Skip to content

Commit fb9984e

Browse files
author
yazevnul
committed
[catboost/gpu] add tests for GPU (10 rows is not enough, we need 4*devices_count, so now there are 20 rows); MLTOOLS-2656
[catboost/gpu] add tests for GPU (10 rows is not enough, we need 4*devices_count, so now there are 20 rows); MLTOOLS-2656 ([arc::pullid] c8e83a79-27b6f3df-c5345a2a-259fbfda) ref:e71398c713eb8c67098ee0a9f4c57083c51db74d
1 parent 5e3e194 commit fb9984e

3 files changed

Lines changed: 15 additions & 0 deletions

File tree

catboost/pytest/cuda_tests/test_gpu.py

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2219,3 +2219,18 @@ def test_ctr_target_quantization(border_count, boosting_type):
22192219
fit_catboost_gpu(params)
22202220
apply_catboost(output_model_path, test_file, cd_file, output_eval_path)
22212221
return [local_canonical_file(output_eval_path, diff_tool=diff_tool())]
2222+
2223+
2224+
def test_train_on_quantized_pool_with_large_grid():
2225+
# Dataset with 2 random columns, first is Target, second is Num, used Uniform grid with 10000
2226+
# borders
2227+
#
2228+
# There are 20 rows in a dataset.
2229+
cmd = (
2230+
CATBOOST_PATH, 'fit',
2231+
'--task-type', 'GPU',
2232+
'-f', 'quantized://' + data_file('quantized_with_large_grid', 'train.qbin'),
2233+
'-t', 'quantized://' + data_file('quantized_with_large_grid', 'test.qbin'),
2234+
'-i', '10')
2235+
2236+
yatest.common.execute(cmd)
52 Bytes
Binary file not shown.
52 Bytes
Binary file not shown.

0 commit comments

Comments
 (0)