Skip to content

Latest commit

 

History

History
9 lines (6 loc) · 872 Bytes

File metadata and controls

9 lines (6 loc) · 872 Bytes

About

This is a unsupervised classifier implemented with Generative Pre-trained Transformer (GPT) models. We use OpenAI's models but any other LLM provider or local models could be used instead.

Prediction

For the prompt, we use a condensed version of the classification instructions that were used by the human labelers when creating the ground truth. The prediction code creates structured output in the form of json files.

Evaluation

We used the same 250 requirements that were used to evaluate the supervised SetFit classifier on unseen data. The evaluation calculates precision, recall, and f1-score (micro, macro, weighted) and stores the results here.