def predict(self, X): check_is_fitted(self, ['X_', 'y_']) X = check_array(X) X = DynamicBayesianClassifier._first_col(X) return self._predict(X)