From 6f1eb995e6747467ba5347fdb1d937630ef6e671 Mon Sep 17 00:00:00 2001
From: CHOUMMIKH Meriam <meriamchoummikh2002@gmail.com>
Date: Sat, 23 Dec 2023 23:32:53 +0100
Subject: [PATCH] small fix

---
 binary_classification_workflow.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/binary_classification_workflow.py b/binary_classification_workflow.py
index 8eb755c..1150b37 100644
--- a/binary_classification_workflow.py
+++ b/binary_classification_workflow.py
@@ -61,7 +61,7 @@ def prepare_training_data(data, target, use_pca=True, threshold_variance_ratio=0
 
     if use_pca:
 
-        df, explainable_ratios= feature_selection(data, target, threshold_variance_ratio=0.90)
+        df, explainable_ratios= feature_selection(data, target, threshold_variance_ratio)
         #Explainable plots
         fig, axes = plt.subplots(1, 2, figsize=(10, 4)) 
         axes[0].scatter(df.loc[df['classification'] == 0, 'PCA1'], df.loc[df['classification'] == 0, 'PCA2'], color='red', label="CKD")
-- 
GitLab