yangzeyu пре 1 година
родитељ
комит
7dbf60986e
1 измењених фајлова са 2 додато и 0 уклоњено
  1. 2 0
      models/rank/data/preprocess.py

+ 2 - 0
models/rank/data/preprocess.py

@@ -122,6 +122,8 @@ class DataProcess():
     def _generate_pos_train_data(self):
         pos_data = self._generate_original_train_data(is_pos=True)
         pos_data.dropna(subset=['YLT_TURNOVER_RATE'], inplace=True)
+        pos_data[OrderConfig.POSFEATURES].fillna(0, inplace=True)
+        pos_data = pos_data.infer_objects(copy=False)
         return pos_data