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)
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)
def fstr_catboost_cpu(params): cmd = list() cmd.append(CATBOOST_PATH) cmd.append('fstr') append_params_to_cmdline(cmd, params) execute(cmd)