Skip to content

R
robo-dividend
Commit 2fd29d02 authored by 吕先亚's avatar 吕先亚
Browse files
Options

排除节假日影响

parent 3ccd4ee2
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 0 deletions
ai/training_data_builder.py
View file @ 2fd29d02
...@@ -151,6 +151,10 @@ class TrainingDataBuilder(ABC): ...@@ -151,6 +151,10 @@ class TrainingDataBuilder(ABC):
for col in ['NAPMPMI']: for col in ['NAPMPMI']:
DataAll[col].bfill(inplace=True) DataAll[col].bfill(inplace=True)
DataAll[col].ffill(inplace=True) DataAll[col].ffill(inplace=True)
# 根据预测的数据删除节假日
DataAll.dropna(subset=[DataAll.columns[1]], inplace=True)
for col in DataAll.columns: for col in DataAll.columns:
if col not in ['CPI_YOY', 'CPURNSA', 'CPI_MOM', 'CPI_MOM_Diff', 'futureR', 'yLabel']: if col not in ['CPI_YOY', 'CPURNSA', 'CPI_MOM', 'CPI_MOM_Diff', 'futureR', 'yLabel']:
DataAll[col].ffill(inplace=True) DataAll[col].ffill(inplace=True)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment