diff --git a/.gitignore b/.gitignore new file mode 100644 index 0000000..5138364 --- /dev/null +++ b/.gitignore @@ -0,0 +1,2 @@ +# virtual environment +env/ \ No newline at end of file diff --git a/README.md b/README.md index 69b2937..b0a9cd0 100644 --- a/README.md +++ b/README.md @@ -13,6 +13,12 @@ This repository contains a PyTorch implementation of our NeruIPS2022 paper "Conv - scikit-learn 1.1.3 - ogb 1.3.5 +### Installation + +``` +pip install -r requirements.txt +``` + ## Datasets We provide the small datasets in the folder '/main/data'. The ogb datasets (ogbn-arxiv and ogbn-papers100M) and non-homophilous datasets (from [LINKX](https://arxiv.org/abs/2110.14446) ) can be downloaded automatically. diff --git a/requirements.txt b/requirements.txt new file mode 100644 index 0000000..0153b20 --- /dev/null +++ b/requirements.txt @@ -0,0 +1,11 @@ +torch==1.11.0 +numpy==1.23.4 +torch-geometric==1.7.2 +tqdm==4.64.1 +scipy==1.9.3 +seaborn==0.12.0 +scikit-learn==1.1.3 +ogb==1.3.5 + + +