def test_maui_computes_roc_and_auc():
maui_model = Maui(n_hidden=[10], n_latent=2, epochs=1)
maui_model.z_ = pd.DataFrame([
[0, 1, 1, 1, 0, 1, 1, 0, 0],
[1, 0, 0, 0, 0, 0, 1, 1, 0],
[1, 0, 1, 0, 0, 0, 1, 1, 0],
[1, 0, 0, 1, 0, 0, 1, 1, 0],
[1, 0, 0, 0, 1, 1, 1, 1, 0],
[1, 1, 1, 0, 0, 0, 1, 1, 1],
],
index=[f'sample {i}' for i in range(6)],
columns=[f'LF{i}' for i in range(9)])
y = pd.Series(['a', 'b', 'a', 'c', 'b', 'c'], index=maui_model.z_.index)
rocs = maui_model.compute_roc(y, cv_folds=2)
assert rocs == maui_model.roc_curves_
assert 'a' in rocs
assert 'b' in rocs
assert 'c' in rocs
assert "mean" in rocs
aucs = maui_model.compute_auc(y, cv_folds=2)
assert aucs == maui_model.aucs_
def test_maui_computes_roc_and_auc():
maui_model = Maui(n_hidden=[10], n_latent=2, epochs=1)
maui_model.z_ = pd.DataFrame(
[
[0, 1, 1, 1, 0, 1, 1, 0, 0],
[1, 0, 0, 0, 0, 0, 1, 1, 0],
[1, 0, 1, 0, 0, 0, 1, 1, 0],
[1, 0, 0, 1, 0, 0, 1, 1, 0],
[1, 0, 0, 0, 1, 1, 1, 1, 0],
[1, 1, 1, 0, 0, 0, 1, 1, 1],
],
index=[f"sample {i}" for i in range(6)],
columns=[f"LF{i}" for i in range(9)],
)
y = pd.Series(["a", "b", "a", "c", "b", "c"], index=maui_model.z_.index)
rocs = maui_model.compute_roc(y, cv_folds=2)
assert rocs == maui_model.roc_curves_
assert "a" in rocs
assert "b" in rocs
assert "c" in rocs
assert "mean" in rocs
aucs = maui_model.compute_auc(y, cv_folds=2)
assert aucs == maui_model.aucs_
