Bagging Regression Mean

Average the predictions of several regressors (for example trees trained on bootstrap samples). predictions has shape (n_models, n_samples).

Implement solve(predictions).

Returns. Return a NumPy array of length n_samples holding the mean prediction of the models for each sample. For a 1-D input the result is a single float.

Examples

Example 1

Input:

solve([[1.0, 2.0], [3.0, 4.0], [5.0, 9.0]])

Output:

[3.0, 5.0]

Example 2

Input:

solve([1.0, 2.0, 3.0])

Output:

2.0