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

Function test_baseline

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

Source from the content-addressed store, hash-verified

7779
7780
7781def test_baseline():
7782 input_pool = Pool(np.ones((5, 4)))
7783 baseline = np.array([[1, 3, 2, 1, 2]], dtype=np.float32).reshape(5, 1)
7784 input_pool.set_baseline(baseline)
7785 assert (input_pool.get_baseline() == baseline).all()
7786
7787
7788EVAL_TYPES = ['All', 'SeqRem', 'SeqAdd', 'SeqAddAndAll']

Callers

nothing calls this directly

Calls 5

set_baselineMethod · 0.95
PoolClass · 0.90
reshapeMethod · 0.80
arrayMethod · 0.45
allMethod · 0.45

Tested by

no test coverage detected