bayes-ch

朴素贝叶斯分类

安装

npm i bayes-ch
// or
yarn add bayes-ch

使用

导入代码

// 导入代码
const Bayes = require('bayes-ch');
// 实例化类
langBayes = new Bayes();

创建数据样本

//训练数据
langBayes.train("学校有多少个校区?分别在哪?", '校区');
langBayes.train("诚信分怎么获取", '诚信分');
langBayes.train("Ein Bayes-Klassifikator (Aussprache: [beɪz], benannt nach dem englischen Mathematiker Thomas Bayes), ist ein aus dem Satz von Bayes hergeleiteter Klassifikator. Er ordnet jedes Objekt der Klasse zu, zu der es mit der größten Wahrscheinlichkeit gehört, oder bei der durch die Einordnung die wenigsten Kosten entstehen. Genau genommen handelt es sich um eine mathematische Funktion, die jedem Punkt eines Merkmalsraums eine Klasse zuordnet.", 'German');
langBayes.train("Naiwny klasyfikator bayesowski – prosty klasyfikator probabilistyczny. Naiwne klasyfikatory bayesowskie są oparte na założeniu o wzajemnej niezależności predyktorów (zmiennych niezależnych). Często nie mają one żadnego związku z rzeczywistością i właśnie z tego powodu nazywa się je naiwnymi. Bardziej opisowe jest określenie – „model cech niezależnych”. Ponadto model prawdopodobieństwa można wyprowadzić korzystając z twierdzenia Bayesa.", 'Polish');

调用数据

//获取每个分类的匹配度
var scores = langBayes.guess('诚信分');
console.log(scores);
//获取最高匹配度的对象
var winner = langBayes.extractWinner(scores);
console.log(winner);

查询结果

{
  '校区': 0.4,
  '诚信分': 0.8571428571428571,
  German: 0.06250000000000003,
  Polish: 0.06250000000000003
}
{ label: '诚信分', score: 0.8571428571428571 }

导出json模型

let jsonData = langBayes.toJSON();
fs.writeFileSync("train.json",jsonData);

使用json模型

var Bayes = require('../bayes');
langBayes = new Bayes();
// 导入数据
const   json = require('./train.json');
// json转换成功实例
langBayes.fromJSON(JSON.stringify(json))
// 推测数据
var scores = langBayes.guess('诚信分');
console.log(scores);
// 提取匹配度最高的数据
var winner = langBayes.extractWinner(scores);
console.log(winner);