@Override
    public Classification<T, K> classify(Collection<T> features) {
        SortedSet<Classification<T, K>> probabilites =
                this.categoryProbabilities(features);
        
//        System.out.println(probabilites);

        if (probabilites.size() > 0) {
            return probabilites.last();
        }
        return null;
    }