Downloads · 30 days
4
1% of all-time downloads
NisargRhino/drug-classification
drug-classification is a machine learning model from NisargRhino. Use it for the machine learning task on the model card, and read the license before you ship it in a product. The card lists the license as unknown.
Downloads · 30 days
4
1% of all-time downloads
All-time downloads
350
Public
Parameters
44.1M
176 MB on disk
Likes
1
Public
Click a slice to open those files.
.safetensors176 MB · 100%
From the Hugging Face model README
Hello Everyone,
This model uses the Roberta / ChemBERTa Machine Learning model, that has been edited and trained for Drug Classification.
It was trained to predict usecases of drugs by inputting SMILES strings. Ex. If you input a drug SMILES used to treat Heart Disease, it will output 'Heart Disease'
Thank you,
Nisarg