loading
index.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
const brain = require('brain.js'),
dataset = require('./dataset.js'); // Для удобства тренировочная база вынесена в отдельные файлы

const net = new brain.recurrent.LSTM(); // Создаем ИНС классификатора LSTM

net.train([...dataset], {log: true, logPeriod: 100, iterations: 1000}); // Качаем бицепсы-трицепсы нашей сети. Максимальное число итераций - 500. Это очень мало, но даже 500 интераций занимают довольно много времени. Чем больше итерацию обучения пройдет сеть - тем точнее будет ответ. 

let text, output;

//** Всего лишь простая функция для удобного вывода результатов работы */
function run(text) {
  output = net.run(text); //Тестируем работу ИНС 
  console.log(`\nФраза "${text}" имеет эмоциональную направленность: ${output}`)
}

// Запускаем ИНС и смотрим результаты
run('Я хорошо провел день') // хорошую
run('У меня все отлично') // хорошую
run('Я плохо себя чувствую') // плохую
run('Это ужасный день') // плохую
node v10.15.2 linux/amd64