Update model names and rounds in Study_Models and Test_Model scripts; add future version ideas in todo.md

Author: DefinetlyNotAI

todo

@@ -0,0 +1,85 @@
+For future version 4n2
+
+```
+[6] Pred=0.000 → Private SSH key
+[18] Pred=0.000 → Private RSA key
+[30] Pred=0.003 → Credit card expiration
+[32] Pred=0.000 → Encrypted password hash
+[38] Pred=0.000 → Date of birth
+[39] Pred=0.000 → Secret question answer
+[44] Pred=0.000 → Private notes
+[47] Pred=0.004 → Encrypted token
+```
+
+Do include these in future version 4n2
+
+Example code for generating sensitive data using the Faker library in Python:
+```python
+import random
+import base64
+from faker import Faker
+
+class DataGenerator:
+    def __init__(self, cfg):
+        self.cfg = cfg
+        self.faker = Faker()
+
+    def sensitive_text(self):
+        field = random.choice(self.cfg.SENSITIVE_FIELDS)
+
+        if field == "ssn":
+            return f"SSN: {self.faker.ssn()}"
+
+        elif field == "credit_card":
+            number = self.faker.credit_card_number()
+            exp = self.faker.credit_card_expire()
+            cvv = self.faker.random_int(min=100, max=999)
+            return f"Credit Card: {number} Exp: {exp} CVV: {cvv}"
+
+        elif field == "email":
+            return f"Email: {self.faker.email()}"
+
+        elif field == "phone_number":
+            return f"Phone: {self.faker.phone_number()}"
+
+        elif field == "address":
+            return f"Address: {self.faker.address().replace(chr(10), ', ')}"
+
+        elif field == "name":
+            return f"Name: {self.faker.name()}"
+
+        elif field == "password":
+            return f"Password: {self.faker.password(length=12, special_chars=True)}"
+
+        elif field == "private_key":
+            # Simulate a realistic RSA private key
+            lines = [base64.b64encode(self.faker.binary(length=32)).decode('ascii') for _ in range(10)]
+            key_body = "\n".join(lines)
+            return f"Private key: -----BEGIN RSA PRIVATE KEY-----\n{key_body}\n-----END RSA PRIVATE KEY-----"
+
+        elif field == "ssh_key":
+            # Simulate realistic OpenSSH private key
+            lines = [base64.b64encode(self.faker.binary(length=32)).decode('ascii') for _ in range(8)]
+            key_body = "\n".join(lines)
+            return f"Private SSH key: -----BEGIN OPENSSH PRIVATE KEY-----\n{key_body}\n-----END OPENSSH PRIVATE KEY-----"
+
+        elif field == "api_key":
+            return f"API Key: {self.faker.sha256()[:32]}"
+
+        elif field == "pin":
+            return f"PIN: {self.faker.random_int(min=1000, max=9999)}"
+
+        elif field == "dob":
+            return f"DOB: {self.faker.date_of_birth(minimum_age=18, maximum_age=90)}"
+
+        elif field == "security_answer":
+            return f"Security Answer: {self.faker.word()}"
+
+        elif field == "jwt_token":
+            header = base64.urlsafe_b64encode(b'{"alg":"HS256","typ":"JWT"}').decode('utf-8').rstrip("=")
+            payload = base64.urlsafe_b64encode(self.faker.json().encode()).decode('utf-8').rstrip("=")
+            signature = base64.urlsafe_b64encode(self.faker.binary(length=16)).decode('utf-8').rstrip("=")
+            return f"JWT: {header}.{payload}.{signature}"
+
+        return "Sensitive info: [REDACTED]"
+```

todo.md

@@ -15,8 +15,8 @@
 from data import test_texts, test_labels
 
 # ----------------- Setup -----------------
-NAME = "Model_Sense.4n1"
-ROUND = 5
+NAME = "Model_SenseMacro.4n1"
+ROUND = 7
 OUTPUT_DIR = f"../{NAME}_Data_Visualization"
 os.makedirs(OUTPUT_DIR, exist_ok=True)
 EMBEDDINGS_DIR = f"../cache/{NAME}/round_{ROUND}/embeddings"  # your embeddings directory

tools/Study_Models.py

@@ -9,8 +9,8 @@
 from vulnscan import SimpleNN
 
 # ---------------- INIT ----------------
-NAME = "Model_Sense.4n1"
-ROUND = 5
+NAME = "Model_SenseMacro.4n1"
+ROUND = 7
 
 # ---------------- LOAD MODEL + EMBEDDER ----------------
 device = "cuda" if torch.cuda.is_available() else "cpu"