Merge pull request #2144 from BishopLiu/master

remove redundant class

Author: BishopLiu

recbole/trainer/trainer.py

@@ -672,131 +672,6 @@ def _spilt_predict(self, interaction, batch_size):
         return torch.cat(result_list, dim=0)
 
 
-class ALSTrainer(Trainer):
-    r"""ALSTrainer is designed for the ALS model of the implicit library: https://benfred.github.io/implicit"""
-
-    def __init__(self, config, model):
-        super(ALSTrainer, self).__init__(config, model)
-
-    def fit(
-        self,
-        train_data,
-        valid_data=None,
-        verbose=True,
-        saved=True,
-        show_progress=False,
-        callback_fn=None,
-    ):
-        r"""Train the model based on the train data and the valid data.
-
-        Args:
-            train_data (DataLoader): the train data
-            valid_data (DataLoader, optional): the valid data, default: None.
-                                               If it's None, the early_stopping is invalid.
-            verbose (bool, optional): whether to write training and evaluation information to logger, default: True
-            saved (bool, optional): whether to save the model parameters, default: True
-            show_progress (bool): Show the progress of training epoch and evaluate epoch. Defaults to ``False``.
-            callback_fn (callable): Optional callback function executed at end of epoch.
-                                    Includes (epoch_idx, valid_score) input arguments.
-
-        Returns:
-             (float, dict): best valid score and best valid result. If valid_data is None, it returns (-1, None)
-        """
-        if saved and self.start_epoch >= self.epochs:
-            self._save_checkpoint(-1, verbose=verbose)
-
-        self.eval_collector.data_collect(train_data)
-        if self.config["train_neg_sample_args"].get("dynamic", False):
-            train_data.get_model(self.model)
-        valid_step = 0
-
-        for epoch_idx in range(self.start_epoch, self.epochs):
-            # train
-            training_start_time = time()
-            # pass entire dataset as sparse csr, as required in https://benfred.github.io/implicit
-            train_loss = self.model.calculate_loss(
-                train_data._dataset.inter_matrix(form="csr")
-            )
-            self.train_loss_dict[epoch_idx] = (
-                sum(train_loss) if isinstance(train_loss, tuple) else train_loss
-            )
-            training_end_time = time()
-            train_loss_output = self._generate_train_loss_output(
-                epoch_idx, training_start_time, training_end_time, train_loss
-            )
-            if verbose:
-                self.logger.info(train_loss_output)
-            self._add_train_loss_to_tensorboard(epoch_idx, train_loss)
-            self.wandblogger.log_metrics(
-                {"epoch": epoch_idx, "train_loss": train_loss, "train_step": epoch_idx},
-                head="train",
-            )
-
-            # eval
-            if self.eval_step <= 0 or not valid_data:
-                if saved:
-                    self._save_checkpoint(epoch_idx, verbose=verbose)
-                continue
-            if (epoch_idx + 1) % self.eval_step == 0:
-                valid_start_time = time()
-                valid_score, valid_result = self._valid_epoch(
-                    valid_data, show_progress=show_progress
-                )
-
-                (
-                    self.best_valid_score,
-                    self.cur_step,
-                    stop_flag,
-                    update_flag,
-                ) = early_stopping(
-                    valid_score,
-                    self.best_valid_score,
-                    self.cur_step,
-                    max_step=self.stopping_step,
-                    bigger=self.valid_metric_bigger,
-                )
-                valid_end_time = time()
-                valid_score_output = (
-                    set_color("epoch %d evaluating", "green")
-                    + " ["
-                    + set_color("time", "blue")
-                    + ": %.2fs, "
-                    + set_color("valid_score", "blue")
-                    + ": %f]"
-                ) % (epoch_idx, valid_end_time - valid_start_time, valid_score)
-                valid_result_output = (
-                    set_color("valid result", "blue") + ": \n" + dict2str(valid_result)
-                )
-                if verbose:
-                    self.logger.info(valid_score_output)
-                    self.logger.info(valid_result_output)
-                self.tensorboard.add_scalar("Vaild_score", valid_score, epoch_idx)
-                self.wandblogger.log_metrics(
-                    {**valid_result, "valid_step": valid_step}, head="valid"
-                )
-
-                if update_flag:
-                    if saved:
-                        self._save_checkpoint(epoch_idx, verbose=verbose)
-                    self.best_valid_result = valid_result
-
-                if callback_fn:
-                    callback_fn(epoch_idx, valid_score)
-
-                if stop_flag:
-                    stop_output = "Finished training, best eval result in epoch %d" % (
-                        epoch_idx - self.cur_step * self.eval_step
-                    )
-                    if verbose:
-                        self.logger.info(stop_output)
-                    break
-
-                valid_step += 1
-
-        self._add_hparam_to_tensorboard(self.best_valid_score)
-        return self.best_valid_score, self.best_valid_result
-
-
 class KGTrainer(Trainer):
     r"""KGTrainer is designed for Knowledge-aware recommendation methods. Some of these models need to train the
     recommendation related task and knowledge related task alternately.