소스 검색

添加pos数据清洗

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