Got rid of reshaping in one_vs_all
Flatten labels vector.
Fixed dimensions of the array returned by predict_one_vs_all.