diff --git a/api/NbLinRegressionModel.py b/api/NbLinRegressionModel.py new file mode 100644 index 00000000..bb29e6d9 --- /dev/null +++ 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" \ No newline at end of file diff --git a/api/models/linear_reg.pkl b/api/models/linear_reg.pkl new file mode 100644 index 00000000..8006301d Binary files /dev/null and b/api/models/linear_reg.pkl differ diff --git a/api/models/vectorizer_lin.pkl b/api/models/vectorizer_lin.pkl new file mode 100644 index 00000000..bcf495e8 Binary files /dev/null and b/api/models/vectorizer_lin.pkl differ