Seq2Seq-多项式展开
此存储库包含使用PyTorch和PyTorch Lightning进行seq2seq任务(即转换)的转换器的实现。
问题
实施一个学习扩展单个变量多项式的深度学习模型,其中该模型将分解序列作为输入并预测扩展序列。 例如:
n*(n-11)=n**2-11*n
n*(n-11)是分解输入
n**2-11*n是扩展目标
扩展的表达式是可交换的,但是只有提供的形式才被认为是正确的。
模型精度
针对预测目标序列和实地真实目标序列之间的严格字符串相等性对模型进行评估。 该模型的精度为0.86 。
指示
分为训练和测试集:
python data.py
训练模型:
python train.py
在测试集上评估模型:
python main.py
运行单元测试:
python -m unittest tests.py