jcmachicao commited on
Commit
70a385d
1 Parent(s): 229898b

Upload 3 files

Browse files
df_carga.xlsx ADDED
Binary file (11.2 kB). View file
 
edusights_20240702_state_dict_CON_31.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4ca4ba27d9b1ff10216d89a39de13bb17afdf85d35f9d07944ca988ec202f9f7
3
+ size 28650
es_class_nn.py ADDED
@@ -0,0 +1,24 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import torch
2
+ import torch.nn as nn
3
+ import pandas as pd
4
+ import numpy as np
5
+
6
+ DROPOUTX = 0.05
7
+
8
+ class SimplePlusNN2(nn.Module):
9
+ def __init__(self, input_size, num_classes):
10
+ super(SimplePlusNN2, self).__init__()
11
+ self.fc1 = nn.Linear(input_size, 64)
12
+ self.relu = nn.ReLU()
13
+ self.fc2 = nn.Linear(64, 8)
14
+ self.relu = nn.ReLU()
15
+ self.fc3 = nn.Linear(8, 1)
16
+ self.dropout = nn.Dropout(DROPOUTX)
17
+ self.sigmoid = nn.Sigmoid()
18
+
19
+ def forward(self, x):
20
+ x = self.relu(self.fc1(x))
21
+ x = self.relu(self.fc2(x))
22
+ x = self.dropout(x)
23
+ x = self.sigmoid(self.fc3(x))
24
+ return x