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

Maki ti possible to not load unusp

Browse Source
This commit is contained in:
Piotr Czapla
2019-03-26 21:29:23 +01:00
parent 24f1d741f0
commit 0a823be17b
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
ulmfit/train_clas.py
+2 -2
View File
@@ -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")