File size: 2,127 Bytes
fb13f9f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 |
import streamlit as st
import pandas as pd
import gensim
import joblib
import numpy as np
import tensorflow as tf
from tensorflow.keras.models import Sequential, Model
from tensorflow.keras.layers import Embedding, TextVectorization, GlobalAveragePooling1D, Input, LSTM, GRU, Dropout, Dense
from scraper import scrape
from func import TextProcess, Label, TextProcess2
lda_model_inf = gensim.models.LdaModel.load('lda.model')
model = tf.keras.models.load_model('model_imp')
# with open('model_imp.pkl', 'rb') as file_1:
# model = joblib.load(file_1)
with open('dictionary.pkl', 'rb') as file_4:
dictionary_inf = joblib.load(file_4)
def run():
# Membuat Title
st.title('Negative Review clustering insights')
# membuat form
with st.form(key='form_parameters'):
text_input = st.text_input('Reviews', 'Barangnya tidak sesuai gambar')
submitted_text = st.form_submit_button('Show topic')
data_inf = {
'text': text_input
}
data_inf = pd.DataFrame([data_inf])
if submitted_text:
data_inf['text_processed'] = data_inf['text'].apply(lambda x: TextProcess(x))
Message_body_inf = data_inf['text_processed']
data_inf['text_processed_2'] = data_inf['text'].apply(lambda x: TextProcess2(x))
Message_body_inf_2 = data_inf['text_processed_2']
y_pred_inf = model.predict(Message_body_inf_2)
y_pred_inf = np.where(y_pred_inf >= 0.5, 1, 0)
if y_pred_inf == 0:
st.write('Negative sentiment')
for i in range(len(Message_body_inf)):
bow_vector = dictionary_inf.doc2bow(Message_body_inf[i])
for index, score in sorted(lda_model_inf[bow_vector], key=lambda tup: -1*tup[1]):
# st.write(f"Score: {round(score*100)}%")
st.write(f"Complaint: {Label(index)}")
# st.write("Complaint: {}".format(Label(index)))
break
st.write('------------------------------------')
else:
st.write('Positive sentiment')
if __name__ == '__main__':
run() |