Update DataLoader.py

DataLoader.py

@@ -2,10 +2,7 @@
 
 import pandas as pd
 import numpy as np
-import requests
-import matplotlib.pyplot as plt
 import yfinance as yf
-from datetime import datetime
 from pygooglenews import GoogleNews
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
 from transformers import pipeline
@@ -29,12 +26,13 @@ class DataLoader:
 
 
     def get_stock_data(self):
-        data = yf.download(self.ticker, period = self.time_period_stock)
+        data = yf.download(self.ticker , period = self.time_period_stock)
         df = pd.DataFrame()
         df['Open'] = data['Open']
         df['Close'] = data['Close']
         df['High'] = data['High']
         df['Low'] = data['Low']
+        df['Volume'] = data['Volume']
 
         return df
 
@@ -108,12 +106,4 @@ class DataLoader:
             for column in news_columns:
                 if combined_data[column][i] == 0:
                     combined_data[column][i] = combined_data[column][i-1] * decay_rate
-        return combined_data
-
-
-
-
-
-
-
-
+        return combined_data