将psyqa做成qa对

Question

想把psyqa这个数据集做成qa对来微调chatglm2-6b，需要把里面的description和answer_text提取出来，然后变成content和summary的字典

Answer 1

【以下回答由 GPT 生成】

import pandas as pd

# 读取数据集
psyqa = pd.read_csv('psyqa.csv')

# 提取description和answer_text列
description = psyqa['description']
answer_text = psyqa['answer_text']

# 将数据转换成字典形式
data = {'content': description.tolist(), 'summary': answer_text.tolist()}

# 打印转换后的数据
print(data)

如果你已经解决了该问题, 非常希望你能够分享一下解决方案, 写成博客, 将相关链接放在评论区, 以帮助更多的人 ^-^