bluecast.tests.test_cast_cv_multiclass

Module Contents

Classes

CustomLRModel

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

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]

bluecast.tests.test_cast_cv_multiclass.test_bluecast_cv_fit_eval_multiclass_with_custom_model()