Fix

correlation_test_all.py

@@ -52,6 +52,7 @@ def main():
 
             if len(prev_price_dict[i["ticker"]]) > LEN_PREV:
                 prev_price_dict[i["ticker"]].pop(0)
+
         prev_price_dict[i["ticker"]].append(float(i["open"]))
 
     for i in av_gain_turn_dict:

model.py

 import csv
 
-import torch.nn as nn
-import torch
+# import torch.nn as nn
+# import torch
 import math
 import os
 import io
 import numpy as np
 import pandas as pd
 import random
-from torch.utils.data import Dataset, DataLoader, random_split
-import torch.nn.functional as F
-from torchvision import transforms
+# from torch.utils.data import Dataset, DataLoader, random_split
+# import torch.nn.functional as F
+# from torchvision import transforms
 
 FILE_NAME = "historical_stock_prices.csv"
 OUTPUT_FILE_NAME = "30-day-stocks.csv"
@@ -178,4 +178,3 @@ def train():
 
 if __name__ == "__main__":
     make_dataset()
\ No newline at end of file
-    train()
\ No newline at end of file