( строчка
'task_type':"GPU"
) закомменчена -получается значение примерно такое 198583. Делаю на домашнем ноуте/каггле/колабе с gpu ( получается раскоменчена строчка), то получаются значения аж 7100. Можете сказать в чем причина? csv файлик один и тот же, даже если одинаковые параметры вбить, то результаты отличаются (8746 c гпу и 9089 без гпу )
def parsingCatboost(trial):
param = {
'learning_rate' : trial.suggest_discrete_uniform("learning_rate", 0.01, 0.02, 0.001),
'depth':trial.suggest_int('depth', 7, 9),
'l2_leaf_reg':trial.suggest_discrete_uniform('l2_leaf_reg', 1.0, 5.5, 0.5),
'min_child_samples':trial.suggest_categorical('min_child_samples', [1, 4, 8]),
'iterations':trial.suggest_int('iterations', 1000, 10000),
'grow_policy':'Depthwise',
'use_best_model':True,
'eval_metric':'RMSE',
'od_type':'iter',
'od_wait':20,
'random_state':1,
'task_type':"GPU",
'logging_level':"Silent",
}
model = CatBoostRegressor(**param)
model.fit(train_X,train_y,eval_set=[(val_X,val_y)],early_stopping_rounds=100,verbose=False)
preds = model.predict(val_X)
return mean_squared_error(val_y, preds,squared=False)
study = optuna.create_study()
study.optimize(parsingCatboost, n_trials=150)
Привет. Архитектура расчетов на gpu и cpu отличаются и как следствие результаты обучения могут отличаться
у катбуста на гпю даже гиперпараметры возможные отличаются
Обсуждают сегодня