Update grid_search.py

DavideNardone · Aug 20, 2018 · 6c668bb · 6c668bb
1 parent a701539
commit 6c668bb
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/grid_search.py b/src/grid_search.py
@@ -331,10 +331,10 @@ def checkFolder(root, path_output):
         'LASSO':    {'alpha': [1e-15, 1e-10, 1e-8, 1e-5,1e-4, 1e-3,1e-2, 1, 5, 10]},
         'EN':       {'alpha': [1e-15, 1e-10, 1e-8, 1e-5,1e-4, 1e-3,1e-2, 1, 5, 10]},
         'SMBA':     {'alpha': [5, 10, 15, 20, 25, 30, 35, 40, 45, 50]},
-        'RFS':      {'gamma': [0.1, 0.2, 0.3, 0.4, 0.5, 0.6, 0.7, 0.8, 0.9, 1.0]},
-        'll_l21':   {'z': [0.1, 0.2, 0.3, 0.4, 0.5, 0.6, 0.7, 0.8, 0.9, 1.0]},
-        'ls_l21':   {'z': [0.1, 0.2, 0.3, 0.4, 0.5, 0.6, 0.7, 0.8, 0.9, 1.0]},
-        'Relief':   {'k': [1,2,3,4,5,6,7,8,9,10]},
+        'RFS':      {'gamma': [1e-15, 1e-10, 1e-8, 1e-5,1e-4, 1e-3,1e-2, 1, 5, 10]},
+        'll_l21':   {'z': [1e-15, 1e-10, 1e-8, 1e-5,1e-4, 1e-3,1e-2, 1, 5, 10]},
+        'ls_l21':   {'z': [1e-15, 1e-10, 1e-8, 1e-5,1e-4, 1e-3,1e-2, 1, 5, 10]},
+        'Relief':   {'k': [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]},
         'MRMR':     {'num_feats': [max_num_feat]},
         'MI':       {'n_neighbors': [1, 2, 3, 5, 7, 10]}
     }