wassname/multifit
1
0
Fork 0
mirror of https://github.com/wassname/multifit.git synced 2026-09-09 11:27:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add ability to create folders when saving ensemble output

Browse Source
This commit is contained in:
Piotr Czapla
2019-05-18 21:59:12 +02:00
parent f38f8c0670
commit 06c32ae2dc
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
ulmfit/__main__.py
+2 -1
View File
@@ -195,7 +195,8 @@ class ULMFiT:
print({"Key": key, "Test Accuracy": (test[0] == ensemble).mean(), "on": gold_labels[key], 'files_count':len(files)})
test[0] = ensemble
if out_template:
out_file = Template(out_template).substitute(key=key)
out_file = Path(Template(out_template).substitute(key=key))
out_file.parent.mkdir(exist_ok=True, parents=True)
test.to_csv(out_file, header=None)
print({"File saved to": out_file})