14 lines
455 B
Python
14 lines
455 B
Python
import pandas as pd
|
|
from datasets import Dataset
|
|
|
|
def get_chat_data():
|
|
a = pd.read_excel("致远速联日常销售问答话术.xlsx")
|
|
a.fillna("", inplace=True)
|
|
data = []
|
|
for i in a.values:
|
|
if i[1] == "": continue
|
|
data.append({"conversations": [{"role": "user", "content": i[1]}, {"role": "assistant", "content": i[2]}]})
|
|
return Dataset.from_pandas(pd.DataFrame(data*10))
|
|
|
|
if __name__ == '__main__':
|
|
get_chat_data() |