706a4e840607d50dc94249150366a23ba720a900,mushroom/approximators/ensemble.py,Ensemble,predict,#Ensemble#Any#,41

Before Change

            The predictions of the model.

        
        y = list()
        for m in self.models:
            y.append(m.predict(x))
        y = np.mean(y, axis=0)

After Change

        
        y_0 = self.models[0].predict(x)
        y = np.zeros((self.n_models,) + y_0.shape)
        y[0] = y_0
        for i, m in enumerate(self.models[1:]):
            y[i + 1] = m.predict(x)
        y = np.mean(y, axis=0)

Instances