Commit Graph

16 Commits

Author SHA1 Message Date
morenol
820201e920 Solve conflic with num-traits (#130)
* Solve conflic with num-traits

* Fix clippy warnings

Co-authored-by: Luis Moreno <morenol@users.noreply.github.com>
2022-05-05 10:39:18 -04:00
Luis Moreno
0b3bf946df chore: fix clippy warnings 2021-06-05 01:41:40 -04:00
Luis Moreno
162bed2aa2 feat: added support to wasm (#94)
* test: run tests also in wasm targets

* fix: install rand with wasm-bindgen por wasm targets

* fix: use actual usize size to access buffer.

* fix: do not run functions that create files in wasm.

* test: do not run in wasm test that panics.

Co-authored-by: Luis Moreno <morenol@users.noreply.github.com>
2021-04-28 15:58:39 -04:00
Luis Moreno
1b42f8a396 feat: Add getters for naive bayes structs (#74)
* feat: Add getters for GaussianNB

* Add classes getter to BernoulliNB

Add classes getter to CategoricalNB

Add classes getter to MultinomialNB

* Add feature_log_prob getter to MultinomialNB

* Add class_count to NB structs

* Add n_features getter for NB

* Add feature_count to MultinomialNB and BernoulliNB

* Add n_categories to CategoricalNB

* Implement feature_log_prob and category_count getter for CategoricalNB

* Implement feature_log_prob for BernoulliNB
2021-02-25 15:44:34 -04:00
Ben Cross
f1cf8a6f08 Added serde feature flags to tests 2021-01-18 10:32:35 +00:00
Ben Cross
762986b271 Cargo format 2021-01-17 21:37:30 +00:00
Ben Cross
e0d46f430b feat: Make SerDe optional 2021-01-17 21:35:03 +00:00
Volodymyr Orlov
810a5c429b feat: consolidates API 2020-12-24 18:36:23 -08:00
Volodymyr Orlov
d22be7d6ae fix: post-review changes 2020-12-24 13:47:09 -08:00
Volodymyr Orlov
dd341f4a12 feat: + builders for algorithm parameters 2020-12-23 12:29:39 -08:00
Volodymyr Orlov
a2be9e117f feat: + cross_validate, trait Predictor, refactoring 2020-12-22 15:41:53 -08:00
Luis Moreno
c9eb94ba93 Derive clone for NB Parameters 2020-12-18 00:39:54 -04:00
Volodymyr Orlov
5a185479a7 feat: NB documentation 2020-12-17 19:00:11 -08:00
morenol
9db993939e Add serde to CategoricalNB (#30)
* Add serde to CategoricalNB

* Implement PartialEq for CategoricalNBDistribution
2020-11-19 16:07:10 -04:00
morenol
72e9f8293f Use log likelihood to make calculations more stable (#28)
* Use log likelihood to make calculations more stable

* Fix problem with class_count in categoricalnb

* Use a similar approach to the one used in scikitlearn to define which are the possible categories of each feature.
2020-11-16 23:56:50 -04:00
morenol
3d4d5f64f6 feat: add Naive Bayes and CategoricalNB (#15)
* feat: Implement Naive Bayes classifier

* Implement CategoricalNB
2020-11-09 15:54:27 -04:00