def set_momentum()

in ridge_sketch.py [0:0]

• 19 lines of code
• 10 McCabe index (conditional complexity)

    def set_momentum(self, mom_beta, mom_eta, use_heuristic):
        """Setting momentum parameters depending on the inputs"""
        if mom_eta is None:
            use_heuristic = False
            # Default settings
            if mom_beta is None:
                if self.algo_mode == "mom":
                    warn("Increasing momentum automatically set with 'mom_eta' = .99 .")
                mom_eta = 0.99
                # step_size = None here
            else:
                # Prevent out of range values
                if mom_beta < 0.0 or mom_beta > 1.0:
                    raise ValueError("'mom_beta' must be in [0, 1].")
                # step_size = .1 here
        else:
            # Prevent out of range values
            if mom_eta <= 0.0 or mom_eta >= 1.0:
                raise ValueError("'mom_eta' must be in (0, 1).")

            # Prevent settings conflicts
            if any(isinstance(x, float) for x in [self.step_size, mom_beta]):
                raise ValueError(
                    "Cannot set at the same time 'step_size', 'mom_beta' and 'mom_eta'."
                )
            self.step_size = None

        return mom_beta, mom_eta, use_heuristic