metadata

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.

Training procedure

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