Uses transfer learning with Berta and Resnet to make predictions. Achieved 78% accuracy. Still a WIP.
pandas
Example usage:
python3 process_data.py --csv ./data/page_views.csv --json ./data/articles.json --output ./output.csv --force
predicting article views based on article contents and associated picture
Jupyter NotebookMIT