Hugging Face's logo

liuganghuggingface
/
InfoAlign-Pretrained

Graph Machine Learning
Model card Files Community

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.

Downloads last month
15
Inference Providers NEW
Graph Machine Learning
This model is not currently available via any of the supported Inference Providers.
The model cannot be deployed to the HF Inference API: The model has no library tag.

Dataset used to train liuganghuggingface/InfoAlign-Pretrained

Collection including liuganghuggingface/InfoAlign-Pretrained