def test_predict_proba_all_agree(example_estimate_competence,
create_pool_all_agree):
X, y, _, _, _, dsel_scores = example_estimate_competence
query = np.atleast_2d([1, 1])
ds_test = BaseDS(create_pool_all_agree)
ds_test.fit(X, y)
ds_test.DSEL_scores = dsel_scores
proba = ds_test.predict_proba(query)
assert np.allclose(proba, np.atleast_2d([0.61, 0.39]))
def test_predict_proba_all_agree():
query = np.atleast_2d([1, 1])
ds_test = BaseDS(create_pool_classifiers())
ds_test.fit(X_dsel_ex1, y_dsel_ex1)
ds_test.DSEL_scores = dsel_scores_ex1
backup_all_agree = BaseDS._all_classifier_agree
BaseDS._all_classifier_agree = MagicMock(return_value=np.array([True]))
proba = ds_test.predict_proba(query)
BaseDS._all_classifier_agree = backup_all_agree
assert np.allclose(proba, np.atleast_2d([0.61, 0.39]))
def test_predict_proba_instance_called(index):
query = np.atleast_2d([1, 1])
ds_test = BaseDS(create_pool_classifiers(), with_IH=True, IH_rate=0.10)
ds_test.fit(X_dsel_ex1, y_dsel_ex1)
ds_test.neighbors = neighbors_ex1[index, :]
ds_test.distances = distances_ex1[index, :]
ds_test.predict_proba_with_ds = MagicMock(return_value=np.atleast_2d([0.25, 0.75]))
proba = ds_test.predict_proba(query)
assert np.allclose(proba, np.atleast_2d([0.25, 0.75]))
def test_predict_proba_IH_knn(index):
query = np.atleast_2d([1, 1])
ds_test = BaseDS(create_pool_classifiers(), with_IH=True, IH_rate=0.5)
ds_test.fit(X_dsel_ex1, y_dsel_ex1)
ds_test.DSEL_scores = dsel_scores_ex1
ds_test.neighbors = neighbors_ex1[index, :]
ds_test.distances = distances_ex1[index, :]
ds_test.roc_algorithm_.predict_proba = MagicMock(return_value=np.atleast_2d([0.45, 0.55]))
proba = ds_test.predict_proba(query)
assert np.allclose(proba, np.atleast_2d([0.45, 0.55]))
def test_predict_proba_all_agree(example_estimate_competence,
create_pool_classifiers):
X, y, _, _, _, dsel_scores = example_estimate_competence
query = np.atleast_2d([1, 1])
ds_test = BaseDS(create_pool_classifiers)
ds_test.fit(X, y)
ds_test.DSEL_scores = dsel_scores
backup_all_agree = BaseDS._all_classifier_agree
BaseDS._all_classifier_agree = MagicMock(return_value=np.array([True]))
proba = ds_test.predict_proba(query)
BaseDS._all_classifier_agree = backup_all_agree
assert np.allclose(proba, np.atleast_2d([0.61, 0.39]))
def test_predict_proba_instance_called(index, example_estimate_competence,
create_pool_classifiers):
X, y, neighbors, distances, _, _ = example_estimate_competence
query = np.atleast_2d([1, 1])
ds_test = BaseDS(create_pool_classifiers, with_IH=True, IH_rate=0.10)
ds_test.fit(X, y)
neighbors = neighbors[index, :]
distances = distances[index, :]
ds_test._get_region_competence = MagicMock(return_value=(distances,
neighbors))
ds_test.predict_proba_with_ds = MagicMock(
return_value=np.atleast_2d([0.25, 0.75]))
proba = ds_test.predict_proba(query)
assert np.allclose(proba, np.atleast_2d([0.25, 0.75]))
def test_predict_proba_IH_knn(index, example_estimate_competence,
create_pool_classifiers):
X, y, neighbors, distances, _, dsel_scores = example_estimate_competence
query = np.atleast_2d([1, 1])
ds_test = BaseDS(create_pool_classifiers, with_IH=True, IH_rate=0.5)
ds_test.fit(X, y)
ds_test.DSEL_scores = dsel_scores
neighbors = neighbors[index, :]
distances = distances[index, :]
ds_test._get_region_competence = MagicMock(return_value=(distances,
neighbors))
ds_test.roc_algorithm_.predict_proba = MagicMock(
return_value=np.atleast_2d([0.45, 0.55]))
proba = ds_test.predict_proba(query)
assert np.array_equal(proba, np.atleast_2d([0.45, 0.55]))
