此模型完全基于keras-transformer,只是把别人的封装拿来简单实践一下
此外,可以和我另一个基于LSTM的机器翻译项目MachineTranslation-LSTM对比学习,二者原始数据一致
程序 | 版本 |
---|---|
python | 3.68 |
tensorflow | 1.13.1 |
Keras | 2.2.4 |
windows10 | |
jupyter |
(注:中文正常输入无需空格)
输入中文 | 输出英文 |
---|---|
我喜欢你! | I like you ! |
你喜欢我吗? | Do you like me ? |
滚蛋! | 听不懂呢。 |
(注:数据量较小可能遇到词典中没有的词,我简单的采用安全回复“听不懂呢。”来应对)
- 数据预处理
get_data
- 模型训练与预测
train&translate
- 运行
train&translate
- 加载
model/W-- 40-0.0563-.h5