test / train /README.md
abdullahmeda's picture
-
a4d40bc
# Train new model
- Download and extract the following datasets in a new folder called datasets:
1. [IMDb movies extensive dataset](https://www.kaggle.com/stefanoleone992/imdb-extensive-dataset)
2. [48K IMDB Movies With Posters](https://www.kaggle.com/rezaunderfit/48k-imdb-movies-with-posters)
- Run `create_dataset.ipynb` to create train.csv and valid.csv
- Run `train.ipynb` to train the model