Commit Graph

12 Commits

Author SHA1 Message Date
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
53351b2ece fix needless-range and clippy::ptr_arg warnings. (#36)
* Fix needless for loop range

* Do not ignore clippy::ptr_arg
2020-12-11 16:52:39 -04:00
morenol
f0b348dd6e feat: BernoulliNB (#31)
* feat: BernoulliNB

* Move preprocessing to a trait in linalg/stats.rs
2020-12-04 20:45:40 -04:00
morenol
4720a3a4eb MultinomialNB (#32)
feat: add MultinomialNB
2020-12-03 09:51:33 -04: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
ad3ac49dde Implement GaussianNB (#27)
* feat: Add GaussianNB
2020-11-19 14:19:22 -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
Luis Moreno
6587ac032b Rename to from_array 2020-11-11 22:23:56 -04:00
Luis Moreno
900078cb04 Implement abstract method to convert a slice to a BaseVector, Implement RealNumberVector over BaseVector instead of over Vec<T> 2020-11-11 22:10:01 -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