Tentando aplicar a rede neural do zero

app.py

@@ -15,10 +15,7 @@ def coeff(coeffs, indeps):
     for i,l in enumerate(layers):
         res = res@l + consts[i]
         if i!=n-1: res = F.relu(res)
-    if torch.sigmoid(res) > 0.01:
-        return torch.tensor(1)
-    else:
-        return torch.tensor(0)
+    return torch.sigmoid(res)
 
 """
 def predict(Pregnancies, Glucose, BloodPressure, SkinThickness, Insulin, BMI, DiabetesPedigreeFunction, Age):