ru4en/bart-large-mnli-agent-router

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.