kennethreitz/coinbin.org
1
0
Fork 0
mirror of https://github.com/kennethreitz/coinbin.org.git synced 2026-06-05 23:10:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

just use plain db for predictions

Browse Source 
Signed-off-by: Kenneth Reitz <me@kennethreitz.org>
This commit is contained in:
Kenneth Reitz
2017-08-27 23:59:48 -04:00
parent cc3957bd24
commit aafff78672
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
predictions.py
+4 -4
View File
@@ -18,8 +18,8 @@ def get_predictions(coin):
q = "SELECT date as ds, value as y from api_coin WHERE name=:coin"
pro_db = records.Database(os.environ['HEROKU_POSTGRESQL_TEAL_URL'])
rows = pro_db.query(q, coin=c.name)
db = records.Database()
rows = db.query(q, coin=c.name)
df = rows.export('df')
@@ -31,7 +31,7 @@ def get_predictions(coin):
model = Prophet(weekly_seasonality=True, yearly_seasonality=True)
model.fit(df)
future_data = model.make_future_dataframe(periods=30, freq='d')
future_data = model.make_future_dataframe(periods=365, freq='d')
forecast_data = model.predict(future_data)
print(forecast_data[['ds', 'yhat', 'yhat_lower', 'yhat_upper']].tail())
@@ -46,7 +46,7 @@ def get_predictions(coin):
# print(forecast_data_orig)
d = forecast_data_orig['yhat'].to_dict()
predictions = []
for i, k in enumerate(list(d.keys())[-30:]):
for i, k in enumerate(list(d.keys())[-365:]):
w = maya.when(f'{i+1} days from now')
predictions.append({
'when': w.slang_time(),