README.md

---
language: en
license: mit
datasets:
  - SoftAge-AI/prompt-eng_dataset
tags:
  - text-classification
  - prompt-classification
  - bart
  - bart-large-mnli
---

# 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](https://huggingface.co/datasets/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.