opentrain 0.0.1

🚂 Fine-tune OpenAI models for text classification, question answering, and more

opentrain

🚂 Fine-tune OpenAI models for text classification, question answering, and more

---

opentrain is a simple Python package to fine-tune OpenAI models for task-specific purposes such as text classification, token classification, or question answering.

Usage

import openai
from opentrain.train import OpenAITrainer
openai.api_key = "<ADD_OPENAI_API_KEY_HERE>"
trainer = OpenAITrainer(model="ada")
trainer.train(
    [
        {
            "prompt": "I love to play soccer ->",
            "completion": " soccer",
        },
        {
            "prompt": "I love to play basketball ->",
            "completion": " basketball",
        },
    ],
)