INF554 Machine Learning I (2018-2019, M. Vazirgiannis, Ecole Polytechnique) - Final project - Guillaume Dufau, Christopher Murray, Léon Zheng
Edges have been deleted at random from a citation network. The objective is to reconstruct the initial network using graph and text information.
Please find in a pdf format the report for this project.
Here are the results of our team - The stormtroopers - top 23%: kaggle/evaluation
The project is composed of two Python files, main.py and main_kaggle.py. The former allows model testing and hyper-parameters tuning on the training set where the latter is constructing the predictions file for kaggle.
Libraries used:
- Scikit-learn
- NetworkX
- nltk
- matplotlib
- Guillaume Dufau - GuillaumeDufau
- Christopher Murray - murrman95
- Leon Zheng - leonzheng2