Cyberbullying Classification Model

This model classifies text into six categories of cyberbullying:

  • religion
  • age
  • ethnicity
  • gender
  • other_cyberbullying
  • not_cyberbullying

Usage

import pickle
import requests

# Load the model and vectorizer
model = pickle.load(open('model.pkl', 'rb'))
vectorizer = pickle.load(open('vectorizer.pkl', 'rb'))
preprocess_func = pickle.load(open('preprocess.pkl', 'rb'))

# Preprocess and predict
def predict(text):
    processed_text = preprocess_func(text)
    vectorized_text = vectorizer.transform([processed_text])
    prediction = model.predict(vectorized_text)[0]
    return prediction
Downloads last month

-

Downloads are not tracked for this model. How to track
Inference Providers NEW
This model isn't deployed by any Inference Provider. ๐Ÿ™‹ Ask for provider support