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_