From cdf9c6fdb2b9bc9722be5a933828dbd62a9a697f Mon Sep 17 00:00:00 2001 From: Harshit Kumar Date: Sun, 3 Mar 2019 15:33:01 +0530 Subject: [PATCH] Fix the number of neurons in last fully connected layer Fix number of neurons in last fully connected layer to 2 (equal to number of classes in diabetes dataset) --- 08_2_dataset_loade_logistic.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/08_2_dataset_loade_logistic.py b/08_2_dataset_loade_logistic.py index 43ba9f4..c027c96 100644 --- a/08_2_dataset_loade_logistic.py +++ b/08_2_dataset_loade_logistic.py @@ -41,7 +41,7 @@ def __init__(self): super(Model, self).__init__() self.l1 = torch.nn.Linear(8, 6) self.l2 = torch.nn.Linear(6, 4) - self.l3 = torch.nn.Linear(4, 1) + self.l3 = torch.nn.Linear(4, 2) self.sigmoid = torch.nn.Sigmoid()