I figured `get_samples` is the fully general function and everything else (including `classify`) are just convenience methods on top of it.

I could split the functionality of `classify` into 2 functions: one that returns the likelihood vectors and another that calls the first function and computes the argmax and converts to one-hot.