neural-network - 使用brain.js神经网络做文本分析

Question

我正在尝试进行一些文本分析，以确定给定的字符串是否...谈论政治。我想我可以创建一个神经网络，其中输入是字符串或单词列表(排序可能很重要？)，输出是字符串是否与政治有关。

但是，brain.js 库只接受 0 到 1 之间的数字或 0 到 1 之间的数字数组的输入。如何以能够完成任务的方式强制我的数据？

Answer 1

new brain.recurrent.LSTM(); 

这对你有用。

例子，

var brain = require('brain.js')
var net = new brain.recurrent.LSTM();
net.train([
  {input: "my unit-tests failed.", output: "software"},
  {input: "tried the program, but it was buggy.", output: "software"},
  {input: "i need a new power supply.", output: "hardware"},
  {input: "the drive has a 2TB capacity.", output: "hardware"},
  {input: "unit-tests", output: "software"},
  {input: "program", output: "software"},
  {input: "power supply", output: "hardware"},
  {input: "drive", output: "hardware"},
]);

console.log("output = "+net.run("drive"));


output = hardware

请参阅此链接=> https://github.com/BrainJS/brain.js/issues/65
这对brain.recurrent.LSTM()有明确的解释和用法