Supported Models
AIVM currently supports the following pre-trained models for specific learning tasks. You can either utilize these existing models or use the training scripts to produce your own custom-trained versions:
BertTiny
- SMS Spam Classification
- Binary classification for detecting spam messages
- Input: Text string
- Output: Binary classification (spam/not spam)
- Movie Rating Sentiment Analysis
- Training Script
- Sentiment analysis for movie reviews
- Input: Text string
- Output: Sentiment score (-1 to 1) for positive and negative
- Tweet Sentiment Analysis
- Training Script
- Sentiment analysis for tweets in
- Input: Text string
- Output: Sentiment score (-1 to 1) for positive, neutral and negative
LeNet5
- Handwritten Digit Recognition (MNIST)
- Classification of handwritten digits
- Input: 28x28 grayscale image
- Output: Digit classification (0-9)
- Cats vs Dogs Classification
- Training Script
- Binary image classification
- Input: 28x28 grayscale image
- Output: Binary classification (cat/dog)