metadata
license: mit
datasets:
- liuganghuggingface/InfoAlign-Data
pipeline_tag: graph-ml
The pretrained model for the paper "Learning Molecular Representation in a Cell".
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.