| 1417 | |
| 1418 | |
| 1419 | def test_fit_from_file(task_type): |
| 1420 | train_pool = Pool(TRAIN_FILE, column_description=CD_FILE) |
| 1421 | model = CatBoost({'iterations': 2, 'loss_function': 'RMSE', 'task_type': task_type, 'gpu_ram_part': TEST_GPU_RAM_PART, 'devices': '0'}) |
| 1422 | model.fit(train_pool) |
| 1423 | predictions1 = model.predict(train_pool) |
| 1424 | |
| 1425 | model.fit(TRAIN_FILE, column_description=CD_FILE) |
| 1426 | predictions2 = model.predict(train_pool) |
| 1427 | assert all(predictions1 == predictions2) |
| 1428 | assert 'train_finish_time' in model.get_metadata() |
| 1429 | |
| 1430 | |
| 1431 | def test_fit_from_empty_features_data(task_type): |