subset.nb2

来自「数据挖掘中的bayes算法,很好的代码」· NB2 代码 · 共 27 行

/*----------------------------------------------------------------------
  domains
----------------------------------------------------------------------*/
dom(x) = ZZ [0, 6];
dom(y) = ZZ [0, 8];
dom(class) = { a, b };

/*----------------------------------------------------------------------
  naive Bayes classifier
----------------------------------------------------------------------*/
nbc(class) = {
  prob(class) = {
    a: 10,
    b: 10 };
  prob(x|class) = {
    a: N(1.1, 0.766667) [10],
    b: N(3.9, 0.766667) [10] };
  prob(y|class) = {
    a: N(1.5, 3.16667) [10],
    b: N(6.5, 3.16667) [10] };
};

/*----------------------------------------------------------------------
  number of attributes: 3
  number of tuples    : 20
----------------------------------------------------------------------*/