this post was submitted on 16 Nov 2023
1 points (100.0% liked)

Machine Learning

1 readers
1 users here now

Community Rules:

founded 1 year ago
MODERATORS
 

I'm looking for suggestions for a transformer model that I can fine-tune for a text classification task. Due to hardware constraints the model has to be fairly small. Something in the order of a 50 MB weight file.

you are viewing a single comment's thread
view the rest of the comments
[–] kwnaidoo@alien.top 1 points 1 year ago

While not a transformer, what about "Gaussian naive Bayes"? It's not the best classifier around but for some tasks - it's good enough. I used it to build a small search term classifier model which basically classifies e-commerce search terms against a category or tag.