Change the `predict` function to `predict_num` and write a general wrapper `predict` method that switches between these.
Change the
predictfunction topredict_numand write a general wrapperpredictmethod that switches between these.