Create train_ag_news.py

train_ag_news.py

@@ -0,0 +1,22 @@
+"""
+Quick training script for AG News dataset
+Usage: python train_ag_news.py
+"""
+from train import train_model
+
+if __name__ == '__main__':
+    train_model(
+        dataset_name='ag_news',
+        embedding_dim=128,        # Balanced size
+        hidden_dim=192,           # Same as emotion for consistency
+        num_layers=1,             # Single layer for simple RNN
+        num_hidden_nodes=96,      # Same as emotion for consistency
+        dropout=0.25,             # Moderate dropout
+        batch_size=64,            # Good batch size
+        max_length=50,            # AG News works well with 50 tokens
+        learning_rate=0.001,      # Slightly lower for stability
+        num_epochs=30,            # Reasonable epochs
+        patience=7,
+        vocab_min_freq=5          # Higher threshold for formal news text
+    )
+