Method predict

bench/embedding_experiment.py:121–123

(self, X: np.ndarray)

                self.b1 -= self.lr * db1

    def predict(self, X: np.ndarray) -> np.ndarray:
        _, _, _, probs = self._forward(X)
        return probs.argmax(axis=1)


def run_experiment():