lin reg

TailUFPB · Oct 27, 2023 · 30d09d8 · 30d09d8
1 parent 53c5d24
commit 30d09d8
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 0 deletions.
diff --git a/api/NbLinRegressionModel.py b/api/NbLinRegressionModel.py
@@ -0,0 +1,16 @@
+import pickle
+
+def make_prediction(my_sentence):
+ with open("./models/vectorizer_lin.pkl", "rb") as f:
+ vectorizer = pickle.load(f)
+
+ with open("./models/linear_reg.pkl", "rb") as f:
+ model = pickle.load(f)
+
+ new_sentence = vectorizer.transform(my_sentence)
+
+ prediction = model.predict(new_sentence)
+ if prediction == 0:
+ return "Negativo"
+ else:
+ return "Positivo"
diff --git a/api/models/linear_reg.pkl b/api/models/linear_reg.pkl
diff --git a/api/models/vectorizer_lin.pkl b/api/models/vectorizer_lin.pkl