«Ошибка ключа» при преобразовании файла CSV в файл JSONLPython

Программы на Python
Ответить
Anonymous
 «Ошибка ключа» при преобразовании файла CSV в файл JSONL

Сообщение Anonymous »


Я пытаюсь создать отформатированный файл jsonl, используя уже созданный файл csv для точной настройки chatgpt-3.5-turbo

Я создал CSV-файл с двумя столбцами «пользователь» и «помощник»


Изображение


Мне нужно создать файл jsonl, в котором каждая строка в файле csv будет выглядеть примерно так:
Изображение


Вот код, который я использую:

импортировать панд как pd DEFAULT_SYSTEM_PROMPT= "ПОВЕДКА, КОТОРАЯ Я БУДУ ИСПОЛЬЗОВАТЬ ДЛЯ ОБУЧЕНИЯ МОДЕЛИ." Защиту create_dataset (пользователь, помощник): возвращаться { "Сообщения": [ {"роль": "система", "контент": DEFAULT_SYSTEM_PROMPT}, {"роль": "пользователь", "контент": пользователь}, {"роль": "помощник", "содержание": помощник}, ] } если __name__ == "__main__": df = pd.read_csv("gptdata_final.csv",coding='cp1252') печать (дф) с open("train.jsonl", "w") как f: для _ введите строку в df.iterrows(): example_str = json.dumps(create_dataset(row["user"], row["assistant"])) f.write(example_str + "\n") И когда я пытаюсь запустить его, я получаю эту ошибку при вызове функции json.dumps() со строками

Traceback (самый последний вызов — последний): Файл «/Users/user/Downloads/new/conv.py», строка 20, в example_str = json.dumps(create_dataset(row["user"], row["assistant"])) Файл «/Users/user/.pyenv/versions/3.10.0/lib/python3.10/site-packages/pandas/core/series.py», строка 1040, в __getitem__ вернуть self._get_value(ключ) Файл «/Users/user/.pyenv/versions/3.10.0/lib/python3.10/site-packages/pandas/core/series.py», строка 1156, в _get_value loc = self.index.get_loc(метка) Файл «/Users/user/.pyenv/versions/3.10.0/lib/python3.10/site-packages/pandas/core/indexes/base.py», строка 3797, в get_loc поднять KeyError(ключ) из ошибки KeyError: «пользователь» Я не понимаю, почему возникает ошибка ключа, хотя столбец «пользователь» существует.
Ответить

Быстрый ответ

Изменение регистра текста: 
Смайлики
:) :( :oops: :roll: :wink: :muza: :clever: :sorry: :angel: :read: *x)
Ещё смайлики…
   
К этому ответу прикреплено по крайней мере одно вложение.

Если вы не хотите добавлять вложения, оставьте поля пустыми.

Максимально разрешённый размер вложения: 15 МБ.

Вернуться в «Python»