X-Git-Url: http://nitlanguage.org diff --git a/lib/nlp/nlp.nit b/lib/nlp/nlp.nit index bc96938..0e4dba3 100644 --- a/lib/nlp/nlp.nit +++ b/lib/nlp/nlp.nit @@ -23,17 +23,17 @@ import vsm redef class NLPDocument # `NLPVector` representing `self`. - var vector: Vector[String] is lazy do - var vector = new Vector[String] + var vector: Vector is lazy do + var vector = new Vector for sentence in sentences do for token in sentence.tokens do if not keep_pos_token(token) then continue var lemma = token.lemma if lemma_black_list.has(lemma) then continue if not vector.has_key(lemma) then - vector[lemma] = 1 + vector[lemma] = 1.0 else - vector[lemma] += 1 + vector[lemma] += 1.0 end end end