/dailyBruin

predicting article views based on article contents and associated picture

Primary LanguageJupyter NotebookMIT LicenseMIT

DailyBruin

Uses transfer learning with Berta and Resnet to make predictions. Achieved 78% accuracy. Still a WIP.

Dependencies

  • pandas

Process Data

Example usage:

python3 process_data.py --csv ./data/page_views.csv --json ./data/articles.json --output ./output.csv --force