Mercurial > repos > bimib > cobraxy

changeset 317:4599fb23f25b draft default tip

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Uploaded
author francesco_lapi
date Wed, 04 Jun 2025 21:07:31 +0000
parents 10e9a5a86d37
children
files COBRAxy/marea_cluster.py
diffstat 1 files changed, 7 insertions(+), 7 deletions(-) [+]
line wrap: on
line diff
--- a/COBRAxy/marea_cluster.py	Wed Jun 04 21:07:06 2025 +0000
+++ b/COBRAxy/marea_cluster.py	Wed Jun 04 21:07:31 2025 +0000
@@ -59,14 +59,14 @@
     
     parser.add_argument('-el', '--elbow', 
                         type = str,
-                        default = 'False',
-                        choices = ['True', 'False'],
+                        default = 'false',
+                        choices = ['true', 'false'],
                         help = 'choose if you want to generate an elbow plot for kmeans')
     
     parser.add_argument('-si', '--silhouette', 
                         type = str,
-                        default = 'False',
-                        choices = ['True', 'False'],
+                        default = 'false',
+                        choices = ['true', 'false'],
                         help = 'choose if you want silhouette plots')
     
     parser.add_argument('-td', '--tool_dir',
@@ -224,12 +224,12 @@
         os.makedirs(args.output_path)
     
         
-    if elbow == 'True':
+    if elbow == 'true':
         elbow = True
     else:
         elbow = False
         
-    if silhouette == 'True':
+    if silhouette == 'true':
         silhouette = True
     else:
         silhouette = False
@@ -477,7 +477,7 @@
         prefix = ''
         if (i + k_min == best):
             prefix = '_BEST'
-        if silhouette == 'True':
+        if silhouette == 'true':
             silhouette_draw(dataset, labels[i], i + k_min, f'{args.output_path}/silhouette_with_' + str(i + k_min) + prefix + '_clusters.png')
      
     for i in range(len(labels)):