【吴恩达团队Tensorflow2.0实践系列课程第三课】TensorFlow2.0中的自然语言处理
这个暑假计划学习NLP,看的是吴恩达老师的视频,已经看到第三课,大家有兴趣的可以看看之前的视频,我觉得讲得非常好。写博客是为了更好的学习,将其中一些细节写下来避免遗忘。
Tokenizer是一个用于向量化文本,或将文本转换为序列(即单词在字典中的下标构成的列表,从1算起)的类。
构造参数
类方法
属性
代码
import tensorflow as tf
from tensorflow import keras
from keras.preprocessing.text import Tokenizer
sentences = [
'I love my dog',
'I love my cat',
'You love my dog! '
]
tokenizer = Tokenizer(num_words = 100,
filters='!"#$%&()*+,-./:;<=>?@[]^_`{|}~tn',
lower=True,
split=' ',
oov_token='<00f>',
char_level=False)
tokenizer.fit_on_texts(sentences)
word_index = tokenizer.word_index
print(word_index)
输出:
{'love': 1, 'my': 2, 'i': 3, 'dog': 4, 'ca': 5, 'you': 6}
我的理解:
将文本中每个词进行编码,然后每个词就有唯一一个索引,这个文本就转换为了数字。
数据文件下载
口令:hygu9z
import numpy as np
import tensorflow as tf
import tensorflow.keras as keras
import os
origin_dir = 'E:\\黑马课程\\数据集'
train_dir = origin_dir + '\\aclImdb\\train'
test_dir = origin_dir + '\\aclImdb\\test'
texts = []
labels = []
for fname in os.listdir(train_dir+'\\neg'):
with open(train_dir+'\\neg\\'+fname,'r',encoding='utf8') as f:
texts.append(f.read())
labels.append(0)
for fname in os.listdir(train_dir+'\\pos'):
with open(train_dir+'\\pos\\'+fname,'r',encoding='utf8') as f:
texts.append(f.read())
labels.append(1)
from sklearn.model_selection import train_test_split
training_sentences,testing_sentences, training_labels,testing_labels = train_test_split(texts,labels, test_size=0.2)
分词与编码:用 keras.preprocessing.text.Tokenizer 匹配文本 texts. 再用 tokenizer 将文本列表转化为数字列表 sequences(列表中的每个元素都是由整数构成的列表)。
word_index 是将单词对应到整数的字典:{‘the’:1, ‘and’:2, ‘a’:3, … }
再用 keras.preprocessing.pad_sequences 将列表中每个元素都变成长为 200的整数列表。 返回值是 ndarray,形状为 (25000,200)
vocab_size = 10000 #每个评论加载的最大数据
max_length = 200
trunc_type='post'
oov_tok = ""
from tensorflow.keras.preprocessing.text import Tokenizer
from tensorflow.keras.preprocessing.sequence import pad_sequences
tokenizer = Tokenizer(num_words = vocab_size, oov_token=oov_tok)
tokenizer.fit_on_texts(training_sentences)
word_index = tokenizer.word_index
sequences = tokenizer.texts_to_sequences(training_sentences)
# 填充或者截断评论
padded = pad_sequences(sequences,maxlen=max_length, truncating=trunc_type)
testing_sequences = tokenizer.texts_to_sequences(testing_sentences)
testing_padded = pad_sequences(testing_sequences,maxlen=max_length)
# 将键值对换,数字标识在前
reverse_word_index = dict([(value, key) for (key, value) in word_index.items()])
# 查看处理后的数据和之前的数据
def decode_review(text):
return ' '.join([reverse_word_index.get(i, '?') for i in text])
简单地建立一个Sequential模型,由三层构成:Embedding,Flatten , Dense.
模型输入参数形状是 (batch_size, 200),表示由 batch_size 个长度200的整数向量构成。
Embedding层的设定参数是(10000,16),表示将10000个整数映射到10000个长度为16的向量。
Flatten将向量转换为一维
Dense全连接层用sigmoid激活函数输出一个值,将其二分类。
训练模型10个epoch,batch_size=32
embedding_dim = 16
model = tf.keras.Sequential([
tf.keras.layers.Embedding(vocab_size, embedding_dim, input_length=max_length),
tf.keras.layers.Flatten(),
tf.keras.layers.Dense(6, activation='relu'),
tf.keras.layers.Dense(1, activation='sigmoid')
])
model.compile(loss='binary_crossentropy',optimizer='adam',metrics=['accuracy'])
model.summary()
# 转换数据,放入模型训练
training_labels_final = np.array(training_labels)
testing_labels_final = np.array(testing_labels)
num_epochs = 10
model.fit(padded, training_labels_final, epochs=num_epochs, validation_data=(testing_padded, testing_labels_final))
import matplotlib.pyplot as plt
acc = history.history['acc']
val_acc = history.history['val_acc']
plt.plot(np.arange(1,len(acc)+1), acc)
plt.plot(np.arange(1,len(acc)+1), val_acc)
plt.legend(['acc','val_acc'])
plt.title('Accuracy (Non-pretrained embedding & LSTM)')
plt.figure()
loss = history.history['loss']
val_loss = history.history['val_loss']
plt.plot(np.arange(1,len(loss)+1), loss)
plt.plot(np.arange(1,len(val_loss)+1), val_loss)
plt.legend(['loss','val_loss'])
plt.title('Loss (Non-pretrained embedding & LSTM)')
plt.show()
预测测试集中前10个,如果需要预测新的文本,需要处理格式为向量
print(model.predict(testing_padded[:10]))