如何使用Pandas格式化數據（格式化情感分析結果的數據格式）

Question

我正在使用 BERT 進行情感分析。 我想將結果轉換為 DataFrame 格式，但我不知道如何轉換。 如果有人知道，請告訴我。

相關網頁如下https://huggingface.co/transformers/main_classes/pipelines.html

>>> pipe = pipeline ("text-classification")
>>> pipe (["This restaurant is awesome", "This restaurant is aweful"])
[{'label':'POSITIVE','score': 0.9998743534088135},
  {'label':'NEGATIVE','score': 0.9996669292449951}]

輸出結果以列表格式輸出。

因此，我想將其轉換為如下所示的數據幀格式。 我應該做什么樣的處理？

Answer 1

嘗試這個：

sentiment = pipe (["This restaurant is awesome", "This restaurant is aweful"])
df = pd.DataFrame(sentiment)