Merge pull request #11 from dsouzadaniel/patch-1

Fixing the tmp_file write fail #10
text-machine-lab · Jul 18, 2018 · 90b3875 · 90b3875
2 parents 441e3d2 + 91e8b1e
commit 90b3875
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/code/machine_learning/crf.py b/code/machine_learning/crf.py
@@ -178,7 +178,8 @@ def predict(clf, X):
     # Dump the model into a temp file
     os_handle,tmp_file = tempfile.mkstemp(dir=tmp_dir, suffix="crf_temp")
     with open(tmp_file, 'wb') as f:
-        f.write(clf)
+        clf_byte = bytearray(clf, 'latin1')
+        f.write(clf_byte)
 
     # Create the Tagger object
     tagger = pycrfsuite.Tagger()