Downloads · 30 days
11
6% of all-time downloads
InfoAlign-Pretrained
liuganghuggingface/InfoAlign-Pretrained
InfoAlign-Pretrained is a graph machine learning model from liuganghuggingface. Use it for the graph machine learning task on the model card, and read the license before you ship it in a product. The card lists the license as mit.
The pretrained model for the paper "Learning Molecular Representation in a Cell".
All-time downloads
187
Public
Repo size
55.3 MB
Likes
1
Public
Repo makeup
Click a slice to open those files.
.pt55.3 MB · 100%
At a glance
- Task
- Graph Machine Learning
- License
- mit
- Access
- Public
- Created
- Jun 19, 2024
- Updated
- Jul 2, 2024
- SHA
- 96977bb2
Model card
From the Hugging Face model README
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.
- Task
- Graph Machine Learning
- License
- mit
- Created
- Jun 19, 2024
- Updated
- Jul 2, 2024