CitGraph employs Graph Neural Networks (GNNs) to analyze citation networks, specifically focusing on node classification tasks.
Leveraging PyTorch Geometric and the Planetoid dataset, this project aims to classify nodes within citation graphs, offering insights into citation patterns and relationships.
This project implements a Graph Neural Network (GNN) for node classification using PyTorch Geometric.
It leverages the Planetoid dataset to demonstrate the application of GNNs in classifying nodes in a citation network.
The project utilizes PyTorch Geometric due to its efficient implementation of GNN layers and convenient dataset handling.
PyTorch Lightning is employed for streamlined training and experimentation.
Challenges encountered include dataset preparation and hyperparameter tuning.
Future features may include implementing more advanced GNN architectures and exploring additional datasets for node classification tasks.
If you haven't already created a virtual environment for your project, you can do so using virtualenv or venv. Here's an example using venv:
python -m venv myenv
Replace myenv with the desired name for your virtual environment.
On Windows, activate the virtual environment using:
myenv\Scripts\activate
On macOS and Linux, use:
source myenv/bin/activate
Replace myenv with the name of your virtual environment.
Once the virtual environment is activated, you can install Jupyter Notebook using pip:
pip install jupyter ipykernel torch lightning torch_geometric
This will install Jupyter Notebook within your virtual environment.
To verify that Jupyter Notebook is installed in your virtual environment, you can run:
jupyter --version
This should display the version of Jupyter Notebook installed within your virtual environment.
You need to create a Jupyter Notebook kernel that is associated with your virtual environment. This allows you to use the packages installed in your virtual environment within Jupyter Notebook.
pip install ipykernel
python -m ipykernel install --user --name=myenv --display-name="name"
Replace myenv with the name of your virtual environment and choose a suitable display name.
Now, you can start Jupyter Notebook from within your virtual environment:
jupyter notebook
This will open a new Jupyter Notebook session in your web browser, and you should be able to select the "My Virtual Environment" kernel when creating a new notebook. This kernel will use the packages installed in your virtual environment.
Run all jupyter notebooks cells
Make a pull request