fix: tfidf bug

cophi-wue · Jan 24, 2019 · 446cf72 · 446cf72
1 parent e14084d
commit 446cf72
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/cophi/model.py b/src/cophi/model.py
@@ -504,7 +504,7 @@ def tfidf(self):
             tf_{t,d} \times log(\frac{N}{df_t})
         """
         tf = self.rel
-        idf = self.stats["documents"] / self.dtm.astype(bool).sum(axis=0)
+        idf = self.stats["documents"] / self.dtm.fillna(0).astype(bool).sum(axis=0)
         return tf * np.log(idf)
 
     @property