Facebook Friend Prediction

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/.

Tools Used

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.