Refactor label handling in Study_Models and update visualization process in todo.md

Fixed bugs for Study_Models.py

Author: DefinetlyNotAI

todo.md

@@ -1,6 +1,8 @@
 For future version 4n2
 
 ```
+[2] Pred=0.438 → Credit Card Number
+[4] Pred=0.001 → Email Address
 [6] Pred=0.000 → Private SSH key
 [18] Pred=0.000 → Private RSA key
 [30] Pred=0.003 → Credit card expiration

tools/Study_Models.py

@@ -169,9 +169,9 @@ def visualize_tsne(model_, dataloader_, filename="Visualize_tSNE.png", use_penul
             else:
                 feat = model_(X)
             all_features.append(feat.cpu().numpy())
-            all_labels.append(y.cpu().numpy())
+            all_labels.append(y.cpu().numpy().ravel())
     all_features = np.vstack(all_features)
-    all_labels = np.hstack(all_labels)
+    all_labels = np.concatenate(all_labels, axis=0)
     n_samples, n_features = all_features.shape
     n_components = min(2, n_samples, n_features)  # adapt automatically
     tsne = TSNE(n_components=n_components, random_state=42, perplexity=max(1, min(30, n_samples - 1)))
@@ -302,8 +302,6 @@ def save_model_summary(model_, filename="Model_Summary.txt"):
 visualize_tsne_custom(model, embedder, test_texts, test_labels)
 print("Running visualize_feature_importance...")
 visualize_feature_importance(input_dim)
-print("Running plot_loss_landscape_3d...")
-plot_loss_landscape_3d(model, dataloader, criterion)
 print("Saving model state dict...")
 save_model_state_dict(model)
 print("Generating model visualization...")
@@ -312,5 +310,7 @@ def save_model_summary(model_, filename="Model_Summary.txt"):
 save_graph(model)
 print("Saving model summary...")
 save_model_summary(model)
-
+print("Running plot_loss_landscape_3d...")
+model_cpu = model.to("cpu")
+plot_loss_landscape_3d(model_cpu, dataloader, criterion)
 print("All visualizations completed. Files saved in 'data/' directory.")