dibend commited on
Commit
45cf157
1 Parent(s): ed74d2e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -21,6 +21,9 @@ def train_predict_wrapper(ticker, start_date, end_date, prediction_days):
21
  data = yf.download(ticker, start=start_date, end=end_date)
22
  data = data["Close"]
23
 
 
 
 
24
  # Train linear regression model
25
  X = data.index.values[:-prediction_days].reshape(-1, 1)
26
  y = data.values[:-prediction_days]
 
21
  data = yf.download(ticker, start=start_date, end=end_date)
22
  data = data["Close"]
23
 
24
+ # Convert dates to Unix timestamps
25
+ data.index = data.index.to_timestamp()
26
+
27
  # Train linear regression model
28
  X = data.index.values[:-prediction_days].reshape(-1, 1)
29
  y = data.values[:-prediction_days]