文本生成.py

# -*- coding: utf-8 -*-
"""文本生成.ipynb

Automatically generated by Colaboratory.

Original file is located at
    https://colab.research.google.com/drive/1JFyRrdUS6eRqL3h2pLPCCUP3_iBh7hkV
"""

import matplotlib as mpl
import matplotlib.pyplot as plt
import numpy as np
import pandas as pd
import tensorflow as tf
from tensorflow import keras
import sklearn 
import os
import sys
import time

print(tf.__version__)
print(sys.version_info)
for module in mpl,np,pd,sklearn,tf,keras:
  print(module.__name__,module.__version__)

# !wget https://storage.googleapis.com/download.tensorflow.org/data/shakespeare.txt

# 莎士比亚文集路径
input_filepath="/content/shakespeare.txt"
# 读取文集
text=open(input_filepath,'r').read()

print(len(text))
print(type(text))
print(text[0:100])

"""100W+个数据集对于NN训练是很小的"""

# 对文集去重并排序
vocab=sorted(set(text))
print(len(vocab))
print(type(vocab))
print(vocab)

"""char-level：区分字母大小写，各种符号"""

# 给字符编号char:idx
char2idx={char:idx for idx,char in enumerate(vocab)}
print(type(char2idx))
print(char2idx)

# idx:char-vocab的下标就是天然的索引
# np的array速度比python的list快
idx2char=np.array(vocab)
print(type(idx2char))
print(idx2char)

# 用{字符：编号}把文集中的字符都转成数字编号
text_as_int=np.array([char2idx[c] for c in text])
print(len(text_as_int))
print(type(text_as_int))

print(text[0:10])
print(text_as_int[0:10])

"""文本First Citi对应的字符索引是[18 47 56 57 58  1 15 47 58 47]

莎士比亚文集:char-level
"""

# 对文本划分好输入、输出 
# e.g.对文本abcde划分得到输入abcd,输出bcde
def split_input_target(id_text):
  return id_text[0:-1],id_text[1:]

# 把id text作为数据集
char_dataset=tf.data.Dataset.from_tensor_slices(text_as_int)
# 给100个序列，则划分成100个输入，100个输出
seq_length=100
# 分批处理，得到多组输入、输出
seq_dataset=char_dataset.batch(seq_length+1,drop_remainder=True)

# 拿出来两个字符，看看是什么具体字符
for ch_id in char_dataset.take(2):
  print(ch_id,idx2char[ch_id.numpy()])

# 拿出来两个句子，看看是什么具体句子
for seq_id in seq_dataset.take(2):
  print(seq_id)
  print(repr(''.join(idx2char[seq_id.numpy()])))
  print(repr(' '.join(idx2char[seq_id.numpy()])))

"""seq_id是用char-level表示的
字符可以预测出空格，就作为单词的分隔了
"""

# 划分单词的输入、输出
# map实现调用函数
seq_dataset=seq_dataset.map(split_input_target)

# 拿出来两个序列，划分输入输出
for item_input,item_output in seq_dataset.take(1):
  print(item_input.numpy())
  print(item_output.numpy())

print(seq_dataset)

"""看例子，18输出47，47输出56，56输出57..."""

# batch处理
batch_size=64
# 洗牌
buffer_size=10000
# shuffle()-随机重新排列此数据集的元素-理解:如果数据集包含10,000个元素但buffer_size设置为1,000个，则shuffle最初将仅从缓冲区中的前1,000个元素中选择一个随机元素。选择一个元素后，其缓冲区中的空间将被下一个（即1,001个）元素替换，并保留1,000个元素缓冲区。
seq_dataset=seq_dataset.shuffle(buffer_size).batch(batch_size,drop_remainder=True)
print(seq_dataset)

vocab_size=len(vocab)
# 资料小，就升维
embedding_dim=256
rnn_units=1024

# 搭建模型写成一个函数
def build_model(vocab_size,embedding_dim,rnn_units,batch_size):
      model=keras.models.Sequential([
                   keras.layers.Embedding(vocab_size,embedding_dim,batch_input_shape=[batch_size,None]),
                  #  stateful=True 是否要把最后返回的状态添加到输出
                  # recurrent_initializer='glorot_uniform'-RNN的权值初始值，均值为0，以0为中心的对称区间均匀分布的随机数
                   keras.layers.SimpleRNN(units=rnn_units,stateful=True,recurrent_initializer='glorot_uniform',return_sequences=True),
                  #  让输出是65个vocab_size的一个
                   keras.layers.Dense(vocab_size),
      ])
      return model

