Skip to content

Commit a7a4e78

Browse files
authored
Update multilayer_perceptron_classifier_from_scratch.py
1 parent 53a35b1 commit a7a4e78

1 file changed

Lines changed: 9 additions & 8 deletions

File tree

machine_learning/multilayer_perceptron_classifier_from_scratch.py

Lines changed: 9 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -142,15 +142,15 @@ def __init__(
142142
self.gamma = gamma # learning_rate decay hyperparameter gamma
143143
self.epoch = epoch
144144
self.hidden_dim = hidden_dim
145-
146-
self.train_loss = []
147-
self.train_accuracy = []
148-
self.test_loss = []
149-
self.test_accuracy = []
145+
146+
self.train_loss: list[float] = []
147+
self.train_accuracy: list[float] = []
148+
self.test_loss: list[float] = []
149+
self.test_accuracy: list[float] = []
150150

151151
self.dataloader = dataloader
152-
self.inter_variable = {}
153-
self.weights1_list = []
152+
self.inter_variable: dict[str, np.ndarray] = {}
153+
self.weights1_list: list[np.ndarray] = []
154154

155155
def get_inout_dim(self) -> tuple[int, int]:
156156
"""
@@ -469,7 +469,8 @@ def train(self) -> None:
469469

470470
w1, w2 = self.initialize()
471471

472-
test_accuracy_list, test_loss_list = [], []
472+
test_accuracy_list: list[float] = []
473+
test_loss_list: list[float] = []
473474

474475
batch_size = 1
475476

0 commit comments

Comments
 (0)