epoch fix

NdR_female_superheros.py

@@ -123,7 +123,7 @@ def run_female_superhero_train():
     optimizer = optim.SGD(linear_model.parameters(), lr=0.01, weight_decay=1e-4)
 
     # Training the Linear Model
-    num_epochs = 1#00
+    num_epochs = 100
     for epoch in range(num_epochs):
         linear_model.train()
         outputs = linear_model(X_train_tensor)

NdR_male_superheros.py

@@ -169,7 +169,7 @@ def run_male_superhero_train():
     optimizer = optim.Adam(neural_model.parameters(), lr=0.001, weight_decay=1e-4)
 
     # Training the Neural Network Model
-    num_epochs = 2#00
+    num_epochs = 200
     for epoch in range(num_epochs):
         neural_model.train()
         outputs = neural_model(X_train_tensor)