Bug in DenseBoWBP

INGEOTEC · Jan 30, 2024 · 5341f63 · 5341f63
1 parent bc26f1d
commit 5341f63
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/EvoMSA/back_prop.py b/EvoMSA/back_prop.py
@@ -216,6 +216,7 @@ def initial_parameters(self, X, y):
         dense_w = self.weights.T
         dense_bias = self.bias
         _ = X[tr] @ dense_w + dense_bias
+        _ = _ / np.linalg.norm(_, axis=1, keepdims=True)
         m = self.estimator_class(**self.estimator_kwargs).fit(_, y[tr])
         W = jnp.array(m.coef_.T)
         W0 = jnp.array(m.intercept_)