Update DataLoader.py

DataLoader.py

@@ -1,7 +1,6 @@
 #Class to fetch news and stock data from the web for a specific ticker and combine them into a dataframe.
 
 import pandas as pd
-import numpy as np
 import yfinance as yf
 from pygooglenews import GoogleNews
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
@@ -104,6 +103,6 @@ class DataLoader:
         #We want to start from the oldest date and work our way to the newest date
         for i in range(1, len(combined_data)):
             for column in news_columns:
-                if combined_data[column][i] == 0:
+                if combined_data[column][i] == 0 and combined_data[column][i-1] != 0:
                     combined_data[column][i] = combined_data[column][i-1] * decay_rate
         return combined_data