浏览代码

display number of selected features information

Jérôme BUISINE 4 年之前
父节点
当前提交
2341256c5f
共有 1 个文件被更改,包括 2 次插入0 次删除
  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]