语言: 中文
数据集:

• huggingartists/little-big
  标签:
• huggingartists
• 歌词
• 语言模型头部
• 因果语言模型

小部件示例:

• 文本: "我是"

本模型由huggingartists创建。

通过演示教程，为您喜爱的艺术家定制专属聊天机器人！

工作原理

了解模型开发过程，请查阅W&B报告。

训练数据

模型基于Little Big的歌词进行训练。

数据集可在此获取：点击访问
调用方式如下：

from datasets import load_dataset  

dataset = load_dataset("huggingartists/little-big")  

探索数据，所有流程均通过W&B artifacts进行追踪记录。

训练流程

模型基于预训练的GPT-2，并针对Little Big的歌词进行微调。

超参数和指标完整记录于W&B训练过程，确保完全透明可复现。

训练完成后，最终模型将被记录并版本化。

使用方法

直接使用文本生成管道调用模型：

from transformers import pipeline  
generator = pipeline('text-generation',  
                     model='huggingartists/little-big')  
generator("我是", num_return_sequences=5)  

或通过Transformers库调用：

from transformers import AutoTokenizer, AutoModelWithLMHead  

tokenizer = AutoTokenizer.from_pretrained("huggingartists/little-big")  

model = AutoModelWithLMHead.from_pretrained("huggingartists/little-big")  

局限性及偏差

本模型存在与GPT-2相同的局限性和偏差。

此外，用户推文数据会进一步影响模型生成文本的倾向性。

关于

由Aleksey Korshuk开发

更多详情请访问项目仓库。