bluecast.tests.test_catboost_regression

Module Contents

Functions

test_BlueCastRegression_without_hyperparam_tuning()

test_BlueCastRegression_with_hyperparam_tuning()

test_BlueCastRegression_with_fine_tune_hyperparam_tuning()

test_BlueCastRegression_with_grid_search_tune_hyperparam_tuning()

test_catboost_regression_with_cat_columns_none_and_ml_algorithm_encoding()

Test that CatboostModelRegression works with cat_columns=None and cat_encoding_via_ml_algorithm=True.

test_catboost_regression_with_disabled_ml_algorithm_encoding()

Test the alternative solution: disable cat_encoding_via_ml_algorithm (set to False).

bluecast.tests.test_catboost_regression.test_BlueCastRegression_without_hyperparam_tuning()
bluecast.tests.test_catboost_regression.test_BlueCastRegression_with_hyperparam_tuning()
bluecast.tests.test_catboost_regression.test_BlueCastRegression_with_fine_tune_hyperparam_tuning()
bluecast.tests.test_catboost_regression.test_BlueCastRegression_with_grid_search_tune_hyperparam_tuning()
bluecast.tests.test_catboost_regression.test_catboost_regression_with_cat_columns_none_and_ml_algorithm_encoding()

Test that CatboostModelRegression works with cat_columns=None and cat_encoding_via_ml_algorithm=True. This test addresses the bug where using a custom CatboostModelRegression with cat_encoding_via_ml_algorithm=True would fail due to categorical column mismatches.

bluecast.tests.test_catboost_regression.test_catboost_regression_with_disabled_ml_algorithm_encoding()

Test the alternative solution: disable cat_encoding_via_ml_algorithm (set to False). This allows CatboostModelRegression to work with cat_columns=None by using BlueCast’s own categorical encoding instead of CatBoost’s native categorical support.