【发布时间】:2020-01-06 16:05:51
【问题描述】:
免责声明: 我是机器学习的新手。如果我问了一个超级愚蠢的问题,请原谅我。
您好,我正在尝试使用 Brain.js 在一组数字中查找模式。
我已经扩展了 Brain.js github 页面中的示例。
const net = new brain.recurrent.LSTMTimeStep({
inputSize: 3,
hiddenLayers: [10],
outputSize: 3
});
net.train([
[1, 1, 1],
[2, 2, 2],
[3, 3, 3],
[4, 4, 4],
[5, 5, 5],
[6, 6, 6],
[7, 7, 7],
[8, 8, 8]
]);
const output = net.run([[7, 7, 7], [8, 8, 8]]);
我试图得到 [9, 9, 9] 的输出,但我主要得到 [8, 8, 8]。
但是,如果我尝试运行 const output = net.run([[5, 5, 5], [6, 6, 6]]);,我很容易得到 [7, 7, 7] 和训练数据序列中其他数字的连续输出。
是否有办法对其进行训练,以便获得所需的输出并将其用于其他模式?
【问题讨论】:
-
你应该将你的数字转换为 0 和 1 之间。其次,一个简单的神经网络就可以了。
标签: javascript machine-learning lstm brain.js