Skip to content

Commit 8988727

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent e0ae20f commit 8988727

1 file changed

Lines changed: 10 additions & 8 deletions

File tree

machine_learning/multilayer_perceptron_classifier_from_scratch.py

Lines changed: 10 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,9 @@
11
import numpy as np
22
from numpy.random import default_rng
3+
34
rng = default_rng(42)
5+
6+
47
class Dataloader:
58
"""
69
DataLoader class for handling dataset, including data shuffling,
@@ -136,12 +139,12 @@ class MLP:
136139
"""
137140

138141
def __init__(
139-
self,
140-
dataloader: Dataloader,
141-
epoch: int,
142-
learning_rate: float,
143-
gamma: float = 1.0,
144-
hidden_dim: int = 2,
142+
self,
143+
dataloader: Dataloader,
144+
epoch: int,
145+
learning_rate: float,
146+
gamma: float = 1.0,
147+
hidden_dim: int = 2,
145148
) -> None:
146149
self.learning_rate = learning_rate
147150
self.gamma = gamma # learning_rate decay hyperparameter gamma
@@ -192,8 +195,7 @@ def initialize(self) -> tuple[np.ndarray, np.ndarray]:
192195
"""
193196

194197
in_dim, out_dim = self.dataloader.get_inout_dim()
195-
w1 = (rng.standard_normal((in_dim + 1, self.hidden_dim)) *
196-
np.sqrt(2.0 / in_dim))
198+
w1 = rng.standard_normal((in_dim + 1, self.hidden_dim)) * np.sqrt(2.0 / in_dim)
197199
w2 = rng.standard_normal((self.hidden_dim, out_dim)) * np.sqrt(
198200
2.0 / self.hidden_dim
199201
)

0 commit comments

Comments
 (0)