|
|
--- |
|
|
license: mit |
|
|
--- |
|
|
|
|
|
# Orthogonal Model of Emotions |
|
|
|
|
|
A Text Classifier created using Sci-Kit Learn |
|
|
|
|
|
## Author |
|
|
|
|
|
C.J. Pitchford |
|
|
|
|
|
## Published |
|
|
|
|
|
18 June 2025 |
|
|
|
|
|
## Usage |
|
|
|
|
|
|
|
|
# Load the model and vectorizer |
|
|
def load_model_and_vectorizer(model_path='naive_bayes_model.pkl', vectorizer_path='vectorizer.pkl'): |
|
|
model = joblib.load(model_path) |
|
|
vectorizer = joblib.load(vectorizer_path) |
|
|
return model, vectorizer |
|
|
|
|
|
# Function to predict the label of a new text |
|
|
def predict_label(text, model, vectorizer): |
|
|
text_vec = vectorizer.transform([text]) |
|
|
prediction = model.predict(text_vec) |
|
|
return prediction[0] |
|
|
|
|
|
# Example usage |
|
|
if __name__ == "__main__": |
|
|
model, vectorizer = load_model_and_vectorizer() |
|
|
new_text = "I really, really hope this works." |
|
|
predicted_label = predict_label(new_text, model, vectorizer) |
|
|
print(f'The predicted label for the text is: {predicted_label}') |
