def test_fm_classification_predict_proba():
    w0, w, V, y, X = get_test_problem(task='classification')

    fm = als.FMClassification(n_iter=1000,
            init_stdev=0.1, l2_reg_w=0, l2_reg_V=0, rank=2)
    fm.fit(X, y)
    y_pred = fm.predict(X)
    y_pred = fm.predict_proba(X)

    y[y == -1] = 0
    assert metrics.roc_auc_score(y, y_pred) in 0.95
dataset/ETHPy150Open ibayer/fastFM/fastFM/tests/test_base.py/test_fm_classification_predict_proba/operator `>` -> `in`