Skip to content

Commit e866454

Browse files
Update README.md
1 parent a8a30f4 commit e866454

File tree

1 file changed

+11
-4
lines changed

1 file changed

+11
-4
lines changed

README.md

Lines changed: 11 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -12,8 +12,12 @@ Please run `pip install -r requirements.txt` to install all the dependencies.
1212

1313
## Usage
1414
### Data Download
15-
As the data size is too large, please click on the link to download the [AMiner Data](https://drive.google.com/file/d/16X9y8iBL62j88YdhDaZQLOheuYlxFWiD/view?usp=sharing).
16-
Once download the data, please unzip the file and further put the unzipped data into the aminer_data folder.
15+
As the data size is too large, please click on the link to download the [AMiner Data](https://drive.google.com/file/d/1TJfviH4--HPp12jQnqbJEpO58GrHCexJ/view?usp=share_link).
16+
17+
Tips: the file size is quite big including (a.) the AMiner data, (b.) the fine-tuned node features (finetune_feature_20301141240.txt), and (c.) the pre-trained pt model file (aminer_node_text_20301141240.pt).
18+
19+
Once finish downloading, please unzip the file and further put the unzipped data (part a) to the "data/aminer_data/" folder; the fine-tuned node features (part b) to "finetune/aminer_data/" folder;
20+
the pre-trained pt file (part c) into the "pretrain/" folder;
1721

1822
### Model Pre-training
1923
For co-modality pre-training, we consider two types of modality combinations, i.e., graph modality and text modality, and
@@ -27,8 +31,8 @@ graph modality and image modality in our paper:
2731

2832
```finetune.py``` contains the code of model finetuning for downstream tasks.
2933

30-
The default setting for AMiner data is all set. If you want to train the model, please run the code ```python main_graph_text_gcl.py``` over AMiner data after installing all required packages.
31-
It may take a while, you can also simply run the code ```python finetune.py``` to reproduce our results. We also provide a sample running log for the code above.
34+
The default setting for AMiner data is all set. If you want to train the model, please run the code ```python main_graph_text_gcl.py``` over AMiner data.
35+
It may take a while, you can also simply run the code ```python finetune.py``` to reproduce our results. We also provide a sample running log below.
3236

3337

3438
## Dataset
@@ -899,5 +903,8 @@ Load model from epoch 199
899903
Model Testing: f1_micro_test: 0.8791 f1_macro_test: 0.8640 auc_test: 0.9805
900904
901905
Process finished with exit code 0
906+
907+
902908
```
903909

910+

0 commit comments

Comments
 (0)