深度排序
学习使用seq2seq模型对数字进行排序。
运行这段代码
调用pip install -r requirements.txt安装所有依赖项。
产生资料
可以使用所有数据
样品电话
python generate.py \
--name="train" \
--size=10000 \
--max_val=256 \
--min_length=2 \
--max_length=256 \
训练
可以通过在设置适当的参数,然后将train.run()设置为在调用,最后一次调用python main.py (是的,我很抱歉,对于未配置命令行参数)。
从上面的示例调用生成的数据集中训练了1个纪元,大约花费了10分钟。
评估
在train.txt上训练模型后,使用生成测试集( name="test" ),然后以与上所述相同的方式运行 ,以查看该模型的一些示例评估。
再
1