Shape of prediction

[LeoLaugier]

Hi,

Thank you for this implementation of Hyperband.

I noticed that in defs_regression, prediction "p" for keras_mlp and rf has shape (n,1) whereas "target" has shape (n,).

I wanted to define my own metric that involved substracting prediction by target at some point. For small arrays it is OK to substract a (n, ) array to a (n, 1) array but for n > 100,000 I got a memory error.

You might want to squeeze prediction p to troubleshoot this problem.

Thank you.