Skip to content

xinbingzhe/bipartite-link-prediction

 
 

Repository files navigation

Bipartite Link Prediction

This project explores various methods for bipartite link prediction using Yelp's Dataset Challenge dataset. In particular, we try to predict which restaurants a particular user will review. This was done as a class project for Stanford's Social and Information Network Analysis class (CS224W). The final report is included in the repository (writeups/final_report.pdf). It requires scikit-learn, networkx, and snap.py to run.

Running

  1. Place Yelp academic datasets in data/provided
  2. Run dataset_maker.py to generate examples
  3. Run any of the following files:
  • dataset_metrics.py (prints various properties of the dataset)
  • random_baseline.py (random predictions)
  • random_walks.py (make predictions using unsupervised random walks)
  • similarity.py (make predictions using heuristic similarity measures)
  • supervised_classifier.py (make predictions using a supervised binary classifier)
  • supervised_random_walks.py (make predictions using supervised random walks, see Backstrom and Leskovec, 2011)
  • svd.py (make predictions using matrix factorization)
  1. Use eval.py to generate model evaluation metrics.

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%