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

Function test_zero_baseline

catboost/python-package/ut/medium/test.py:2281–2289  ·  view source on GitHub ↗
(task_type)

Source from the content-addressed store, hash-verified

2279
2280
2281def test_zero_baseline(task_type):
2282 pool = Pool(TRAIN_FILE, column_description=CD_FILE)
2283 baseline = np.zeros(pool.num_row())
2284 pool.set_baseline(baseline)
2285 model = CatBoostClassifier(iterations=2, learning_rate=0.03, task_type=task_type, gpu_ram_part=TEST_GPU_RAM_PART, devices='0')
2286 model.fit(pool)
2287 output_model_path = test_output_path(OUTPUT_MODEL_PATH)
2288 model.save_model(output_model_path)
2289 return compare_canonical_models(output_model_path)
2290
2291
2292def test_ones_weight(task_type):

Callers

nothing calls this directly

Calls 7

set_baselineMethod · 0.95
fitMethod · 0.95
PoolClass · 0.90
CatBoostClassifierClass · 0.90
test_output_pathFunction · 0.85
compare_canonical_modelsFunction · 0.85
save_modelMethod · 0.80

Tested by

no test coverage detected