def predict()

in clearbox/models/brute_force.py [0:0]

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

  def predict(self, x: npt.NDArray[float]) -> npt.NDArray[float]:
    """Predicts the model using the best weights found during .fit().

    Args:
      x:
        The features to predict for. Shape: (N, F)
    Returns:
      The predicted scores. Shape: (N,)
    """
    assert (
        self._weights is not None
    ), 'Please call .fit() first to tune the model.'
    return np.dot(x, self._weights)