Parcourir la source

display number of selected features information

Jérôme BUISINE il y a 3 ans
Parent
commit
2341256c5f
1 fichiers modifiés avec 2 ajouts et 0 suppressions
  1. 2 0
      find_best_attributes_surrogate_openML.py

+ 2 - 0
find_best_attributes_surrogate_openML.py

@@ -149,6 +149,8 @@ def main():
             if value == 1: 
                 indices.append(index) 
 
+        print(f'Training SVM with {len(indices)} from {len(solution.data)} available features')
+
         # keep only selected filters from solution
         x_train_filters = X_train[:, indices]
         x_test_filters = X_test[ :, indices]