ソースを参照

enable filter of attributes

Jérôme BUISINE 2 年 前
コミット
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)