Upload 3 files

edusights_20240702_state_dict_CON_31.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4ca4ba27d9b1ff10216d89a39de13bb17afdf85d35f9d07944ca988ec202f9f7
+size 28650

es_class_nn.py

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