lkrsnik/cjvt-srl-tagging
1
0
Fork 0
forked from kristjan/cjvt-srl-tagging
Code Issues Pull Requests Projects Releases Wiki Activity

fixed paths

Browse Source
This commit is contained in:
voje
2019-02-27 17:32:19 +01:00
parent 5c9cf59723
commit b4c7ac5427
13 changed files with 40 additions and 41 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tools/fillpred_model/step1.py
View File

@@ -51,4 +51,4 @@ if __name__ == "__main__":
print(i, df.shape)
print(ndf.head())
ndf.to_pickle(OUTFILE)
ndf.to_pickle(Path(OUTFILE))

4
tools/fillpred_model/step2.py
View File

@@ -27,4 +27,6 @@ if __name__ == "__main__":
clf_full = DecisionTreeClassifier()
clf_full.fit(X, y)
pickle.dump(clf_full, open(OUTFILE, "wb"))
with open(OUTFILE, "wb") as fp:
pickle.dump(clf_full, fp)