for (int i = 0; i < numClasses; i++) {
double probability = getProbabilityForClass(document, i);
distribution.set(i, probability);
}
double maxProbability = distribution.max();
double probabilitySum = 0.0d;
// we normalize it back
for (int i = 0; i < numClasses; i++) {
double probability = distribution.get(i);
double normalizedProbability = FastMath.exp(probability - maxProbability