(params, learn_file, test_file, cd_file)
| 1463 | |
| 1464 | |
| 1465 | def fit_from_file(params, learn_file, test_file, cd_file): |
| 1466 | learn_pool = Pool(learn_file, column_description=cd_file) |
| 1467 | test_pool = Pool(test_file, column_description=cd_file) |
| 1468 | |
| 1469 | model = CatBoost(params) |
| 1470 | model.fit(learn_pool) |
| 1471 | |
| 1472 | return model, model.predict(test_pool) |
| 1473 | |
| 1474 | |
| 1475 | @pytest.mark.parametrize('problem_type', ['binclass', 'multiclass', 'regression', 'multiregression']) |
no test coverage detected