From 0a823be17bdbb788b33e5fcbd63d0cede94fa23f Mon Sep 17 00:00:00 2001 From: Piotr Czapla Date: Tue, 26 Mar 2019 21:29:23 +0100 Subject: [PATCH] Maki ti possible to not load unusp --- ulmfit/train_clas.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/ulmfit/train_clas.py b/ulmfit/train_clas.py index b0c6a5f..479e8ba 100644 --- a/ulmfit/train_clas.py +++ b/ulmfit/train_clas.py @@ -217,8 +217,8 @@ class CLSHyperParams(LMHyperParams): val_df = pd.read_csv(val_fn, header=None) else: val_df = None - - unsup_df = pd.read_csv(self.dataset_path / f'{prefix}unsup.csv', header=None) + unsup_fn = self.dataset_path / f'{prefix}unsup.csv' + unsup_df = pd.read_csv(unsup_fn, header=None) if unsup_fn.exists() else trn_df[:0] if val_df is None: print("Validation set not found using 10% of trn")