给定许多序列预测一个序列

数据挖掘 机器学习 算法 顺序
2021-10-11 06:37:29

我正在尝试找到适合此用例的算法:

我的数据:一堆固定大小的整数数组,例如

[0,2,3,4,5]
[1,2,3,1,5]
[4,1,2,4,5]
...

输入:一个整数数组,输出:对数组其余部分的预测,例如

[0,1] -> [2,4,2]
[3] -> [1,3,2,5,5]
[3,2,4,1] -> [4]
2个回答

所以,问题是如何对以下问题建模:输入序列的一个子部分,然后输出序列的其余部分。这是一个序列到序列的问题。

为此,正如 Derek O 所逃避的那样,我建议使用编码器-解码器架构。在编码器中,您将对输入序列进行编码(使用 RNN/LSTM),这将为您提供“隐藏表示”,然后通过解码器(RNN/LSTM)传递您的隐藏表示,然后将隐藏表示解码为树脂序列整数。

这是一篇提供有关编码器-解码器模型的更多详细信息的文章:https ://towardsdatascience.com/understanding-encoder-decoder-sequence-to-sequence-model-679e04af4346

这取决于这些数字代表什么。

如果存在顺序/时间依赖性(即后续输出取决于先前的输入),我建议使用 LSTM-RNN。