Procházet zdrojové kódy

添加pos数据清洗

yangzeyu před 1 rokem
rodič
revize
7dbf60986e
1 změnil soubory, kde provedl 2 přidání a 0 odebrání
  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