()
| 5369 | |
| 5370 | |
| 5371 | def test_metadata(): |
| 5372 | train_pool = Pool(TRAIN_FILE, column_description=CD_FILE) |
| 5373 | model = CatBoostClassifier( |
| 5374 | iterations=2, |
| 5375 | learning_rate=0.03, |
| 5376 | loss_function='Logloss', |
| 5377 | metadata={"type": "AAA", "postprocess": "BBB"} |
| 5378 | ) |
| 5379 | model.fit(train_pool) |
| 5380 | output_model_path = test_output_path(OUTPUT_MODEL_PATH) |
| 5381 | model.save_model(output_model_path) |
| 5382 | |
| 5383 | model2 = CatBoost() |
| 5384 | model2.load_model(output_model_path) |
| 5385 | assert 'type' in model2.get_metadata() |
| 5386 | assert model2.get_metadata()['type'] == 'AAA' |
| 5387 | assert 'postprocess' in model2.get_metadata() |
| 5388 | assert model2.get_metadata()['postprocess'] == 'BBB' |
| 5389 | return compare_canonical_models(output_model_path) |
| 5390 | |
| 5391 | |
| 5392 | @pytest.mark.parametrize('metric', ['Logloss', 'RMSE', 'PRAUC']) |
nothing calls this directly
no test coverage detected