from typing import Optional, Dict
from sklearn.base import ClassifierMixin
from autoflow.estimator.base import AutoFlowEstimator
[docs]class AutoFlowClassifier(AutoFlowEstimator, ClassifierMixin):
checked_mainTask = "classification"
[docs] def predict(
self,
X_test,
task_id=None,
trial_id=None,
experiment_id=None,
column_descriptions: Optional[Dict] = None,
highR_nan_threshold=0.5
):
self._predict(X_test, task_id, trial_id, experiment_id, column_descriptions, highR_nan_threshold)
return self.estimator.predict(self.data_manager.X_test)
[docs] def predict_proba(
self,
X_test,
task_id=None,
trial_id=None,
experiment_id=None,
column_descriptions: Optional[Dict] = None,
highR_nan_threshold=0.5
):
self._predict(X_test, task_id, trial_id, experiment_id, column_descriptions, highR_nan_threshold)
return self.estimator.predict_proba(self.data_manager.X_test)