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