ChatGPT插件例子 编写 Python 文档字符串
如何为给定的 Python 函数创建文档字符串的示例。我们指定 Python 版本,粘贴代码,然后在注释中询问文档字符串,并给出文档字符串的特征开头 (""")。
提问
# Python 3.7
def randomly_split_dataset(folder, filename, split_ratio=[0.8, 0.2]):
df = pd.read_json(folder + filename, lines=True)
train_name, test_name = “train.jsonl”, “test.jsonl”
df_train, df_test = train_test_split(df, test_size=split_ratio[1], random_state=42)
df_train.to_json(folder + train_name, orient='records', lines=True)
df_test.to_json(folder + test_name, orient='records', 线=True)
randomly_split_dataset('finetune_data/', 'dataset.jsonl')
# 上述函数的精心制作的高质量文档字符串:
"""
回答
""" 该函数根据给定的分割比例,将一个数据集随机分割成训练集和测试集两部分。
参数:
folder(str):包含该数据集的文件夹的路径。
filename(str):数据集文件的名称。
split_ratio(list):两个浮点数的列表,表示训练集和测试集的比例。
返回:无
"""
API请求
import os
import openai
openai.api_key = os.getenv("OPENAI_API_KEY")
response = openai.Completion.create(
model="text-davinci-003",
prompt="# Python 3.7\n \ndef randomly_split_dataset(folder, filename, split_ratio=[0.8, 0.2]):\n df = pd.read_json(folder + filename, lines=True)\n train_name, test_name = \"train.jsonl\", \"test.jsonl\"\n df_train, df_test = train_test_split(df, test_size=split_ratio[1], random_state=42)\n df_train.to_json(folder + train_name, orient='records', lines=True)\n df_test.to_json(folder + test_name, orient='records', lines=True)\nrandomly_split_dataset('finetune_data/', 'dataset.jsonl')\n \n# An elaborate, high quality docstring for the above function:\n\"\"\"",
temperature=0,
max_tokens=150,
top_p=1.0,
frequency_penalty=0.0,
presence_penalty=0.0,
stop=["#", "\"\"\""]
)