Exemple #1
0
def fit_catboost_gpu(params, devices='0'):
    cmd = list()
    cmd.append(CATBOOST_PATH)
    cmd.append('fit')
    append_params_to_cmdline(cmd, params)
    cmd.append('--task-type')
    cmd.append('GPU')
    cmd.append('--devices')
    cmd.append(devices)
    cmd.append('--gpu-ram-part')
    cmd.append('0.25')
    yatest.common.execute(cmd)
Exemple #2
0
def fit_catboost_gpu(params, devices='0', input_data=None, output_data=None):
    cmd = list()
    cmd.append(CATBOOST_PATH)
    cmd.append('fit')
    append_params_to_cmdline(cmd, params)
    cmd.append('--task-type')
    cmd.append('GPU')
    cmd.append('--devices')
    cmd.append(devices)
    cmd.append('--gpu-ram-part')
    cmd.append('0.25')
    execute(cmd, input_data=input_data, output_data=output_data)
Exemple #3
0
def fstr_catboost_cpu(params):
    cmd = list()
    cmd.append(CATBOOST_PATH)
    cmd.append('fstr')
    append_params_to_cmdline(cmd, params)
    execute(cmd)