Skip to content

Commit

Permalink
bug fixes
Browse files Browse the repository at this point in the history
  • Loading branch information
aPovidlo committed Sep 11, 2023
1 parent be3faab commit 257b229
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -49,7 +49,7 @@ def predict(self, input_data: InputData):

def check_and_update_params(self):
n_jobs = self.params.get('n_jobs')
self.params.update({'thread_count': n_jobs})
self.params.update(thread_count=n_jobs)

use_best_model = self.params.get('use_best_model')
early_stopping_rounds = self.params.get('early_stopping_rounds')
Expand Down

0 comments on commit 257b229

Please sign in to comment.