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)