Update app.py
Browse files
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]
|