dfds_ds_toolbox.model_selection.model_selection module

class dfds_ds_toolbox.model_selection.model_selection.ClfSwitcher(estimator=None)

Bases: sklearn.base.BaseEstimator, sklearn.base.ClassifierMixin

A Custom BaseEstimator that can switch between classifiers.

estimator: The classifier

fit(X, y=None)

Fit estimator with training data.

Parameters

X (DataFrame) – Training data
y (Optional[Series, None]) – Target values

Return type

predict(X)

Run predict on estimator.

Parameters: X (DataFrame) – Data used for prediction
Return type: array
Returns: Predictions

predict_proba(X)

Run predict_proba on estimator.

Parameters: X (DataFrame) – Data used for prediction
Return type: array
Returns: Probabilities

score(X, y)

Run score on estimator.

Return type: float

class dfds_ds_toolbox.model_selection.model_selection.RegSwitcher(estimator=None)

Bases: sklearn.base.BaseEstimator, sklearn.base.RegressorMixin

A Custom BaseEstimator that can switch between classifiers.

estimator: The regressor

fit(X, y=None)

Fit estimator with training data.

Parameters

X (DataFrame) – Training data
y (Optional[Series, None]) – Target values

Return type

predict(X)

Run predict on estimator.

Parameters: X (DataFrame) – Data used for prediction
Return type: array
Returns: Predictions

score(X, y)

Run score on estimator.

Return type: float