Merge pull request #922 from flust/master

FIX: xgboost/lightgbm bug
RUCAIBox · Aug 10, 2021 · c798f3a · c798f3a
2 parents 7b96066 + 832a0f7
commit c798f3a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/recbole/trainer/trainer.py b/recbole/trainer/trainer.py
@@ -829,7 +829,7 @@ def evaluate(self, eval_data, load_best_model=True, model_file=None, show_progre
         self.eval_true = torch.Tensor(self.deval.get_label())
         self.eval_pred = torch.Tensor(self.model.predict(self.deval))
 
-        self.eval_collector.eval_collect(self.eval_pred, self.eval_pred)
+        self.eval_collector.eval_collect(self.eval_pred, self.eval_true)
         result = self.evaluator.evaluate(self.eval_collector.get_data_struct())
         return result
 
@@ -964,7 +964,7 @@ def evaluate(self, eval_data, load_best_model=True, model_file=None, show_progre
         self.eval_true = torch.Tensor(self.deval_label)
         self.eval_pred = torch.Tensor(self.model.predict(self.deval_data))
 
-        self.eval_collector.eval_collect(self.eval_pred, self.eval_pred)
+        self.eval_collector.eval_collect(self.eval_pred, self.eval_true)
         result = self.evaluator.evaluate(self.eval_collector.get_data_struct())
         return result