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