release: 完成了mvp,具有基本的模型训练,语料生成和推理功能
This commit is contained in:
25
prompt/base.py
Normal file
25
prompt/base.py
Normal file
@@ -0,0 +1,25 @@
|
||||
class create_dataset:
|
||||
"""用于生成微调数据集模板的类"""
|
||||
|
||||
template = """
|
||||
项目名为:{}
|
||||
请依据以下该项目官方文档的部分内容,创造合适的对话数据集用于微调一个了解该项目的小模型的语料,要求兼顾文档中间尽可能多的信息点,使用中文
|
||||
文档节选:{}
|
||||
按照如下json格式返回:{}
|
||||
"""
|
||||
|
||||
@staticmethod
|
||||
def create(*args: any) -> str:
|
||||
"""根据提供的任意数量参数生成数据集模板
|
||||
|
||||
Args:
|
||||
*args: 任意数量的参数,将按顺序填充到模板中
|
||||
|
||||
Returns:
|
||||
格式化后的模板字符串
|
||||
"""
|
||||
return create_dataset.template.format(*args)
|
||||
|
||||
|
||||
if __name__=="__main__":
|
||||
print(create_dataset.create("a", "b", "c"))
|
Reference in New Issue
Block a user