/virality-predictor

A article's virality predictor based on article text and user's interactions.

Primary LanguageJupyter Notebook

virality-predictor

Data: https://www.kaggle.com/gspmoreira/articles-sharing-reading-from-cit-deskdrop

This repository contains following files

  • requirements.txt - Libraries used in making predictor
  • article_text_features.ipynb - Features generated from article content (text) & stores it in data/article_info.csv
  • virality_predictor.ipynb - Model and their evaluation
  • Data - Contains data
    • shared_articles.csv - from kaggle
    • users_interactions.csv - from kaggle
    • article_info.csv - features extracted from article text (article_text_features.ipynb)