Also, let's make it take a `MutableVec` and fill in the probability distribution, rather than creating a fresh vector. That will make this more efficient - we can reuse the same vector for every item we want to predict for. Excess memory allocation is one of the biggest performance killers in Java.