logs implemented, something off (probably with probability of elements not in training data)