Tonic commited on
Commit
d3a5a47
·
1 Parent(s): fe5cebb

adds chronos dependencies correctly

Browse files
Files changed (2) hide show
  1. app.py +2 -1
  2. requirements.txt +2 -2
app.py CHANGED
@@ -129,7 +129,8 @@ def make_prediction(symbol: str, timeframe: str = "1d", prediction_days: int = 5
129
  with torch.inference_mode():
130
  prediction = pipe.predict(
131
  context=context,
132
- prediction_length=prediction_days
 
133
  ).detach().cpu().numpy()
134
 
135
  # Reshape prediction to get mean and std
 
129
  with torch.inference_mode():
130
  prediction = pipe.predict(
131
  context=context,
132
+ prediction_length=prediction_days,
133
+ num_samples=100
134
  ).detach().cpu().numpy()
135
 
136
  # Reshape prediction to get mean and std
requirements.txt CHANGED
@@ -1,7 +1,7 @@
1
  # first pip install numpy scipy scikit-learn all seperately
2
 
3
 
4
- pandas_datareader
5
  # numpy
6
  #--find-links https://download.pytorch.org/whl/torch_stable.html
7
  #torch==1.11.0+cu113
@@ -103,7 +103,7 @@ fastapi
103
  gunicorn
104
  uvicorn
105
  # git+https://github.com/amazon-science/chronos-forecasting.git
106
- chronos-forecasting
107
  scikit-learn>=1.0.0
108
 
109
  python-binance
 
1
  # first pip install numpy scipy scikit-learn all seperately
2
 
3
 
4
+ pandas_datareader>=0.10.0
5
  # numpy
6
  #--find-links https://download.pytorch.org/whl/torch_stable.html
7
  #torch==1.11.0+cu113
 
103
  gunicorn
104
  uvicorn
105
  # git+https://github.com/amazon-science/chronos-forecasting.git
106
+ chronos-forecasting>=1.5.2
107
  scikit-learn>=1.0.0
108
 
109
  python-binance