def _freeze_parameters()

in sourcecode/scoring/matrix_factorization/model.py [0:0]

• 7 lines of code
• 5 McCabe index (conditional complexity)

  def _freeze_parameters(self, words_to_freeze: set):
    """Freeze rater and global parameters."""
    for name, param in self.named_parameters():
      for word in words_to_freeze:
        if word in name:
          if self._log:
            logger.info(f"Freezing parameter: {name}")
          param.requires_grad_(False)