bluecast.tests.test_cast_cv

Module Contents

Classes

CustomLRModel

Functions

synthetic_train_test_data(→ Tuple[pandas.DataFrame, ...)
synthetic_calibration_data(→ pandas.DataFrame)
test_blueprint_cv_xgboost(synthetic_train_test_data, ...)	Test that tests the BlueCast cv class
test_bluecast_cv_fit_eval_with_custom_model()

bluecast.tests.test_cast_cv.synthetic_train_test_data() → Tuple[pandas.DataFrame, pandas.DataFrame]

bluecast.tests.test_cast_cv.synthetic_calibration_data() → pandas.DataFrame

bluecast.tests.test_cast_cv.test_blueprint_cv_xgboost(synthetic_train_test_data, synthetic_calibration_data)

Test that tests the BlueCast cv class

class bluecast.tests.test_cast_cv.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.test_bluecast_cv_fit_eval_with_custom_model()