Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add Multinomial and Bernoulli Naive Bayes variants #4053

Merged
merged 26 commits into from
Jul 22, 2021
Merged

Add Multinomial and Bernoulli Naive Bayes variants #4053

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
26 commits
Select commit Hold shift + click to select a range
d0ecf57
Initial gaussian naive bayes variant
cjnolet Feb 19, 2020
053f8f3
Adding tests for gaussian naive bayes
cjnolet Feb 19, 2020
3d4ca96
Merge branch '013-fea-kmeans-da' into fea-ext-naive_bayes_variants
cjnolet Feb 27, 2020
7a0bf65
A couple changes. Starting to test end to end to at least verify the …
cjnolet Feb 27, 2020
ddda055
Merge branch 'branch-0.15' into fea-ext-naive_bayes_variants
cjnolet Jun 11, 2020
caea9d2
A little progress. There are still some different things to consider …
cjnolet Jun 12, 2020
7477a7b
A few updates to the test. We are producing the right results, just a…
cjnolet Jun 12, 2020
a1b95ad
Initial port of Bernoulli & Categorical naive bayes. Also adding `bin…
cjnolet Jun 24, 2020
2ca8938
Merge branch 'branch-21.08' into fea-ext-naive_bayes_variants
lowener Jun 10, 2021
29d8173
Merge branch 'branch-21.08' into fea-ext-naive_bayes_variants
lowener Jun 15, 2021
e144d28
Update naive bayes refactor
lowener Jun 16, 2021
a89f7cd
Update Gaussian Naive Bayes code
lowener Jun 29, 2021
4a71a32
Adding working version of GNB and Bernoulli and their test
lowener Jul 12, 2021
42dc80c
Update init and binarize primitive
lowener Jul 12, 2021
130d4ae
Adding working CategoricalNB
lowener Jul 13, 2021
f9480d9
Separating code to keep Multinomial and Bernoulli only
lowener Jul 13, 2021
f87cafd
Merge branch 'branch-21.08' into 21.08-multinomial-nb
lowener Jul 13, 2021
9945d93
Fix style
lowener Jul 13, 2021
8f4eee9
Update copyright
lowener Jul 14, 2021
e6cb185
Fix docstring
lowener Jul 14, 2021
3063586
Update tests to compare with sklearn and factorize naive bayes init
lowener Jul 19, 2021
63133c4
Fix style
lowener Jul 19, 2021
defbf41
Fix count_features_dense write order
lowener Jul 20, 2021
c63dab4
Fix class_prior parameter
lowener Jul 21, 2021
dea0218
Update doc and api.rst
lowener Jul 22, 2021
e54b783
Fix style
lowener Jul 22, 2021
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
7 changes: 5 additions & 2 deletions docs/source/api.rst
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -258,10 +258,13 @@ Multiclass Classification
.. autoclass:: cuml.multiclass.OneVsRestClassifier
:members:

Mutinomial Naive Bayes
Naive Bayes
----------------------

.. autoclass:: cuml.MultinomialNB
.. autoclass:: cuml.naive_bayes.MultinomialNB
:members:

.. autoclass:: cuml.naive_bayes.BernoulliNB
:members:

Stochastic Gradient Descent
Expand Down
3 changes: 2 additions & 1 deletion python/cuml/naive_bayes/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
#
# Copyright (c) 2020, NVIDIA CORPORATION.
# Copyright (c) 2020-2021, NVIDIA CORPORATION.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
Expand All @@ -15,3 +15,4 @@
#

from cuml.naive_bayes.naive_bayes import MultinomialNB
from cuml.naive_bayes.naive_bayes import BernoulliNB
Loading