Upload agrega_sentimientos.py

agrega_sentimientos.py

@@ -0,0 +1,27 @@
+import pandas as pd
+from pysentimiento import create_analyzer
+def genera_excel_sentimientos(noticias):
+    titulos_buscados=noticias.titulo
+    from pysentimiento import create_analyzer
+    analyzer = create_analyzer(task="sentiment", lang="es")
+    sentimientos=[]
+    negativos=[]
+    neutros=[]
+    positivos=[]
+    for titulo in titulos_buscados:
+        prediccion=analyzer.predict(titulo)
+        sentimientos.append(prediccion.output)
+        negativos.append(prediccion.probas['NEG'])
+        neutros.append(prediccion.probas['NEU'])
+        positivos.append(prediccion.probas['POS']) 
+        print(prediccion)
+    
+    noticias=noticias.copy()
+    noticias['sentimiento']=sentimientos
+    noticias['pond_negativos']=negativos
+    noticias['pond_neutro']=neutros
+    noticias['pond_positivo']=positivos
+
+    #noticias.to_excel("noticias_con_sentimientos.xlsx")
+
+    return noticias