ensure_predictions_shape#

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

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

previous

ensure_dataframe_type

next

ensure_predictions_proba