ensure_predictions_shape#

ensure_predictions_shape(pred: ndarray, data: Sequence) ndarray[source]#

Ensure the predictions are in the right shape and if so return them. else raise error.