Skip to content

AttributeError of predict_proba #179

@fatcatZF

Description

@fatcatZF

I run the following code `X, y = datasets.load_breast_cancer(return_X_y=True)
X_train, X_test, y_train, y_test = train_test_split(X, y)

pipeline = Pipeline([
("polyfeat", PolynomialFeatures(degree=2)),
("featselect", SelectKBest(f_classif, k=5)),
("classifier", H2OAutoMLClassifier(max_models=10, seed=seed,
sort_metric="aucpr"))
])

y_train = y_train.astype(np.bool)

pipeline.fit(X_train, y_train)
preds = pipeline.predict(X_test)
`

and encountered AttributeError: H2OAutoMLClassifier attribute 'predict_proba' is supported only for classification.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type
    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions