File size: 533 Bytes
d89bb1c 58739e0 d89bb1c 96977bb |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
---
license: mit
datasets:
- liuganghuggingface/InfoAlign-Data
pipeline_tag: graph-ml
---
The pretrained model for the paper ["Learning Molecular Representation in a Cell"](https://arxiv.org/abs/2406.12056v1).
The model is trained with the following configurations:
- **emb_dim**: 300
- **model**: gin-virtual
- **norm_layer**: batch_norm
- **num_layer**: 5
- **prior**: 1.0e-09
- **readout**: sum
- **threshold**: 0.8
- **walk_length**: 4
Please place the config.YAML file in the `ckpt` folder and rename it to `pretrain.yaml`. |