Source code for autoflow.ensemble.mean.regressor

from sklearn.base import RegressorMixin, BaseEstimator
import numpy as np


class MeanRegressor(BaseEstimator, RegressorMixin):
    def __init__(self,models):
        self.models = models

    def predict(self,X):
        preds = [model.predict(X) for model in self.models]
        pred_arr = np.array(preds)
        pred = np.average(pred_arr, axis=0)
        return pred