Update es_class_nn.py

es_class_nn.py

@@ -3,16 +3,16 @@ import torch.nn as nn
 import pandas as pd
 import numpy as np
 
-DROPOUTX = 0.05
+DROPOUTX = 0.10
 
 class SimplePlusNN2(nn.Module):
     def __init__(self, input_size, num_classes):
         super(SimplePlusNN2, self).__init__()
-        self.fc1 = nn.Linear(input_size, 64)
+        self.fc1 = nn.Linear(input_size, 128)
         self.relu = nn.ReLU()
-        self.fc2 = nn.Linear(64, 8)
+        self.fc2 = nn.Linear(128, 32)
         self.relu = nn.ReLU()
-        self.fc3 = nn.Linear(8, 1)
+        self.fc3 = nn.Linear(32, 1)
         self.dropout = nn.Dropout(DROPOUTX)
         self.sigmoid = nn.Sigmoid()