---
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`.