def predict(self, X): # scikit-learn checks X = check_array(X) return np.array([self._predict(x) for x in X])