def predict(self, dataset: Any):
    'Performs prediction'
    self.test = deepcopy(dataset)
    if ('orbit' in self.name):
        prediction = self.model.predict(self.test)
    elif ('nprophet' in self.name):
        future = self.model.make_future_dataframe(self.train, periods=len(self.test))
        prediction = self.model.predict(future).rename(columns={'yhat1': self.pred_col})
    prediction = prediction[[self.date_col, self.pred_col]]
    self.prediction = prediction
    return self.prediction
Performs prediction
def predict(self, dataset: Any):
    
    self.test = deepcopy(dataset)
    if ('orbit' in self.name):
        prediction = self.model.predict(self.test)
    elif ('nprophet' in self.name):
        future = self.model.make_future_dataframe(self.train, periods=len(self.test))
        prediction = self.model.predict(future).rename(columns={'yhat1': self.pred_col})
    prediction = prediction[[self.date_col, self.pred_col]]
    self.prediction = prediction
    return self.prediction