Mastodon Feed: Post

Mastodon Feed

lloydmeta ("Lloyd") wrote:

Trained a multiclass classification model, built on top of google-bert/bert-base-uncased, tuned on the Drug Review Dataset (Drugs.com). It's useful for making a best attempt classification for the condition someone has, based on their review of a drug...

Done as more of a learning exercise / refresher on the whole ML scene after being basically away for 8 (!) years. So much has changed .. tooling, education, etc are MUCH improved.

https://huggingface.co/lloydmeta/drug-bert

#genai #ml #transformers #ai