This repository contains a machine learning model for predicting who would be friends in a social network. We empirically validate our findings via the Facebook dataset from SNAP.
A summary of our findings can be found at https://paul-tqh-nguyen.github.io/facebook_friend_prediction/.
The following tools were utilized:
- NetworkX
- karateclub
- scikit-learn
- Optuna
- TensorBoard
- PyTorch
- PyTorch Lightning
- Joblib
- Pandas
- Matplotllib
- nvgpu
- NumPy
- tqdm
Other Python libraries used include contextlib, typing-extensions, more-itertools, multiprocessing, typing, collections, argparse, json, random, os, and logging.