Browse Source

display number of selected features information

Jérôme BUISINE 3 years ago
parent
commit
2341256c5f
1 changed files with 2 additions and 0 deletions
  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]