Fixing the tmp_file write fail text-machine-lab#10

Updated the encoding of "clf" to enable tmp_file write.
HabibMrad · Jul 18, 2018 · 91e8b1e · 91e8b1e
1 parent 441e3d2
commit 91e8b1e
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()