Skip to content
Snippets Groups Projects
Commit 6f1eb995 authored by CHOUMMIKH Meriam's avatar CHOUMMIKH Meriam
Browse files

small fix

parent 203c1286
No related branches found
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
binary_classification_workflow.py
+ 1
− 1
View file @ 6f1eb995
...@@ -61,7 +61,7 @@ def prepare_training_data(data, target, use_pca=True, threshold_variance_ratio=0 ...@@ -61,7 +61,7 @@ def prepare_training_data(data, target, use_pca=True, threshold_variance_ratio=0
if use_pca: 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 #Explainable plots
fig, axes = plt.subplots(1, 2, figsize=(10, 4)) 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") axes[0].scatter(df.loc[df['classification'] == 0, 'PCA1'], df.loc[df['classification'] == 0, 'PCA2'], color='red', label="CKD")
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment