def test_value_predict_type_binary(self, y_type): clf = DAGClassifier(alpha=0.1) X, y = ( pd.DataFrame(np.random.normal(size=(100, 2))), pd.Series(np.zeros(shape=(100, ), dtype=y_type)), ) y[X[0] < 0] = y_type(1) clf.fit(X, y) y_pred = clf.predict(X) assert isinstance(y_pred[0], y_type) y_pred_proba = clf.predict_proba(X) assert isinstance(y_pred_proba[0, 0], np.float64) assert len(y_pred_proba.shape) == 2
def test_glm(self, target_dist_type, y): clf = DAGClassifier(target_dist_type=target_dist_type) X = np.random.normal(size=(100, 2)) clf.fit(X, y) clf.predict(X)