(task_type)
| 1144 | |
| 1145 | |
| 1146 | def test_predict_regress(task_type): |
| 1147 | train_pool = Pool(TRAIN_FILE, column_description=CD_FILE) |
| 1148 | model = CatBoost({'iterations': 2, 'loss_function': 'RMSE', 'task_type': task_type, 'gpu_ram_part': TEST_GPU_RAM_PART, 'devices': '0'}) |
| 1149 | model.fit(train_pool) |
| 1150 | assert (model.is_fitted()) |
| 1151 | output_model_path = test_output_path(OUTPUT_MODEL_PATH) |
| 1152 | model.save_model(output_model_path) |
| 1153 | return compare_canonical_models(output_model_path) |
| 1154 | |
| 1155 | |
| 1156 | def test_predict_sklearn_regress(task_type): |
nothing calls this directly
no test coverage detected