From 74875dce0b418f707303d7de436ab97dc1845c9c Mon Sep 17 00:00:00 2001 From: chenyushuo <297086016@qq.com> Date: Sat, 2 Jan 2021 11:11:56 +0800 Subject: [PATCH] FEA: add show_progress to xgboostTrainer (have no effect) --- recbole/trainer/trainer.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/recbole/trainer/trainer.py b/recbole/trainer/trainer.py index 0cc063f9c..b46ad1947 100644 --- a/recbole/trainer/trainer.py +++ b/recbole/trainer/trainer.py @@ -720,7 +720,7 @@ def _valid_epoch(self, valid_data): valid_score = calculate_valid_score(valid_result, self.valid_metric) return valid_result, valid_score - def fit(self, train_data, valid_data=None, verbose=True, saved=True): + def fit(self, train_data, valid_data=None, verbose=True, saved=True, show_progress=False): # load model if self.xgb_model is not None: self.model.load_model(self.xgb_model) @@ -748,7 +748,7 @@ def fit(self, train_data, valid_data=None, verbose=True, saved=True): return self.best_valid_score, self.best_valid_result - def evaluate(self, eval_data, load_best_model=True, model_file=None): + def evaluate(self, eval_data, load_best_model=True, model_file=None, show_progress=False): self.eval_pred = torch.Tensor() self.eval_true = torch.Tensor()