def METHOD_NAME(self, X, **kwargs):
    X = dt.Frame(X)
    X = self.basic_impute(X)
    X = X.to_numpy()
    model, _, _, _ = self.get_model_properties()
    if self.num_classes == 1:
        preds = model.METHOD_NAME(X)
    else:
        preds = model.predict_proba(X)
    return preds