model=build_model(vocab_size=vocab_size,embedding_dim=embedding_dim,rnn_units=rnn_units,batch_size=batch_size)
model.summary()

model.variables

# 从序列中拿出一个take(1)看一看模型对输入处理完的输出，简单验证能否做预测
for input_example_batch,target_example_batch in seq_dataset.take(1):
    # 把model当成函数来用了，其实是调用了call方法-使实例（对象）像函数一样被调用
    example_batch_predictions=model(input_example_batch)
    print(example_batch_predictions.shape)

print(example_batch_predictions)

"""浮点数理解-softmax输出就是概率值，再对应到具体的一类中，这里就是字符类"""

# 输入是100个char,输出也是100个char
# categorical()从分类分布中抽取样本
# logits:2-D Tensor with shape [batch_size, num_classes]，这里是[100,1]的2维张量
sample_indices=tf.random.categorical(logits=example_batch_predictions[0],num_samples=1)
print(sample_indices)
# sqeeze:转为100的向量
sample_indices=tf.squeeze(sample_indices,axis=-1)
print(sample_indices)

# 看一组输入、输出
print("Input:",repr("".join(idx2char[input_example_batch[0]])))
print()
print("Output:",repr("".join(idx2char[target_example_batch[0]])))
print()
print("Predictions:",repr("".join(idx2char[sample_indices])))

"""预测效果不咋地"""

def loss(labels,logits):
  return keras.losses.sparse_categorical_crossentropy(labels,logits,from_logits=True)

model.compile(optimizer='adam',loss=loss)
# 真实标签和预测标签的区别
example_loss=loss(target_example_batch,example_batch_predictions)
print(example_loss.shape)
print(example_loss.numpy().mean())

# 保存模型-经过训练之后的模型效果会比上边那个没有经过训练的效果要好

# 定义一个文件夹来保存模型
output_dir="/content/text_generation_checkpoints"
if not os.path.exists(output_dir):
  os.mkdir(output_dir)

# 保存最后一次迭代的模型
checkpoint_prefix=os.path.join(output_dir,'ckpt_{epoch}')
checkpoint_callback=keras.callbacks.ModelCheckpoint(
                                             filepath=checkpoint_prefix,
                                             #  只保存权重值
                                             save_weights_only=True
                                                  )

epochs=100
history=model.fit(seq_dataset,epochs=epochs,callbacks=[checkpoint_callback])

# 看看最好的模型
tf.train.latest_checkpoint(output_dir)

# 再创建一个新模型，去加载已经保存过的权重值
output_dir="/content/text_generation_checkpoints"
model2=build_model(vocab_size,embedding_dim,rnn_units,batch_size=1)
model2.load_weights(tf.train.latest_checkpoint(output_dir))

# 1个样本，None表示变长序列
model2.build(tf.TensorShape([1,None]))
model2.summary()

# 实现文本生成
def generate_text(model,start_string,num_generate=1000):
  # 输入
  input_eval=[char2idx[ch] for ch in start_string]
  print(input_eval)
  # 升维-在axis=0方向上
  input_eval=tf.expand_dims(input_eval,0)
  print(input_eval)

  text_generated=[]
  # 连续调用模型用reset_states()
  model.reset_states()
  # 逐个预测输出字符
  for _ in range(num_generate):
  
      predictions=model(input_eval)
      # squeeze降维：消掉batch_size维度，变成predictions:[input_eval_len,vocab_size]
      predictions=tf.squeeze(predictions,0)
      print(predictions)
      # 倒序、抽取样本、降维成1维
      predicted_id=tf.random.categorical(predictions,num_samples=1)[-1,0].numpy()
      print(predicted_id)
      # 得到一个预测字符，append到生成文本中
      text_generated.append(idx2char[predicted_id])
      # 得到的预测再去作为下一次的输入字符
      input_eval=tf.expand_dims([predicted_id],0)
  return start_string+''.join(text_generated)


new_text=generate_text(model2,"amazing:")
print(new_text)