《动手学深度学习》循环神经网络
pytorch版循环神经网络实现 import torch import torch . nn as nn import time import math import sys def load_data_jay_lyrics ( ) : """加载周杰伦歌词数据集""" with open ( '/Users/nick/Documents/dataset/jaychou_lyrics.txt' , 'r' ) as f : corpus_chars = f . read ( ) corpus_chars = corpus_chars . replace ( '\n' , ' ' ) . replace ( '\r' , ' ' ) corpus_chars = corpus_chars [ 0 : 10000 ] idx_to_char = list ( set ( corpus_chars ) ) char_to_idx = dict ( [ ( char , i ) for i , char in enumerate ( idx_to_char ) ] ) vocab_size = len ( char_to_idx ) corpus_indices = [ char_to_idx [ char ] for char in corpus_chars ] return