def predict(self, *args, **kwargs):
    if (not self.fitted):
        raise NotFittedError
    return self.predict_return
def predict(self, *args, **kwargs):
    if (not self.fitted):
        raise NotFittedError
    return self.predict_return