Spaces:
Paused
Paused
Tonic commited on
Commit ·
d3a5a47
1
Parent(s): fe5cebb
adds chronos dependencies correctly
Browse files- app.py +2 -1
- 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
|