Predicting popularity of images with reddit community scores
- Reddit Crawler (to download data off reddit)
- praw 7.5.0
- Pillow 9.0.1
- Requests 2.27.1
- A suitable reddit API OAuth token
Place Reddit data under data/reddit
Collate Reddit dataset
python -m dataset.collate_reddit_data
Train model
CUBLAS_WORKSPACE_CONFIG=:4096:8 python -m train.train_model