A simple Graph Neural Network (GNN) project that recommends new friendships in a small social network.
Built using PyTorch Geometric in Google Colab or Locally (Python 3.9+).
- Each person = Node
- Each friendship = Edge
- GNN learns embeddings for people
- Link prediction suggests possible new friendships
- 📜 friend_recommendation.ipynb → Notebook (Colab version)
- 📜 friend_recommendation.py → Python script
- 📜 requirements.txt → Dependencies
- 📜 README.md → Documentation
- Open Google Colab
- Create a new notebook
- Install dependencies:
!pip install torch torchvision torchaudio
!pip install torch-scatter torch-sparse torch-cluster torch-spline-conv torch-geometric -q
git clone https://github.com/your-username/friend-recommendation-gnn.git
cd friend-recommendation-gnn
python -m venv venv
venv\Scripts\activate
pip install -r requirements.txt
python friend_recommendation.py