https://github.com/mljs/naive-bayes
Naive bayes classifier.
https://github.com/mljs/naive-bayes
Last synced: about 1 year ago
JSON representation
Naive bayes classifier.
- Host: GitHub
- URL: https://github.com/mljs/naive-bayes
- Owner: mljs
- License: mit
- Created: 2015-07-05T19:44:13.000Z (almost 11 years ago)
- Default Branch: master
- Last Pushed: 2019-06-29T11:15:57.000Z (almost 7 years ago)
- Last Synced: 2025-04-01T15:41:58.597Z (about 1 year ago)
- Language: JavaScript
- Size: 1.24 MB
- Stars: 48
- Watchers: 14
- Forks: 14
- Open Issues: 6
-
Metadata Files:
- Readme: README.md
- Changelog: History.md
- License: LICENSE
Awesome Lists containing this project
README
# ml-naivebayes
[![NPM version][npm-image]][npm-url]
[![build status][travis-image]][travis-url]
[![Test coverage][codecov-image]][codecov-url]
[![npm download][download-image]][download-url]
Naive bayes classifiers.
## Installation
`npm install ml-naivebayes`
## [API Documentation](https://mljs.github.io/naive-bayes/)
## Usage
### [GaussianNB](./src/GaussianNB.js)
```js
// assuming that you created Xtrain, Xtest, Ytrain, Ytest
import { GaussianNB } from 'ml-naivebayes';
var model = new GaussianNB();
model.train(Xtrain, Ytrain);
var predictions = model.predict(Xtest);
```
### [MultinomialNB](./src/MultinomialNB.js)
```js
// assuming that you created Xtrain, Xtest, Ytrain, Ytest
import { MultinomialNB } from 'ml-naivebayes';
var model = new MultinomialNB();
model.train(Xtrain, Ytrain);
var predictions = model.predict(Xtest);
```
## Authors
* [Jefferson Hernandez](https://github.com/JeffersonH44)
## License
[MIT](./LICENSE)
[npm-image]: https://img.shields.io/npm/v/ml-naivebayes.svg?style=flat-square
[npm-url]: https://npmjs.org/package/ml-naivebayes
[travis-image]: https://img.shields.io/travis/mljs/naive-bayes/master.svg?style=flat-square
[travis-url]: https://travis-ci.org/mljs/naive-bayes
[codecov-image]: https://img.shields.io/codecov/c/github/mljs/naive-bayes.svg?style=flat-square
[codecov-url]: https://codecov.io/github/mljs/naive-bayes
[download-image]: https://img.shields.io/npm/dm/ml-naivebayes.svg?style=flat-square
[download-url]: https://npmjs.org/package/ml-naivebayes