Skip to content

Commit ed2a07c

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

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

machine_learning/rbf_network.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -90,7 +90,7 @@ def _compute_rbf_activations(self, X: np.ndarray) -> np.ndarray: # noqa: N803
9090

9191
return activations
9292

93-
def train(self, X: np.ndarray, y: np.ndarray) -> None: # noqa: N803
93+
def train(self, X: np.ndarray, y: np.ndarray) -> None: # noqa: N803
9494
"""
9595
Train the RBFNN using KMeans clustering and least-squares fitting.
9696
@@ -132,7 +132,7 @@ def train(self, X: np.ndarray, y: np.ndarray) -> None: # noqa: N803
132132
# weights = (A^T A)^-1 A^T y, where A is the activation matrix
133133
self.weights = np.linalg.lstsq(activations, y, rcond=None)[0]
134134

135-
def predict(self, X: np.ndarray) -> np.ndarray: # noqa: N803
135+
def predict(self, X: np.ndarray) -> np.ndarray: # noqa: N803
136136
"""
137137
Make predictions using trained RBFNN.
138138

0 commit comments

Comments
 (0)