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
9b221979da
fix: clippy, documentation and formatting
2020-12-22 16:35:28 -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
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
