config.py : contains configration like batch size etc.
create_folds: create k folds on dataset.
dataset.py : custom dataloader
engine.py : train and validation functions
lstm.py : bidirectional lstm model
train.py : use to train
spctr01/IMDb
Folders and files
| Name | Name | Last commit date | ||
|---|---|---|---|---|