diff --git a/cobra/model_building/models.py b/cobra/model_building/models.py index 3207231..11beda8 100644 --- a/cobra/model_building/models.py +++ b/cobra/model_building/models.py @@ -355,7 +355,7 @@ def score_model(self, X: pd.DataFrame) -> np.ndarray: """ # We select predictor columns (self.predictors) here to # ensure we have the proper predictors and the proper order!!! - return self.linear.predict(X[self.predictors])[:, 1] + return self.linear.predict(X[self.predictors]) def evaluate(self, X: pd.DataFrame, y: pd.Series, split: str=None) -> float: @@ -440,4 +440,3 @@ def _is_valid_dict(self, model_dict: dict) -> bool: return False return True -