diff --git a/api/DataProcesser.py b/api/DataProcesser.py index 1efcf5e3..3a233cb9 100644 --- a/api/DataProcesser.py +++ b/api/DataProcesser.py @@ -1,7 +1,7 @@ import pandas as pd from NbNewsModel import NbNewsModel from NbEmotionsModel import make_prediction -from NbLinRegressionModel import make_prediction +from NbLinRegressionModel import make_prediction_nblin from io import BytesIO import nltk import re @@ -45,12 +45,12 @@ def classify_emotions(self, df): self.preprocess_text).apply(make_prediction) result_csv = df # converte o df pra csv return result_csv - + def lin_regression_model(self, df): df['output_column'] = df['input_column'].apply( - self.preprocess_text).apply(make_prediction) + self.preprocess_text).apply(make_prediction_nblin) result_csv = df # converte o df pra csv return result_csv - + # TODO métodos com o processamento de classificação diff --git a/api/NbLinRegressionModel.py b/api/NbLinRegressionModel.py index bb29e6d9..4281320a 100644 --- a/api/NbLinRegressionModel.py +++ b/api/NbLinRegressionModel.py @@ -1,6 +1,6 @@ import pickle -def make_prediction(my_sentence): +def make_prediction_nblin(my_sentence): with open("./models/vectorizer_lin.pkl", "rb") as f: vectorizer = pickle.load(f) @@ -8,7 +8,7 @@ def make_prediction(my_sentence): model = pickle.load(f) new_sentence = vectorizer.transform(my_sentence) - + prediction = model.predict(new_sentence) if prediction == 0: return "Negativo" diff --git a/main.js b/main.js index 9e53bcf7..22071684 100644 --- a/main.js +++ b/main.js @@ -20,10 +20,10 @@ function createWindow() { show: false, // Ocultar a janela até que esteja pronta //fullscreen: true, }); - win.loadURL("http://localhost:3000/"); // carrega a página hospedada localmente - //win.loadFile(path.join(__dirname, "./build/index.html")); // carrega a build do react + //win.loadURL("http://localhost:3000/"); // carrega a página hospedada localmente + win.loadFile(path.join(__dirname, "./build/index.html")); // carrega a build do react - /* + /* win.loadURL( url.format({ pathname: path.join(__dirname, "./build/index.html"), // relative path to the HTML-file