From 6f4db9de1aa12f4c083a15f18755a41adda7b473 Mon Sep 17 00:00:00 2001 From: Piotr Czapla Date: Sat, 18 May 2019 22:50:55 +0200 Subject: [PATCH] Fix ensemble output (by removing index column) --- ulmfit/__main__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ulmfit/__main__.py b/ulmfit/__main__.py index c0d27e4..b6c4b66 100644 --- a/ulmfit/__main__.py +++ b/ulmfit/__main__.py @@ -197,7 +197,7 @@ class ULMFiT: if out_template: 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) + test.to_csv(out_file, header=None, index=False) print({"File saved to": out_file}) def eval(self, glob="data/mldoc/*-1/models/sp30k/lstm_nl4.m", dataset_template='${ds_name}', name=None,