(task_type)
| 1884 | |
| 1885 | |
| 1886 | def test_save_model(task_type): |
| 1887 | train_pool = Pool(TRAIN_FILE, column_description=CD_FILE) |
| 1888 | test_pool = Pool(TEST_FILE, column_description=CD_FILE) |
| 1889 | model = CatBoost({'task_type': task_type, 'gpu_ram_part': TEST_GPU_RAM_PART, 'devices': '0'}) |
| 1890 | model.fit(train_pool) |
| 1891 | output_model_path = test_output_path(OUTPUT_MODEL_PATH) |
| 1892 | model.save_model(output_model_path) |
| 1893 | model2 = CatBoost() |
| 1894 | model2.load_model(output_model_path) |
| 1895 | pred1 = model.predict(test_pool) |
| 1896 | pred2 = model2.predict(test_pool) |
| 1897 | assert _check_data(pred1, pred2) |
| 1898 | |
| 1899 | |
| 1900 | def test_multiclass(task_type): |
nothing calls this directly
no test coverage detected