@ -33,6 +33,7 @@ Specially, BGCF contains two main modules. The first is sampling, which produce
aggregate the neighbors sampling from nodes consisting of mean aggregator and attention aggregator.
# [Dataset](#contents)
Note that you can run the scripts based on the dataset mentioned in original paper or widely used in relevant domain/network architecture. In the following sections, we will introduce how to run the scripts using the related dataset below.
- Dataset size:
Statistics of dataset used are summarized as below:
@ -61,10 +62,6 @@ aggregate the neighbors sampling from nodes consisting of mean aggregator and at
sh run_process_data_ascend.sh [SRC_PATH]
```
- Launch
```
# Generate dataset in mindrecord format for Amazon-Beauty.
sh ./run_process_data_ascend.sh ./data
# [Features](#contents)
@ -128,12 +125,12 @@ Parameters for both training and evaluation can be set in config.py.
```python
"learning_rate": 0.001, # Learning rate
"num_epochs": 600, # Epoch sizes for training
"num_epoch": 600, # Epoch sizes for training
"num_neg": 10, # Negative sampling rate
"raw_neighs": 40, # Num of sampling neighbors in raw graph
"gnew_neighs": 20, # Num of sampling neighbors in sample graph
"input_dim": 64, # User and item embedding dimension
"l2_coeff": 0.03 # l2 coefficient
"l2": 0.03 # l2 coefficient
"neighbor_dropout": [0.0, 0.2, 0.3]# Dropout ratio for different aggregation layer
"num_graphs":5 # Num of sample graph
```
@ -200,8 +197,8 @@ Parameters for both training and evaluation can be set in config.py.
@ -30,6 +30,7 @@ Graph Attention Networks(GAT) was proposed in 2017 by Petar Veličković et al.
Note that according to whether this attention layer is the output layer of the network or not, the node update function can be concatenate or average.
# [Dataset](#contents)
Note that you can run the scripts based on the dataset mentioned in original paper or widely used in relevant domain/network architecture. In the following sections, we will introduce how to run the scripts using the related dataset below.
- Dataset size:
Statistics of dataset used are summerized as below:
@ -175,7 +176,7 @@ Parameters for both training and evaluation can be set in config.py.
@ -28,6 +28,8 @@ GCN contains two graph convolution layers. Each layer takes nodes features and a
# [Dataset](#contents)
Note that you can run the scripts based on the dataset mentioned in original paper or widely used in relevant domain/network architecture. In the following sections, we will introduce how to run the scripts using the related dataset below.
| Dataset | Type | Nodes | Edges | Classes | Features | Label rate |
mindspore-ci-bot
4 years ago
committed by
Gitee