javascript - Brain js NaN 当预测是单词组合是一个完整的句子时

Question

我是 Brain js 的新手，我试图让它预测单词的组合是否是一个完整的句子。 https://codepen.io/AtanasBobev/pen/zQzZrP?editors=0010

    const config = {
    hiddenLayers: [3]
};

const net = new brain.NeuralNetwork(config);

net.train([{input: ["Hello, I'm John Walker."], output: [1]},
           {input: ["This is on you!"], output: [1]},
           {input: ["Who are you?"], output: [1]},
             {input: ["Let's go."], output: [1]},
           {input: ["John kik"], output: [0]},
           {input: ["This is"], output: [0]}
          ]);

const output = net.run(["I'm Stil."]);  

alert(output);
//Output: NaN

我知道数据不足以进行良好的预测，但我仍然期望一个介于 0-1 之间的值。可能是什么问题？

Answer 1

这可以通过 Brain 的 LSTM 函数来完成。这是它在您的数据集上的运行情况(还有一些要训练的示例):

const net = new brain.recurrent.LSTM();

net.train([
  { input: "Hello, I'm John Walker.", output: "complete" },
  { input: "This is on you!", output: "complete" },
  { input: "John kik", output: "incomplete" },
  { input: "This is", output: "incomplete" },
  { input: "Great job.", output: "complete" },
  { input: "When I hear a", output: "incomplete" }
]);

输出:

> net.run("I'm Stil.");
"incomplete"
> net.run("Great job!")
"complete"

LSTM docs