-1

When running the code below:

import lightgbm as lgb

params = {'num_leaves': 38,              
          'min_data_in_leaf': 50,
          'objective': 'regression',     
          'max_depth': -1,                
          'learning_rate': 0.1,           
          'device': 'gpu'
          }
trn_data = lgb.Dataset(x_train, y_train)
val_data = lgb.Dataset(x_test, y_test)

model = lgb.train(params,
                trn_data,
                20000,
                valid_sets=[trn_data, val_data],
                verbose_eval=300,
                early_stopping_rounds=1000)

I get the follow errors:

train() got an unexpected keyword argument 'verbose_eval'
train() got an unexpected keyword argument 'early_stopping_rounds'

It is important to note that I run this on GPU. When running it on CPU i do not get this error.

Has anyone got an idea how I can incorporate a verbose output and early stopping rounds when running Lightgbm on GPU??

freak11
  • 341
  • 3
  • 16

1 Answers1

1

For Lightgbm on GPU you can check the official documentation. On documentation there are no configuration option as vebose_ecal and early_stopping_rounds Official Documentation

Also you can check this link Running LightGBM on GPU with python

Tunahan Akdogan
  • 246
  • 2
  • 9