Написание детектора типа шутки на JavaScript с использованием Brain.js

Написание детектора типа шутки на JavaScript с использованием Brain.js

Алгоритм машинного обучения в JavaScript, который можно использовать для классификации смысла шутки и прогнозирования ее юмористического характера.

const brain = require('brain.js');

// Create a new neural network
const net = new brain.recurrent.LSTM();

// Define the training data
const trainingData = [
  { input: 'Why was the math book sad? Because it had too many problems.', output: 'humorous' },
  { input: 'Why couldn\'t the bicycle stand up by itself? Because it was two-tired.', output: 'humorous' },
  { input: 'What do you get when you cross a snowman and a vampire? Frostbite.', output: 'humorous' },
  { input: 'Why was the computer cold? It left its Windows open.', output: 'humorous' },
  { input: 'What do you get when you cross a sheep and a kangaroo? A woolly jumper.', output: 'humorous' },
  { input: 'What do you get when you cross a snake and a pie? A python.', output: 'not humorous' }
];

// Train the neural network
net.train(trainingData, {
  errorThresh: 0.025,  // error threshold to reach
  iterations: 20000,   // maximum training iterations
  log: true,           // console.log() progress periodically
  logPeriod: 10        // number of iterations between logging
});

// Test the neural network
const output = net.run('Why was the math book sad? Because it had too many problems.');
console.log(output); // 'humorous'

Этот алгоритм использует библиотеку brain.js для создания рекуррентной нейронной сети и обучения ее набору шуток, помеченных как юмористические или не юмористические. Затем нейронная сеть может предсказать, является ли данная шутка юмористической, на основе шаблонов и корреляций, извлеченных из данных обучения.