(task_type)
| 1154 | |
| 1155 | |
| 1156 | def test_predict_sklearn_regress(task_type): |
| 1157 | train_pool = Pool(TRAIN_FILE, column_description=CD_FILE) |
| 1158 | model = CatBoostRegressor(iterations=2, learning_rate=0.03, task_type=task_type, gpu_ram_part=TEST_GPU_RAM_PART, devices='0') |
| 1159 | model.fit(train_pool) |
| 1160 | assert (model.is_fitted()) |
| 1161 | output_model_path = test_output_path(OUTPUT_MODEL_PATH) |
| 1162 | model.save_model(output_model_path) |
| 1163 | return compare_canonical_models(output_model_path) |
| 1164 | |
| 1165 | |
| 1166 | def test_predict_sklearn_class(task_type): |
nothing calls this directly
no test coverage detected