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
- 14