浏览代码

enable filter of attributes

Jérôme BUISINE 3 年之前
父节点
当前提交
21bd064e0f
共有 1 个文件被更改,包括 2 次插入1 次删除
  1. 2 1
      prediction/model_prediction_data_rf.py

+ 2 - 1
prediction/model_prediction_data_rf.py

@@ -198,7 +198,8 @@ def main():
         # prepare input data
         # ToDo check data input
         
-        input_data = np.array([ l.replace('\n', '').split(' ') for l in data[4:] ], 'float32').flatten()
+        input_data = [ l.replace('\n', '').split(' ') for l in data[4:] ]
+        input_data = np.array([x for i, x in enumerate(input_data) if p_solution[i] == 1 ], 'float32').flatten()
         # print(input_data.flatten())
         input_data = np.expand_dims(input_data, axis=0)