手把手 | 用Python语言模型和LSTM做一个Drake饶舌歌词生成器

简介:

未来AI的主要应用是在建立能够学习数据然后生成原创内容的网络。这个想法已经充分应用于在自然语言处理(NLP)领域,这也是AI社区能够搭建出所谓语言模型的原因:语言模型的前提是学习句子在文章段落中的组成结构,从而生成新的内容。

在这篇文章中,我想尝试生成与很受欢迎的加拿大说唱歌手Drake(a.k.a. #6god)风格类似的说唱歌词,这肯定是件很有趣的事儿。

另外,我还想分享一下常规的机器学习项目渠道,因为我发现很多同学想做一些小项目,但不知道该从何处入手。

获取数据

首先,我们开始搜集Drake的曲库,为了节省时间我直接写了个爬虫,从网页metrolyrics.com抓取歌词。

import urllib.request as urllib2

from bs4 import BeautifulSoup


import pandas as pd


import re


from unidecode import unidecode





quote_page = 'http://metrolyrics.com/{}-lyrics-drake.html'


filename = 'drake-songs.csv'


songs = pd.read_csv(filename)


for index, row in songs.iterrows():


 page = urllib2.urlopen(quote_page.format(row['song']))


 soup = BeautifulSoup(page, 'html.parser')


 verses = soup.find_all('p', attrs={'class': 'verse'})


 lyrics = ''


 for verse in verses:

 text = verse.text.strip()

 text = re.sub(r"\[.*\]\n", "", unidecode(text))

 if lyrics == '':

 lyrics = lyrics + text.replace('\n', '|-|')


 else:


 lyrics = lyrics + '|-|' + text.replace('\n', '|-|')


 songs.at[index, 'lyrics'] = lyrics


 print('saving {}'.format(row['song']))


 songs.head()


print('writing to .csv')


songs.to_csv(filename, sep=',', encoding='utf-8')

我用了一个大家都很熟悉的Python包BeautifulSoup来抓取网页,这里参考了一位大牛Justin Yek的教程,我只花了五分钟就学会了使用。说明一下,上面的代码中我在循环里使用了songs这一数据格式,是因为我事先定义了想获得的歌曲。

教程:

https://medium.freecodecamp.org/how-to-scrape-websites-with-python-and-beautifulsoup-5946935d93fe

3d640b23467330d47b960083acaf2ef63e7d7c84

用DataFrame存储了所有的歌曲歌词

运行爬虫之后,我就得到了以合适的结构存储歌词的csv文件,下一步开始对数据进行预处理并且搭建模型。

模型介绍

现在我们来看看模型是如何生成文本的,这部分你要着重理解,因为这是真正的干货。我将先从模型设计和生成歌词模型中的关键组成部分讲起,然后,我们就可以直接进入实施阶段。

搭建语言模型主要有两种方法:

1.字符级(Character-leve)模型,

2.词汇级(Word-level)模型。

这两者的主要区别在于模型的输入和输出,接下来就具体讲解一下两个模型的工作原理。

字符级模型

在字符级模型中,输入是一连串的字符seed(种子),模型负责预测下一个字符,然后用seed + new_char组合来生成再下一个字符,以此类推。注意,因为我们每次输入的长度应保持一致,所以实际上在每次迭代输入时都要丢掉一个字符。我们可以看一个简单的直观的例子:

b96bb6762c47f799db43aa7c86acc93176ab454c

字符级模型生成词的迭代过程

每次迭代时,模型都是在给定种子字符的基础上预测下一个最可能生成的字符,或者利用条件概率,即找到概率P(new_char|seed)的最大值,其中new_char是字母表中的任一字母。

在此例中,字符表指所有英文字母和间隔符号的集合。(说明,字母表可以根据你的需要包含不同的字母,主要取决于你生成的语言种类)。

词汇级模型

词汇级模型和字符级模型非常相似,但是它用来生成下一个单词而非字符。这里举一个简单的例子来说明这一点:

1daef68b0d861ecf653f5b5f72f0a0306ec4d4da

图3. 词汇级模型生成词汇的迭代过程

现在在这个模型中,我们以一个词汇为单位向前寻找下一个词汇,而非字符。因此,我们想找到概率P(new_word|seed)的最大值,其中new_word是任一词汇。

这里要注意的是,这里我们搜索的范围比字符级要大得多。字符集模型中,我们只需从字符表中查找大概30个字符,但词汇级中每次迭代搜索的范围远远大于这个数量,因此每次迭代的运行速度更慢,但既然我们生成的是一整个词而不只是一个字符,所以也不算太糟糕。

关于词汇级模型,我最后想说明一点,我们可以通过在数据集中搜索独特的词汇来生成更加多样的词汇(这一步通常在数据预处理阶段进行)。由于词汇量可以无限大,我们其实有很多提高生成词汇性能的算法,比如词嵌入,不过关于这个问题可以再写一篇文章了。

这篇文章主要关注字符级模型,因为它更易于实施和理解,也更容易转化为复杂的词汇级模型。

数据预处理

针对字符级模型,我们将按照以下步骤进行数据预处理:

1.标记字符

对字符级模型而言,输入应该是基于字符而非字符串的形式。所以,我们首先要将歌词的每一行转变成字符的集合。

2.定义字符表

上一步,我们获得了歌词中所有可能出现的字符,接下来需要找出所有独特的字符。由于整个数据集并不大(只有140首歌),简单起见,我只保留所有英文字母以及一些特殊符号(比如空格),而忽略数字和其他的信息(因为数据集很小,我宁愿让模型少预测一些字符)。

3.创建训练序列

这里我们会用到滑动窗口的概念。通过沿着句子拖动一个固定长度的窗口,我们将建立用于训练的数据序列。下面的这张图很好地展示了滑动窗口的操作:

4f298bbf5cb2d7fe9e7cad69df9bffe432fc9f1c

图4. 用滑动窗口获得输入/输出

我们通过每次平移一个字符,得到相应长度为20个字符的模型输入和长度为1个字符的模型输出。每次只平移一格的额外好处就是大大扩展了数据集的大小。

4.标注编码训练序列

最后,我们不想直接处理原始字符(尽管理论上讲每个字符都是一个数字,所以你也可以说ASCII码已经帮我们为每个字符完成了编码)。我们要做的是用唯一的数字和每个字符一一对应,这一步就是所谓的标签编码。同时,我们要建立两个非常重要的映射:character-to-index (字符到索引)和index-to-character(索引到字符)。有了这两个映射,我们就能将字母表中任意的字符编码成对应的数字,同理,也能将模型输出的数字索引解码获得相应的字符。

5.数据集的独热编码

因为我们用的是分类数据,就是说所有字符都可以被归为某个类别,所以我们要将字符编码成输入列的形式。

当我们完成以上五个步骤以后,基本就大功告成了,接下来只需要搭建和训练模型。如果你想深入更多细节,以下是五个步骤的代码供参考。

ed2ea9e9deab7350896d765a5fcbb38a1f5122d7

3.建立模型

我们将用循环神经网络(RNN),更具体的说是长短期记忆网络(LSTM),基于前面出现的字符集来预测下一个字符。如果这两个概念都听着陌生的话,我也提供了相关概念的快速复习

RNN快速复习

通常,你看到的网络就是一个网状,从很多点汇聚到一个单点输出。如下图所示:

299ae118de005b20f0ed80a48678e1a458b9f399

图5. 神经网络示意图

这里的神经网络是单点输入,单点输出。它适用于输入是不连续的情况,因为输入的顺序不会影响到输出结果。但是在我们的案例中,输入字符的顺序是非常重要的,因为顺序决定了对应的单词。

而RNN可以接收连续的输入,同时将前一个节点的输出作为参数输入下一个节点,从而解决输入顺序的问题。

c7eee3ffa8dd6f4a39ec2f1abab3d758c6ad3d54

图6. 简易RNN示意图

例如,基于序列Tryna_keep_it_simple,提取的下一个字符就应该是 _。这正是我们想让神经网络做的。神经网络的输入将是 T — > x<1>, r -> x<2>, n -> x<3>... e-> x<n> ,对应输出字符就是一个空格 y -> _ 。

LSTM快速复习

简单的RNN网络仍存在一些问题,它不善于将非常前端的元胞信息传递到后端元胞。例如,句子Tryna keep it simple is a struggle for me中最后一个词me,如果不往回看前面出现了什么单词,那么这个单词是很难预测准确的(很可能就被预测成了Baka,cat,potato之类)。

而LSTM能够很好地解决这个问题,它在每个元胞中存储部分前面发生的事件信息(即前面出现的单词)。如下图所示:

17192f433a3c9eecf2778529319b8877d38c8f69

图7. LSTM示意图,摘自Andrew Ng的深度学习课程

不仅传递前一个元胞的输出a<n> ,同时包含之前元胞输入信息的c<n> 也作为了下一个元胞的输入的一部分。这使得LSTM能够更好地保留上下文的信息,并适用于语言建模的预测。

编程建模

我之前学过一点Keras,所以这次就以Keras为框架编程搭建模型。其实也可以选择自己搭建模型框架,但这样会花费更多的时间。

# create sequential network, because we are passing activations


# down the network


model = Sequential()

# add LSTM layer

model.add(LSTM(128, input_shape=(maxlen, len(chars))))

# add Softmax layer to output one character

model.add(Dense(len(chars)))

model.add(Activation('softmax'))

# compile the model and pick the loss and optimizer

model.compile(loss='categorical_crossentropy', optimizer=RMSprop(lr=0.01))

# train the model

model.fit(x, y, batch_size=128, epochs=30)

以上可见,我们搭建了LSTM模型并且使用了批处理,利用数据子集分批进行训练而不是一次输入所有的数据,这样可以稍微提高一点训练的速度。

4、生成歌词

训练完模型,接下来介绍如何生成下一个字符。我们首先要用用户输入的简单字符串作为随机种子。接着,我们把种子作为网络的输入来预测下一个字符,重复这个过程直到我们生成了一些新的歌词,类似于图2中所示。

以下是一些生成的歌词的例子。

注意:这些歌词都没被审核过,阅读时请自行甄别。

29c08a3ba25852239da5d56a2cf21470e1455488

你可能会注意到,生成的单词有的是没有意义的,这是字符级模型的一个常见问题。这是因为输入序列经常在单词的中间被切断,使得神经网络模型学习并生成对其输入而言是有意义,但是我们看来很奇怪的新单词。

这也是在词汇级模型中可以解决的问题,但是对于仅以200行代码建立的模型来说,字符级模型所达到的效果仍然令人印象深刻。

其他应用

在这里演示的字符级模型的歌词预测功能可以被扩展到其他更有用的应用上。

例如,可以利用相同的原理对iPhone键盘上要输入的下一个单词进行预测。

81434b7395c6a5806aa60376bd719a1d0111e1d6

图8. 键盘输入预测下一个单词

想象假如你建立一个高准确度的Python语言模型,不但能够自动填充关键词或者变量名,还可以填充大段的代码,这将帮助码农们节省多少时间啊!

你可能也注意到了文中的代码并不完整,遗漏了一些片段。请到我的Github获取更多的细节,学习搭建你自己的项目模型。


原文发布时间为:2018-06-14

本文作者:文摘菌

本文来自云栖社区合作伙伴“大数据文摘”,了解相关信息可以关注“大数据文摘”。

相关文章
|
1天前
|
存储 程序员 Python
深入理解Python中的生成器和迭代器
本文将深入探讨Python中生成器和迭代器的概念、原理以及它们在编程中的应用。通过详细解析生成器和迭代器的工作机制,读者将能够更好地理解Python中的迭代器协议、生成器函数以及生成器表达式的使用方法。同时,本文还将介绍生成器和迭代器在Python中的一些常见应用场景,并提供一些实用的编程技巧和最佳实践。
|
2天前
|
机器学习/深度学习 存储 数据可视化
【视频】LSTM模型原理及其进行股票收盘价的时间序列预测讲解|附数据代码2
【视频】LSTM模型原理及其进行股票收盘价的时间序列预测讲解|附数据代码
|
2天前
|
机器学习/深度学习 数据可视化 TensorFlow
【视频】LSTM模型原理及其进行股票收盘价的时间序列预测讲解|附数据代码1
【视频】LSTM模型原理及其进行股票收盘价的时间序列预测讲解|附数据代码
|
2天前
|
机器学习/深度学习 自然语言处理 算法
Python遗传算法GA对长短期记忆LSTM深度学习模型超参数调优分析司机数据|附数据代码
Python遗传算法GA对长短期记忆LSTM深度学习模型超参数调优分析司机数据|附数据代码
|
8天前
|
机器学习/深度学习 数据采集 前端开发
【Python机器学习专栏】模型泛化能力与交叉验证
【4月更文挑战第30天】本文探讨了机器学习中模型泛化能力的重要性,它是衡量模型对未知数据预测能力的关键。过拟合和欠拟合影响泛化能力,而交叉验证是评估和提升泛化能力的有效工具。通过K折交叉验证等方法,可以发现并优化模型,如调整参数、选择合适模型、数据预处理、特征选择和集成学习。Python中可利用scikit-learn的cross_val_score函数进行交叉验证。
|
8天前
|
机器学习/深度学习 数据可视化 前端开发
【Python机器学习专栏】机器学习模型评估的实用方法
【4月更文挑战第30天】本文介绍了机器学习模型评估的关键方法,包括评估指标(如准确率、精确率、召回率、F1分数、MSE、RMSE、MAE及ROC曲线)和交叉验证技术(如K折交叉验证、留一交叉验证、自助法)。混淆矩阵提供了一种可视化分类模型性能的方式,而Python的scikit-learn库则方便实现这些评估。选择适合的指标和验证方法能有效优化模型性能。
|
8天前
|
机器学习/深度学习 算法 前端开发
【Python机器学习专栏】机器学习中的模型融合技术
【4月更文挑战第30天】模型融合,即集成学习,通过结合多个模型提升预测性能。常见方法包括:Bagging(如Random Forest)、Boosting(如AdaBoost、XGBoost)和Stacking。Python中可使用`scikit-learn`实现,例如BaggingClassifier示例。模型融合是机器学习中的强大工具,能提高整体性能并适应复杂问题。
|
8天前
|
机器学习/深度学习 Python
【Python 机器学习专栏】模型选择中的交叉验证与网格搜索
【4月更文挑战第30天】交叉验证和网格搜索是机器学习中优化模型的关键技术。交叉验证通过划分数据集进行多次评估,如K折和留一法,确保模型性能的稳定性。网格搜索遍历预定义参数组合,寻找最佳参数设置。两者结合能全面评估模型并避免过拟合。Python中可使用`sklearn`库实现这一过程,但需注意计算成本、过拟合风险及数据适应性。理解并熟练应用这些方法能提升模型性能和泛化能力。
|
8天前
|
机器学习/深度学习 数据可视化 TensorFlow
【Python 机器学习专栏】使用 TensorFlow 构建深度学习模型
【4月更文挑战第30天】本文介绍了如何使用 TensorFlow 构建深度学习模型。TensorFlow 是谷歌的开源深度学习框架,具备强大计算能力和灵活编程接口。构建模型涉及数据准备、模型定义、选择损失函数和优化器、训练、评估及模型保存部署。文中以全连接神经网络为例,展示了从数据预处理到模型训练和评估的完整流程。此外,还提到了 TensorFlow 的自动微分、模型可视化和分布式训练等高级特性。通过本文,读者可掌握 TensorFlow 基本用法,为构建高效深度学习模型打下基础。
|
8天前
|
机器学习/深度学习 PyTorch TensorFlow
【Python机器学习专栏】循环神经网络(RNN)与LSTM详解
【4月更文挑战第30天】本文探讨了处理序列数据的关键模型——循环神经网络(RNN)及其优化版长短期记忆网络(LSTM)。RNN利用循环结构处理序列依赖,但遭遇梯度消失/爆炸问题。LSTM通过门控机制解决了这一问题,有效捕捉长距离依赖。在Python中,可使用深度学习框架如PyTorch实现LSTM。示例代码展示了如何定义和初始化一个简单的LSTM网络结构,强调了RNN和LSTM在序列任务中的应用价值。