Skip to content

Classification Models

Pre-trained models for image, text, and tabular classification.

Image Classification

Model Trained On Classes Size Framework License Link
ResNet-50 ImageNet 1000 97 MB PyTorch/ONNX BSD HF
EfficientNet-B0 ImageNet 1000 21 MB TF/PyTorch Apache-2.0 HF
MobileNet v3 ImageNet 1000 22 MB TF/PyTorch Apache-2.0 TF Hub
ViT-Base ImageNet-21K 21K 330 MB PyTorch Apache-2.0 HF
ConvNeXt-Tiny ImageNet 1000 110 MB PyTorch Apache-2.0 HF

Text Classification

Model Trained On Task Size Browser? Link
distilbert-base-uncased-finetuned-sst-2 SST-2 Sentiment (pos/neg) 67 MB Yes HF
roberta-base-go_emotions GoEmotions 28 emotions 125 MB Yes HF
finbert Financial texts Fin sentiment 125 MB Yes HF
distilbert-base-uncased-mnli MultiNLI Zero-shot NLI 67 MB Yes HF
twitter-roberta-base-sentiment Twitter 3-class sentiment 125 MB Yes HF

Tabular Classification

For tabular data, pre-trained models are less common — but these libraries let you train classifiers in the browser:

Library Algorithms Browser? Link
TensorFlow.js Neural nets, transfer learning Yes tensorflow.org/js
ml5.js KNN, neural nets Yes ml5js.org
Brain.js Neural nets Yes brain.js.org
scikit-learn (via Pyodide) RF, SVM, XGBoost Yes (slow) pyodide.org