Результаты бесконечности

Question

Я пытаюсь вычислить tf и idf коллекции документов. Моя проблема в том, что я получаю бесконечность, и я не хочу бесконечности, но мне нужны двойные результаты.

Вот мой код:

String key = entry.getKey().toString(); 
Integer value = entry.getValue().intValue(); 
System.out.println("key " + key + " value " + value); 

BigDecimal tf=new BigDecimal(entry.getValue()/collection.size()); 
BigDecimal idf=new BigDecimal(Math.log(counter/entry.getValue())); 
BigDecimal tfidf =new BigDecimal((tf.doubleValue())*(idf.doubleValue())); 
System.out.println("TF*IDF: " + key + ": " + tfidf.doubleValue()); 

вход: Карта, где строка является ключевым словом и Integer, сколько раз появляется слово в документе.

выход: ключевое слово + TF * IDF: номер

Любая идея, как решить эту проблему?

Answer 1

Вы можете иметь проблемы в следующей строке:

BigDecimal tf = new BigDecimal(entry.getValue()/collection.size()); 

Вы perfroming целое подразделение. Это может быть усечено до нуля. Попробуйте изменить это

BigDecimal tf=new BigDecimal((double) entry.getValue()/collection.size()); 

Та же проблема здесь:

new BigDecimal(Math.log(counter/entry.getValue())); 

изменить его

new BigDecimal(Math.log((double) counter/entry.getValue()));