Update project_1.py

lrm22005 · Oct 24, 2023 · 3fb1e8c · 3fb1e8c
1 parent 473f0bf
commit 3fb1e8c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/project_1.py b/project_1.py
@@ -72,13 +72,13 @@ def load_data(data_path, labels_path, dataset_size=10, train=True, standardize=T
             else:
                 continue  # Skip other file formats
 
-            X_data_original.append(time_freq_tensor.clone())  # Store a copy of the original data
+            # X_data_original.append(time_freq_tensor.clone())  # Store a copy of the original data
             X_data.append(time_freq_tensor)
 
             segment_names.append(seg)  # Store segment names
 
     X_data = torch.cat(X_data, 0)
-    X_data_original = torch.cat(X_data_original, 0)
+    # X_data_original = torch.cat(X_data_original, 0)
 
     if standardize:
         X_data = standard_scaling(X_data)  # Standardize the data