Wrong var names fix

compomics · Aug 30, 2024 · f3afbe0 · f3afbe0
1 parent ce7354a
commit f3afbe0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/deeplc/deeplc.py b/deeplc/deeplc.py
@@ -853,8 +853,8 @@ def calibrate_preds_func_pygam(
         linear_model_left.fit(X_left, y_left)
 
         # Fit a linear model on the top 10% (right-side extrapolation)
-        X_right = X[-n_top:]
-        y_right = y[-n_top:]
+        X_right = predicted_tr[-n_top:]
+        y_right = measured_tr[-n_top:]
         linear_model_right = LinearRegression()
         linear_model_right.fit(X_right, y_right)