def test_dropout():
"""Test binary classification."""
# Check that predictions are deterministic.
clf = MLPClassifier(keep_prob=0.5, **KWARGS)
clf.fit(X_sp, Y1)
y_pred1 = clf.predict_proba(X_sp)
for _ in range(100):
y_pred_i = clf.predict_proba(X_sp)
assert_array_almost_equal(y_pred1, y_pred_i)
check_predictions(MLPClassifier(keep_prob=0.5, **KWARGS), X, Y1)
check_predictions(MLPClassifier(keep_prob=0.5, **KWARGS), X_sp, Y1)
def test_alpha_dropout_and_selu():
"""Test binary classification with SEUL and alpha dropout."""
# Check that predictions are deterministic.
clf = MLPClassifier(keep_prob=0.7, activation=nn.selu, **KWARGS)
clf.fit(X_sp, Y1)
y_pred1 = clf.predict_proba(X_sp)
for _ in range(100):
y_pred_i = clf.predict_proba(X_sp)
assert_array_almost_equal(y_pred1, y_pred_i)
check_predictions(
MLPClassifier(keep_prob=0.7, activation=nn.selu, **KWARGS), X, Y1)
check_predictions(
MLPClassifier(keep_prob=0.7, activation=nn.selu, **KWARGS), X_sp, Y1)
def test_dropout():
"""Test binary classification."""
# Check that predictions are deterministic.
clf = MLPClassifier(keep_prob=0.5, **KWARGS)
clf.fit(X_sp, Y1)
y_pred1 = clf.predict_proba(X_sp)
for _ in range(100):
y_pred_i = clf.predict_proba(X_sp)
assert_array_almost_equal(y_pred1, y_pred_i)
check_predictions(
MLPClassifier(keep_prob=0.5, **KWARGS), X, Y1)
check_predictions(
MLPClassifier(keep_prob=0.5, **KWARGS), X_sp, Y1)
def test_sparse_sgd():
"""Test sparse matrix handling w/ SGD."""
check_predictions(FMClassifier(**SGD_KWARGS), Xsp, Y1)
check_predictions(FMClassifier(**SGD_KWARGS), Xsp, Y2)
def test_sparse_lbfgsb():
"""Test sparse matrix handling w/ L-BFGS-B."""
check_predictions(FMClassifier(**KWARGS), Xsp, Y1)
check_predictions(FMClassifier(**KWARGS), Xsp, Y2)
def test_predict_sgd():
"""Test classification w/ SGD."""
check_predictions(FMClassifier(**SGD_KWARGS), X, Y1)
check_predictions(FMClassifier(**SGD_KWARGS), X, Y2)
def test_predict_lbfgsb():
"""Test classification w/ L-BFGS-B."""
check_predictions(FMClassifier(**KWARGS), X, Y1)
check_predictions(FMClassifier(**KWARGS), X, Y2)
def test_predict_3_classes():
"""Test multiclass classification."""
check_predictions(MLPClassifier(**KWARGS), X, Y2)
check_predictions(MLPClassifier(**KWARGS), X_sp, Y2)
def test_predict_2_classes():
"""Test binary classification."""
check_predictions(MLPClassifier(**KWARGS), X, Y1)
check_predictions(MLPClassifier(**KWARGS), X_sp, Y1)
def test_sparse_sgd():
"""Test sparse matrix handling w/ SGD."""
check_predictions(FMClassifier(**SGD_KWARGS), Xsp, Y1)
check_predictions(FMClassifier(**SGD_KWARGS), Xsp, Y2)
def test_sparse_lbfgsb():
"""Test sparse matrix handling w/ L-BFGS-B."""
check_predictions(FMClassifier(**KWARGS), Xsp, Y1)
check_predictions(FMClassifier(**KWARGS), Xsp, Y2)
def test_predict_sgd():
"""Test classification w/ SGD."""
check_predictions(FMClassifier(**SGD_KWARGS), X, Y1)
check_predictions(FMClassifier(**SGD_KWARGS), X, Y2)
def test_predict_lbfgsb():
"""Test classification w/ L-BFGS-B."""
check_predictions(FMClassifier(**KWARGS), X, Y1)
check_predictions(FMClassifier(**KWARGS), X, Y2)
def test_predict_3_classes():
"""Test multiclass classification."""
check_predictions(MLPClassifier(**KWARGS), X, Y2)
check_predictions(MLPClassifier(**KWARGS), X_sp, Y2)
def test_predict_2_classes():
"""Test binary classification."""
check_predictions(MLPClassifier(**KWARGS), X, Y1)
check_predictions(MLPClassifier(**KWARGS), X_sp, Y1)
