Graph augmentation using node content for social circle analysis
This is the implementation of our graph augmentation algorithm for detecting social circles in online social networks. This is a class project for 10701 - Introduction to Machine Learning (PhD) - Carnegie Mellon University.
Project report can be accessed here: http://www.andrew.cmu.edu/user/reletreb/papers/ProjectReport.pdf
Packages needed: sklearn, networkx, numpy, scipy, matplotlib.pyplot, tensorflow, community
Note: community package --> https://github.com/taynaud/python-louvain
Running the code:
- Edit execute.py by adding the desired parameters
- Run execute.py