`train`

Module Contents

`train`(→ src.model_finetuning.model.BertLightningModel)
`predict`(→ pandas.DataFrame)
`main`()

train.train(config: dict, train_df: pandas.DataFrame, val_df: pandas.DataFrame, verbose: bool = False) → src.model_finetuning.model.BertLightningModel

train.predict(config: dict, model: src.model_finetuning.model.BertLightningModel, df: pandas.DataFrame) → pandas.DataFrame