def predict(self, X):
    if (len(X.shape) == 1):
        X = X.reshape(1, (- 1))
    return self.model.predict(X)
def predict(self, X):
    if (len(X.shape) == 1):
        X = X.reshape(1, (- 1))
    return self.model.predict(X)