databoyface commited on
Commit
707aa50
·
verified ·
1 Parent(s): f914843

Added author, date, usage

Browse files
Files changed (1) hide show
  1. README.md +35 -3
README.md CHANGED
@@ -1,3 +1,35 @@
1
- ---
2
- license: mit
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ ---
4
+
5
+ # Orthogonal Model of Emotions
6
+
7
+ ## Author
8
+
9
+ C.J. Pitchford
10
+
11
+ ## Published
12
+
13
+ 18 June 2025
14
+
15
+ ## Usage
16
+
17
+
18
+ # Load the model and vectorizer
19
+ def load_model_and_vectorizer(model_path='naive_bayes_model.pkl', vectorizer_path='vectorizer.pkl'):
20
+ model = joblib.load(model_path)
21
+ vectorizer = joblib.load(vectorizer_path)
22
+ return model, vectorizer
23
+
24
+ # Function to predict the label of a new text
25
+ def predict_label(text, model, vectorizer):
26
+ text_vec = vectorizer.transform([text])
27
+ prediction = model.predict(text_vec)
28
+ return prediction[0]
29
+
30
+ # Example usage
31
+ if __name__ == "__main__":
32
+ model, vectorizer = load_model_and_vectorizer()
33
+ new_text = "I really, really hope this works."
34
+ predicted_label = predict_label(new_text, model, vectorizer)
35
+ print(f'The predicted label for the text is: {predicted_label}')