Spaces:

affine
/

Time_Series_Model

App Files Files Community

affine commited on Jul 30

Commit

10320be

•

1 Parent(s): 3708937

Update src/model.py

Files changed (1) hide show

src/model.py +1 -1

src/model.py CHANGED Viewed

@@ -7,7 +7,7 @@ import pandas as pd
 import torch
 from prophet.serialize import model_to_json, model_from_json
 from pytorch_forecasting import Baseline, TemporalFusionTransformer, TimeSeriesDataSet
-from pytorch_forecasting.models.temporal_fusion_transformer.tuning import optimize_hyperparameters
 import pickle
 # at beginning of the script
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")

 import torch
 from prophet.serialize import model_to_json, model_from_json
 from pytorch_forecasting import Baseline, TemporalFusionTransformer, TimeSeriesDataSet
+# from pytorch_forecasting.models.temporal_fusion_transformer.tuning import optimize_hyperparameters
 import pickle
 # at beginning of the script
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")