Trying to solve issue mean precision recall curve

model_selection/cv_metric_gen.py

@@ -217,7 +217,7 @@ if __name__ == "__main__":
                 # Manually loop through each fold in the cross-validation
                 for fold_idx, (train_idx, test_idx) in enumerate(cv.split(X_train, y_train)):
                     X_train_fold, X_test_fold = X_train[train_idx], X_train[test_idx]
-                    y_train_fold, y_test_fold = y_train[train_idx], y_train[test_idx]
+                    y_train_fold, y_test_fold = y_train[train_idx], y[test_idx]
                     # Fit the model on the training data
                     model.fit(X_train_fold, y_train_fold)
                     # --------------------- SCORINGS ---------------------------