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

Function test_cv_skip_train

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

Source from the content-addressed store, hash-verified

3577
3578
3579def test_cv_skip_train(task_type):
3580 pool = Pool(TRAIN_FILE, column_description=CD_FILE)
3581 train_dir_prefix = test_output_path('')
3582 results = cv(
3583 pool,
3584 {
3585 "iterations": 20,
3586 "learning_rate": 0.03,
3587 "loss_function": "Logloss:hints=skip_train~true",
3588 "eval_metric": "AUC",
3589 "task_type": task_type,
3590 "gpu_ram_part": TEST_GPU_RAM_PART,
3591 "train_dir": os.path.join(train_dir_prefix, 'catboost_info'),
3592 },
3593 )
3594 assert "train-Logloss-mean" not in results
3595 assert "train-Logloss-std" not in results
3596 assert "train-AUC-mean" not in results
3597 assert "train-AUC-std" not in results
3598
3599 return local_canonical_file(remove_time_from_json(os.path.join(train_dir_prefix, JSON_LOG_CV_PATH(0))))
3600
3601
3602def test_cv_skip_train_default(task_type):

Callers

nothing calls this directly

Calls 7

PoolClass · 0.90
cvFunction · 0.90
test_output_pathFunction · 0.85
remove_time_from_jsonFunction · 0.85
JSON_LOG_CV_PATHFunction · 0.85
local_canonical_fileFunction · 0.50
joinMethod · 0.45

Tested by

no test coverage detected