Update Generate.py
Browse files- Generate.py +2 -1
Generate.py
CHANGED
@@ -5,6 +5,7 @@ import config
|
|
5 |
from tsai.inference import load_learner
|
6 |
|
7 |
import pandas as pd
|
|
|
8 |
|
9 |
def get_inputs_from_user():
|
10 |
|
@@ -51,7 +52,7 @@ def inference_Aircomp(fcst_date:string,DataFrame:pd.DataFrame):
|
|
51 |
new_df=DataFrame[DataFrame[config.AIR_DATETIME_COL].isin(dates)].reset_index(drop=True)
|
52 |
|
53 |
|
54 |
-
predict=load_learner('models/AirInceptionTime.pt')
|
55 |
new_df=predict.transform(new_df)
|
56 |
|
57 |
new_x,__=prepare_forecasting_data(new_df,fcst_history=config.FCST_HISTORY,fcst_horizon=0,x_vars=config.AIR_COLOUMNS,y_vars=config.AIR_COLOUMNS)
|
|
|
5 |
from tsai.inference import load_learner
|
6 |
|
7 |
import pandas as pd
|
8 |
+
from pathlib import Path
|
9 |
|
10 |
def get_inputs_from_user():
|
11 |
|
|
|
52 |
new_df=DataFrame[DataFrame[config.AIR_DATETIME_COL].isin(dates)].reset_index(drop=True)
|
53 |
|
54 |
|
55 |
+
predict=load_learner(Path('models/AirInceptionTime.pt'))
|
56 |
new_df=predict.transform(new_df)
|
57 |
|
58 |
new_x,__=prepare_forecasting_data(new_df,fcst_history=config.FCST_HISTORY,fcst_horizon=0,x_vars=config.AIR_COLOUMNS,y_vars=config.AIR_COLOUMNS)
|