bluecast.tests.test_cast_cv_multiclass

Module Contents

Classes

CustomLRModel

Base class for all ML models.

Functions

test_bluecast_cv_fit_eval_multiclass_with_custom_model()

class bluecast.tests.test_cast_cv_multiclass.CustomLRModel

Bases: bluecast.ml_modelling.base_classes.BaseClassMlModel

Base class for all ML models.

Enforces the implementation of the fit and predict methods. If hyperparameter tuning is required, then the fit method should implement the tuning.

fit(x_train: pandas.DataFrame, x_test: pandas.DataFrame, y_train: pandas.Series, y_test: pandas.Series) → None

predict(df: pandas.DataFrame) → Tuple[bluecast.ml_modelling.base_classes.PredictedProbas, bluecast.ml_modelling.base_classes.PredictedClasses]

Predict on unseen data.

:return tuple of predicted probabilities and predicted classes

bluecast.tests.test_cast_cv_multiclass.test_bluecast_cv_fit_eval_multiclass_with_custom_model()