File size: 404 Bytes
05293b5
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
# 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