Merge pull request #342 from ales-erjavec/fixes/bow-binary

[FIX] bagofwords: Use vectorized 'BINARY' local weighting
biolab · May 4, 2018 · ec1745a · ec1745a
2 parents abcec40 + 1aa8ff2
commit ec1745a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/orangecontrib/text/vectorization/bagofwords.py b/orangecontrib/text/vectorization/bagofwords.py
@@ -46,7 +46,7 @@ class BowVectorizer(BaseVectorizer):
 
     wlocals = OrderedDict((
         (COUNT, lambda tf: tf),
-        (BINARY, lambda tf: int(tf > 0)),
+        (BINARY, lambda tf: np.greater(tf, 0, dtype=np.int)),
         (SUBLINEAR, lambda tf: 1 + np.log(tf)),
     ))