yangzeyu 1 жил өмнө
parent
commit
7dbf60986e

+ 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