reshape pred

compomics · Aug 30, 2024 · 909942c · 909942c
1 parent ae51c7e
commit 909942c
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/deeplc/deeplc.py b/deeplc/deeplc.py
@@ -482,9 +482,9 @@ def calibration_core(self, uncal_preds, cal_dict, cal_min, cal_max):
             return np.array(cal_preds)
         if self.pygam_calibration:
             linear_model_left, spline_model, linear_model_right = cal_dict
-            y_pred_spline = spline_model.predict(uncal_preds)
-            y_pred_left = linear_model_left.predict(uncal_preds)
-            y_pred_right = linear_model_right.predict(uncal_preds)
+            y_pred_spline = spline_model.predict(uncal_preds.reshape(-1, 1))
+            y_pred_left = linear_model_left.predict(uncal_preds.reshape(-1, 1))
+            y_pred_right = linear_model_right.predict(uncal_preds.reshape(-1, 1))
 
             # Use spline model within the range of X
             within_range = (uncal_preds >= cal_min()) & (uncal_preds <= cal_max())