nlp: use new vector representation

[nit.git] / lib / nlp / nlp.nit

diff --git a/lib/nlp/nlp.nit b/lib/nlp/nlp.nit
index bc96938..0e4dba3 100644 (file)
--- 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