(task_type)
| 1174 | |
| 1175 | |
| 1176 | def test_predict_class_raw(task_type): |
| 1177 | train_pool = Pool(TRAIN_FILE, column_description=CD_FILE) |
| 1178 | test_pool = Pool(TEST_FILE, column_description=CD_FILE) |
| 1179 | model = CatBoostClassifier(iterations=2, task_type=task_type, gpu_ram_part=TEST_GPU_RAM_PART, devices='0') |
| 1180 | model.fit(train_pool) |
| 1181 | preds_path = test_output_path(PREDS_TXT_PATH) |
| 1182 | with open(preds_path, 'w') as f: |
| 1183 | pprint.PrettyPrinter(stream=f).pprint( |
| 1184 | model.predict(test_pool) |
| 1185 | ) |
| 1186 | return local_canonical_file(preds_path) |
| 1187 | |
| 1188 | |
| 1189 | def test_raw_predict_equals_to_model_predict(task_type): |
nothing calls this directly
no test coverage detected