def predict(self,X): proba = self.predict_proba(X=X) y_pred = sp.argmax(proba,axis=1) return sp.array(y_pred)