---
base_model:
- openai/clip-vit-base-patch16
---

# Fine-Tuned ViT on FERPlus Dataset

This model is a fine-tuned version of ViT on the [FERPlus dataset](https://www.kaggle.com/datasets/arnabkumarroy02/ferplus), which is more balanced ferplus dataset as owner claimed.

## Model Details
- **Base Model**: [Clip-ViT-Base-Patch16](https://huggingface.co/openai/clip-vit-base-patch16)
- **Dataset**: FERPlus, which contains grayscale images of faces labeled with emotion categories.
- **Task**: Emotion Classification
- **Accuracy** %82.4
- **Labels**:
  - 0: Angry
  - 1: Contempt
  - 2: Disgust
  - 3: Fear
  - 4: Happy
  - 5: Neutral
  - 6: Sad
  - 7: Surprise

### Training Hyperparameters
- **Batch Size**: 16
- **Epochs**: 1
- **Learning Rate**: 2e-5
- **Weight Decay**: 0.01