Downloads · 30 days
31
2% of all-time downloads
custom-gcn
Huhujingjing/custom-gcn
custom-gcn is a graph machine learning model from Huhujingjing. Use it for the graph machine learning task on the model card, and read the license before you ship it in a product. It is set up for transformers. The card lists the license as mit.
All-time downloads
1.3K
Public
Repo size
35.7 KB
Likes
2
Public
Repo makeup
Click a slice to open those files.
.bin35.7 KB · 72%
At a glance
- Task
- Graph Machine Learning
- Library
- transformers
- License
- mit
- Model type
- gcn
- Access
- Public
- Created
- Jul 1, 2023
- Updated
- Jul 1, 2023
- SHA
- 1f0c4c29
Datasets
Model card
From the Hugging Face model README
pip install torch_scatter rdkit
pip install torch_geometric
example
from transformers import AutoModel
device = 'cpu'
smiles = ['CCCC', 'C#C/C(C)=C(/[CH2])C', 'NNNNN']
model = AutoModel.from_pretrained("Huhujingjing/custom-gcn", trust_remote_code=True).to(device)
output = model.predict_smiles(smiles)
print(output)
- Task
- Graph Machine Learning
- Library
- transformers
- Type
- gcn
- License
- mit
- Languages
- en
- Created
- Jul 1, 2023
- Updated
- Jul 1, 2023