项目文件夹

文件

T

文件历史

Mufei Li a260a6e615 [Example] Fix GATNE-T (#2061 )

* Update README.md

* Update

* Fix

Co-authored-by: Ubuntu <ubuntu@ip-172-31-1-5.us-west-2.compute.internal>

2020-08-19 17:01:56 +08:00

scripts

[Model] GATNE-T (#1470 )

2020-05-05 22:35:25 -07:00

src

[Example] Fix GATNE-T (#2061 )

2020-08-19 17:01:56 +08:00

README.md

[Example] Fix GATNE-T (#2061 )

2020-08-19 17:01:56 +08:00

requirements.txt

[Model] GATNE-T (#1470 )

2020-05-05 22:35:25 -07:00

README.md

Representation Learning for Attributed Multiplex Heterogeneous Network (GANTE)

Paper link: https://arxiv.org/abs/1905.01669
Author's code repo: https://github.com/THUDM/GATNE. Note that only GATNE-T is implemented here.

Requirements

requirements

pip install -r requirements.txt

Datasets

To prepare the datasets:

```
mkdir data
cd data
```
Download datasets from the following links:
Unzip the datasets

Training

Run with following (available dataset: "example", "youtube", "amazon")

python src/main.py --input data/example

To run on "twitter" dataset, use

python src/main.py --input data/twitter --eval-type 1

Results

All the results match the official code with the same hyper parameter values, including twiiter dataset (auc, pr, f1 is 76.29, 76.17, 69.34, respectively).

	auc	pr	f1
amazon	96.88	96.31	92.12
youtube	82.29	80.35	74.63
twitter	72.40	74.40	65.89
example	94.65	94.57	89.99