BART-Large-MNLI Prompt Classification Model

This model is fine-tuned from facebook/bart-large-mnli on the prompt engineering dataset for classifying different types of prompts.

Model description

This model classifies prompts into different categories based on their type.

Intended uses & limitations

This model is intended for classifying text prompts for AI systems. It should be used for understanding the intent or type of a given prompt.

Training data

The model was trained on the SoftAge-AI/prompt-eng_dataset.

Training procedure

The model was trained for 3 epochs with a batch size of 4 and gradient accumulation steps of 8.

Downloads last month
12
Safetensors
Model size
407M params
Tensor type
F32
·
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Dataset used to train ru4en/bart-large-mnli-tool-router-new