https://github.com/snap-research/nodedup
Node Duplication Improves Cold-start Link Prediction
https://github.com/snap-research/nodedup
cold-start data-augmentation data-augmentation-strategies gnn gnns graph graphml link-prediction ml recommendation-system recommender-system
Last synced: 3 months ago
JSON representation
Node Duplication Improves Cold-start Link Prediction
- Host: GitHub
- URL: https://github.com/snap-research/nodedup
- Owner: snap-research
- License: other
- Created: 2023-10-04T18:27:28.000Z (about 2 years ago)
- Default Branch: main
- Last Pushed: 2024-06-20T01:56:06.000Z (over 1 year ago)
- Last Synced: 2025-03-05T23:22:20.129Z (7 months ago)
- Topics: cold-start, data-augmentation, data-augmentation-strategies, gnn, gnns, graph, graphml, link-prediction, ml, recommendation-system, recommender-system
- Language: Python
- Homepage:
- Size: 74.2 KB
- Stars: 2
- Watchers: 6
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
# Node Duplication Improves Cold-start Link Prediction
## Introduction
This repository contains the source code for the paper: [Node Duplication Improves Cold-start Link Prediction](https://arxiv.org/pdf/2402.09711)
## Requirements
The code was developed and tested with Python 3.10.9. The enviromental requirements are listed as in `requirements.txt`. Please run the following code to install all the requirements:
```
pip install -r requirements.txt
```
## Prepare the datasets
IGB-100K dataset needs to be pre-downloaded and pre-processed. Please install IGB package and download IGB-100k datset as follows:
```
git clone https://github.com/IllinoisGraphBenchmark/IGB-Datasets.git
cd IGB-Datasets/
pip install .
cd IGB-Datasets
python
from igb import dataloader
from igb import download
download.download_dataset(path='../IGB-Datasets', dataset_type='homogeneous', dataset_size='tiny')
```
Then pre-process IGB-100K datasets with the following command:
```
cd data
python igb_process.py
```
Other datasets can be directly downloaded when running the experimental code.
### Usage
#### Transductive Setting
Please run the following command to reproduce the results in Table 1:
```
bash scripts/transductive.sh
```
#### Production Setting
Please run the following command to reproduce the results in Table 2 and Table 7 in the paper:
```
bash scripts/inductive.sh
```
## Citation
If you use this code in your research, please cite the following paper:
```bibtex
@article{guo2024node,
title={Node Duplication Improves Cold-start Link Prediction},
author={Guo, Zhichun and Zhao, Tong and Liu, Yozen and Dong, Kaiwen and Shiao, William and Shah, Neil and Chawla, Nitesh V},
journal={arXiv preprint arXiv:2402.09711},
year={2024}
}
```