mxnet.gluon.rnn及mxnet.symbol实现LSTM教程

基于mxnet.symbol的基本使用以及模型加载与保存
mxnet基本使用以及模型加载与保存
mxnet.symbolRNN-GRU-LSTM-Bi官网教程
基于mxnet的LSTM实现(mx.rnn.LSTMCell+symbol)

基于mxnet.gluon.rnn的基本使用以及模型加载与保存
LSTM Mxnet Implementation-手写
mxnet.gluon.rnn.LSTM中文教程
mxnet.gluon.rnn.GRU官网教程

模型导出

python 复制代码
import mxnet as mx
hidden_size = 128
num_layer = 2
word_emb_dim = 128
seq_len = 64
layer = mx.gluon.rnn.GRU(hidden_size, num_layer, bidirectional=True, )
layer.initialize()
# seq_len = 5 batch_size = 3 input_size = 128
input = mx.nd.random.uniform(shape=(seq_len, 3, word_emb_dim))
# by default zeros are used as begin state
output = layer(input)
# manually specify begin state.
# num_layers = 6 batch_size = 3 num_hidden = 128
h0 = mx.nd.random.uniform(shape=(num_layer*2, 3, hidden_size))
output, hn = layer(input, h0)
# 模型保存
layer.export('model')
# 模型加载
sym = mx.sym.load('model-symbol.json')
mod = mx.mod.Module(symbol=sym, context=mx.cpu(), label_names=None)
mod.bind(data_shapes=[('data', (5, 3, hidden_size))])
mod.load_params('model-0000.params')
print('Load Successfully!')

MXNet/Gluon:网络和参数的存取方式

相关推荐
机器学习之心3 小时前
一区北方苍鹰算法优化+创新改进Transformer!NGO-Transformer-LSTM多变量回归预测
算法·lstm·transformer·北方苍鹰算法优化·多变量回归预测·ngo-transformer
只怕自己不够好4 小时前
RNN与LSTM,通过Tensorflow在手写体识别上实战
rnn·tensorflow·lstm
cv小白菜1 天前
多算法模型(BI-LSTM GRU Mamba ekan xgboost)实现功率预测
机器学习·gru·lstm·时间序列·功率预测
这该死迷人的编程3 天前
深度学习:循环神经网络的计算复杂度,顺序操作和最大路径长度
人工智能·rnn·深度学习
余炜yw4 天前
【循环神经网络】:教AI写周杰伦风格的歌词
人工智能·rnn·深度学习
拓端研究室TRL5 天前
MATLAB用CNN-LSTM神经网络的语音情感分类深度学习研究
深度学习·神经网络·matlab·cnn·lstm
宝贝儿好6 天前
【NLP】第二章:RNN 循环神经网络
人工智能·pytorch·rnn·深度学习·神经网络·语言模型·自然语言处理
Hello.Reader6 天前
循环神经网络(RNN)全面解析
人工智能·rnn·深度学习
一去不复返的通信er6 天前
LSTM(长短期记忆网络)详解
人工智能·rnn·深度学习·神经网络·lstm
小叮当爱咖啡7 天前
RNN深度学习案例:LSTM火灾温度预测
rnn·深度学习·lstm