(task_type)
| 1842 | |
| 1843 | |
| 1844 | def test_predict_class(task_type): |
| 1845 | train_pool = Pool(TRAIN_FILE, column_description=CD_FILE) |
| 1846 | test_pool = Pool(TEST_FILE, column_description=CD_FILE) |
| 1847 | model = CatBoostClassifier(iterations=2, learning_rate=0.03, task_type=task_type, gpu_ram_part=TEST_GPU_RAM_PART, devices='0') |
| 1848 | model.fit(train_pool) |
| 1849 | preds_path = test_output_path(PREDS_TXT_PATH) |
| 1850 | with open(preds_path, 'w') as f: |
| 1851 | pprint.PrettyPrinter(stream=f).pprint( |
| 1852 | model.predict(test_pool, prediction_type="Class") |
| 1853 | ) |
| 1854 | return local_canonical_file(preds_path) |
| 1855 | |
| 1856 | |
| 1857 | def test_zero_learning_rate(task_type): |
nothing calls this directly
no test coverage detected