Update clustering.py

clustering.py

@@ -858,25 +858,22 @@ def map_sentiment(estrella):
         return "positivo"
 
 
-def classify_sentiment(texto):
-    resultado = classifier(texto)[0]
-    sentimiento = map_sentiment(resultado["label"])
-    return (
-        sentimiento,
-        resultado["score"],
-    )
-
-
-def classify_sentiment_df(data, comment_col="comment"):
-
+def classify_sentiment_df(data, comment_col="comment", num_threads=4):
     def classify_sentiment(texto):
         resultado = classifier(texto)[0]
         sentimiento = map_sentiment(resultado["label"])
         return sentimiento, resultado["score"]
 
-    data["sentimiento"], data["confianza"] = zip(
-        *data[comment_col].apply(classify_sentiment)
-    )
+    def process_row(texto):
+        # Esta función procesa una fila y retorna los valores necesarios
+        return classify_sentiment(texto)
+
+    # Crear un ThreadPoolExecutor con el número de hilos especificado
+    with ThreadPoolExecutor(max_workers=num_threads) as executor:
+        results = list(executor.map(process_row, data[comment_col]))
+
+    # Dividir los resultados en columnas para "sentimiento" y "confianza"
+    data["sentimiento"], data["confianza"] = zip(*results)
 
     return data