微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

为什么语言翻译 ML 模型不起作用?

如何解决为什么语言翻译 ML 模型不起作用?

我使用 TensorFlow 函数式 API 创建了一个语言翻译模型。

这是我用过的编码器-解码器模型。

# encoder 
encoder = tf.keras.Input(shape=(None,))
enc_embd = tf.keras.layers.Embedding(vocab_train,embedding_dim)(encoder)

encoder_gru = tf.keras.layers.GRU(units,return_sequences=True,return_state=True,kernel_regularizer=tf.keras.regularizers.L2(0.001))
output_e,hidden_e = encoder_gru(enc_embd)

# decoder
decoder = tf.keras.Input(shape=(None,))
dec_embd = tf.keras.layers.Embedding(vocab_label,embedding_dim)(decoder)

decoder_gru = tf.keras.layers.GRU(units,kernel_regularizer=tf.keras.regularizers.L2(0.001))
output_d,hidden_d = decoder_gru(dec_embd,initial_state = hidden_e)

final_output = tf.keras.layers.Dense(vocab_label,activation='softmax',kernel_regularizer=tf.keras.regularizers.L2(0.001))
output_f = final_output(output_d)

为了一一翻译单词,我自定义了预测模型。

# customize for prediction
encoder_model = tf.keras.Model(encoder,hidden_e) 

state = tf.keras.Input(shape=(None,))
output,hidden_dest = decoder_gru(dec_embd,initial_state = state)
output_result = final_output(output)

decoder_model = tf.keras.Model([decoder,state],[hidden_dest,output_result])

在训练期间,TensorFlow 准确度指标显示准确度有所增加,但预测不准确。这是完整的代码,我已经注释掉了每一行,以便您轻松理解。请看一下。

https://www.kaggle.com/devanshusingh/tensorflow

在这代码中,我认为调用函数有问题,请帮我找出问题所在。

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。