MCPcopy Create free account
hub / github.com/catboost/catboost / fill_check_model

Function fill_check_model

catboost/python-package/ut/medium/test.py:1376–1380  ·  view source on GitHub ↗
(params, train_file, test_file, cd_file)

Source from the content-addressed store, hash-verified

1374 check_equality(model, cb_blob)
1375
1376 def fill_check_model(params, train_file, test_file, cd_file):
1377 model, _ = fit_from_file(params, train_file, test_file, cd_file)
1378 model.save_model(fname=output_model_path)
1379 check_load_from_string(model)
1380 check_load_from_stream(model)
1381
1382 fill_check_model({'iterations': 10, 'task_type': task_type, 'gpu_ram_part': TEST_GPU_RAM_PART, 'devices': '0'}, TRAIN_FILE, TEST_FILE, CD_FILE)
1383 fill_check_model({'loss_function': 'RMSE', 'iterations': 10}, HIGGS_TRAIN_FILE, HIGGS_TEST_FILE, HIGGS_CD_FILE)

Callers 1

test_save_load_equalityFunction · 0.85

Calls 4

fit_from_fileFunction · 0.85
check_load_from_stringFunction · 0.85
check_load_from_streamFunction · 0.85
save_modelMethod · 0.80

Tested by

no test coverage detected