未验证 提交 d17ede81 编写于 作者: H Huang Zhengjie 提交者: GitHub

Merge pull request #53 from Liwb5/develop

update README and config for ogb mol graph classification
......@@ -24,3 +24,9 @@ python -u main.py --config hiv_config.yaml
export CUDA_VISIBLE_DEVICES=2
python -u main.py --config pcba_config.yaml
```
### Experiment results
| model | hiv (rocauc)| pcba (prcauc)|
|-------|-------------|--------------|
| GIN |0.7719 (0.0079) | 0.2232 (0.0018) |
task_name: hiv.0423
task_name: hiv
use_cuda: True
seed: 1667
seed: 15391
dataset_name: ogbg-molhiv
eval_metrics: null
task_type: null
......@@ -31,7 +31,7 @@ num_workers: 12
output_dir: ./outputs/
# trainging config
epoch: 40
epoch: 50
learning_rate: 0.0001
lr_scheduler: linear_warmup_decay
weight_decay: 0.01
......
task_name: pcba.0423
task_name: pcba
use_cuda: True
seed: 1667
seed: 28994
dataset_name: ogbg-molpcba
eval_metrics: null
task_type: null
......@@ -31,7 +31,7 @@ num_workers: 12
output_dir: ./outputs/
# trainging config
epoch: 40
epoch: 50
learning_rate: 0.005
lr_scheduler: linear_warmup_decay
weight_decay: 0.01
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册