I have a function that calculates the probability of belonging to category k ~ {1, 2, ..., K} based on