Fixes tests & adds the "count" argument to dlm.ldl.LDL.gen_cmat. (#4)

* Fixes tests

* Adds the 'count' argument to dlm.ldl.LDL.gen_cmat.

Author: msaito8623

discriminative_lexicon_model/ldl.py

@@ -16,10 +16,10 @@ def __init__ (self, words=None, embed_or_df=None, cmat=False, smat=False,
                     events=events, freqs=freqs)
         return None
 
-    def gen_cmat (self, words=None, gram=3, noise=False, freqs=None):
+    def gen_cmat (self, words=None, gram=3, count=True, noise=False, freqs=None):
         if not (words is None):
             self.words = words
-        self.cmat = lm.gen_cmat(self.words, gram=gram, noise=noise, freqs=freqs)
+        self.cmat = lm.gen_cmat(self.words, gram=gram, count=count, noise=noise, freqs=freqs)
         return None
 
     def gen_smat (self, embed_or_df, words=None, noise=False, freqs=None):

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "discriminative_lexicon_model"
-version = "2.0.1"
+version = "2.0.2"
 description = "Python-implementation of Discriminative Lexicon Model / Linear Discriminative Learning"
 
 license = "MIT"