Skip to main content

Example: Toxic Comment Classifier

Open In Colab

In this example, we will illustrate the process of utilizing the Predibase SDK to fine-tune the Llama-8B-Instruct model for the task of classifying toxic comments on social media platforms. This comprehensive example will guide you through the entire end-to-end process, starting with environment setup, moving through data preparation, fine-tuning the model, and performing batch inference, and culminating with an in-depth model quality evaluation.