Skip to content

Latest commit

 

History

History
90 lines (80 loc) · 2.18 KB

bayes-nlu.md

File metadata and controls

90 lines (80 loc) · 2.18 KB
Raw

Bayes NLU

You can train a Bayes NLU classifier (indicating language) with utterances and their intents. Then you can give a different utterance, and get the classifications for each intent, sorted descending by the score value.

const { BayesNLU } = require('node-nlp');

async function main() {
  const classifier = new BayesNLU({ language: 'fr' });
  classifier.add('Bonjour', 'greet');
  classifier.add('bonne nuit', 'greet');
  classifier.add('Bonsoir', 'greet');
  classifier.add("J'ai perdu mes clés", 'keys');
  classifier.add('Je ne trouve pas mes clés', 'keys');
  classifier.add('Je ne me souviens pas où sont mes clés', 'keys');
  await classifier.train();
  const classifications = classifier.getClassifications('où sont mes clés');
  console.log(classifications);
  // [ { label: 'keys', value: 0.9878048780487805 },
  // { label: 'greet', value: 0.01219512195121951 } ]
}
main();

Or you can get only the best classification

const { BayesNLU } = require('node-nlp');

async function main() {
  const classifier = new BayesNLU({ language: 'fr' });
  classifier.add('Bonjour', 'greet');
  classifier.add('bonne nuit', 'greet');
  classifier.add('Bonsoir', 'greet');
  classifier.add("J'ai perdu mes clés", 'keys');
  classifier.add('Je ne trouve pas mes clés', 'keys');
  classifier.add('Je ne me souviens pas où sont mes clés', 'keys');
  await classifier.train();
  const classifications = classifier.getBestClassification('où sont mes clés');
  console.log(classifications);
  // { label: 'keys', value: 0.9878048780487805 }
}
main();

Currently 40 languages are supported:

  • Arabic (ar)
  • Armenian (hy)
  • Basque (eu)
  • Bengali (bn)
  • Catala (ca)
  • Chinese (zh)
  • Czech (cs)
  • Danish (da)
  • Dutch (nl)
  • English (en)
  • Farsi (fa)
  • Finnish (fi)
  • French (fr)
  • Galician (gl)
  • German (de)
  • Greek (el)
  • Hindi (hi)
  • Hungarian (hu)
  • Indonesian (id)
  • Irish (ga)
  • Italian (it)
  • Japanese (ja)
  • Korean (ko)
  • Lithuanian (lt)
  • Malay (ms)
  • Nepali (ne)
  • Norwegian (no)
  • Polish (pl)
  • Portuguese (pt)
  • Romanian (ro)
  • Russian (ru)
  • Serbian (sr)
  • Slovene (sl)
  • Spanish (es)
  • Swedish (sv)
  • Tagalog (tl)
  • Tamil (ta)
  • Thai (th)
  • Turkish (tr)
  • Ukrainian (uk)