def METHOD_NAME():
    """Test thresholding of multiclass predictions."""
    y = np.random.rand(10, 5)
    y = y / np.sum(y, axis=1)[:, np.newaxis]
    y_thresh = threshold_predictions(y)
    assert y_thresh.shape == (10,)
    assert (y_thresh == np.argmax(y, axis=1)).all()