NisargRhino/drug-classification

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