From f38f8c0670846744e80d76b386fd0478cd7642ef Mon Sep 17 00:00:00 2001 From: Piotr Czapla Date: Sat, 18 May 2019 21:48:32 +0200 Subject: [PATCH] fix issue in the databunch name generation --- ulmfit/train_clas.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ulmfit/train_clas.py b/ulmfit/train_clas.py index a8c7e0b..1c926f8 100644 --- a/ulmfit/train_clas.py +++ b/ulmfit/train_clas.py @@ -321,7 +321,7 @@ class CLSHyperParams(LMHyperParams): args['text_cols'] = list(trn_df.columns.values)[1:] args['mark_fields'] = True lm_suffix = self.bptt if self.bptt != 70 else "" - lm_suffix = self.use_tst_for_lm if "" else "-notst" + lm_suffix += self.use_tst_for_lm if "" else "-notst" data_lm = self.lm_databunch(f'lm{lm_suffix}', train_df=lm_trn_df, valid_df=lm_val_df, bs=bs, force=force, bptt=self.bptt, **args) args['vocab'] = data_lm.vocab data_cls = self.cls_databunch(cls_name, train_df=trn_df, valid_df=val_df, bs=bs, force=force, **args)