STCKMRKTFRCST - Linear Regression
STCKMRKTFRCST - Linear Regression
import numpy as np
import os
%matplotlib inline
df_Stock = Stock
df_Stock = df_Stock.rename(columns={'Close(t)':'Close'})
df_Stock.head()
df_Stock.tail(5)
df_Stock.shape
df_Stock.columns
df_Stock['Close'].plot(figsize=(10, 7))
plt.ylabel('Price', fontsize=14)
plt.xlabel('Time', fontsize=14)
plt.show()
df_Stock = df_Stock.drop(columns='Date_col')
def create_train_test_set(df_Stock):
target = df_Stock['Close_forcast']
data_len = df_Stock.shape[0]
# Splitting features and target into train, validation and test samples
lr = LinearRegression()
lr.fit(X_train, Y_train)
"""
"""
Y_train_pred = lr.predict(X_train)
Y_val_pred = lr.predict(X_val)
Y_test_pred = lr.predict(X_test)
print(' ')
print(' ')
df_pred['Predicted'] = Y_val_pred
df_pred = df_pred.reset_index()
df_pred.loc[:, 'Date'] = pd.to_datetime(df_pred['Date'],format='%Y-%m-%d')
df_pred
df_pred[['Actual', 'Predicted']].plot()