Merge pull request L1aoXingyu#24 from jaeyung1001/patch-2

Author: L1aoXingyu

02-Logistic Regression/Logistic_Regression.py

@@ -29,16 +29,17 @@
 test_loader = DataLoader(test_dataset, batch_size=batch_size, shuffle=False)
 
 # 定义 Logistic Regression 模型
-class logsticRegression(nn.Module):
+class Logistic_Regression(nn.Module):
     def __init__(self, in_dim, n_class):
-        super(logsticRegression, self).__init__()
-        self.logstic = nn.Linear(in_dim, n_class)
+        super(Logistic_Regression, self).__init__()
+        self.logistic = nn.Linear(in_dim, n_class)
 
     def forward(self, x):
-        out = self.logstic(x)
+        out = self.logistic(x)
         return out
 
-model = logsticRegression(28 * 28, 10)  # 图片大小是28x28
+
+model = Logistic_Regression(28 * 28, 10)  # 图片大小是28x28
 use_gpu = torch.cuda.is_available()  # 判断是否有GPU加速
 if use_gpu:
     model = model.cuda()