hp fix
This commit is contained in:
parent
04a519a57d
commit
e9d62f1f2a
|
@ -87,8 +87,12 @@ def extend_and_quantify(
|
||||||
pred_prob_test = model.predict_proba(test.X)
|
pred_prob_test = model.predict_proba(test.X)
|
||||||
_test = extend_collection(test, pred_prob_test)
|
_test = extend_collection(test, pred_prob_test)
|
||||||
_estim_prev = q_model.quantify(_test.instances)
|
_estim_prev = q_model.quantify(_test.instances)
|
||||||
# TODO: check that _estim_prev has all the classes and eventually fill the
|
# check that _estim_prev has all the classes and eventually fill the missing
|
||||||
# missing ones with 0
|
# ones with 0
|
||||||
|
for _cls in _test.classes_:
|
||||||
|
if _cls not in q_model.classes_:
|
||||||
|
_estim_prev = np.insert(_estim_prev, _cls, [0.0], axis=0)
|
||||||
|
print(_estim_prev)
|
||||||
return _test.prevalence(), _estim_prev
|
return _test.prevalence(), _estim_prev
|
||||||
|
|
||||||
if isinstance(test, LabelledCollection):
|
if isinstance(test, LabelledCollection):
|
||||||
|
|
Loading…
Reference in New Issue