You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Copy file name to clipboardExpand all lines: README.md
+11-4Lines changed: 11 additions & 4 deletions
Original file line number
Diff line number
Diff line change
@@ -12,8 +12,12 @@ Please run `pip install -r requirements.txt` to install all the dependencies.
12
12
13
13
## Usage
14
14
### 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;
17
21
18
22
### Model Pre-training
19
23
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:
27
31
28
32
```finetune.py``` contains the code of model finetuning for downstream tasks.
29
33
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.
32
36
33
37
34
38
## Dataset
@@ -899,5 +903,8 @@ Load model from epoch 199
899
903
Model Testing: f1_micro_test: 0.8791 f1_macro_test: 0.8640 auc_test: 0.9805
0 commit comments