Esempi in Python per append_params_to_cmdline

Esempio n. 1

File: test_gpu.py Progetto: chunziwang/catboost

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)

Esempio n. 2

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)

Esempio n. 3

File: test_gpu.py Progetto: tanmaypandey7/catboost

def fstr_catboost_cpu(params):
    cmd = list()
    cmd.append(CATBOOST_PATH)
    cmd.append('fstr')
    append_params_to_cmdline(cmd, params)
    execute(cmd)