Fix

correlation_test_all.py

@@ -52,7 +52,8 @@ 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"]))
+
+        prev_price_dict[i["ticker"]].append(float(i["open"]))
 
     for i in av_gain_turn_dict:
         if count_pos_dict[i] > 0 and count_turn_dict[i] > 0 and count_dict[i] > 0:

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"
@@ -177,5 +177,4 @@ def train():
                 correct = 0.0
 
 if __name__ == "__main__":
-    make_dataset()
-    train()
\ No newline at end of file
+    make_dataset()
\ No newline at end of file