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

Changes moved to different branch

Browse Source
This commit is contained in:
NAUSICAA\Julian
2019-02-20 17:16:08 -03:00
parent 61ae344fda
commit c5a93d19fe
1 changed files with 0 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
ulmfit/train_clas.py
-2
View File
@@ -202,8 +202,6 @@ class CLSHyperParams(LMHyperParams):
cls_name = f'{cls_name}noise{noise}'
args = self.tokenizer_to_fastai_args(sp_data_func=lambda: trn_df[1], use_moses=use_moses)
args['text_cols'] = list(trn_df.columns.values)[1:]
args['mark_fields'] = True
data_lm = self.lm_databunch('lm', train_df=lm_trn_df, valid_df=lm_val_df, bs=bs, force=force, **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)