From 25e71282fdd3a917c620da42345c8419d4599182 Mon Sep 17 00:00:00 2001
From: Devolta05 <sandyom07@gmail.com>
Date: Thu, 7 Oct 2021 11:49:00 +0530
Subject: [PATCH] Added Categorical Naive Bayes to classifier config dictionary

---
 blobcity/config/classifier_config.py | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/blobcity/config/classifier_config.py b/blobcity/config/classifier_config.py
index eb1170e..1111505 100644
--- a/blobcity/config/classifier_config.py
+++ b/blobcity/config/classifier_config.py
@@ -134,7 +134,7 @@ class classifier_config:
                 'booster':{'str':['gbtree', 'gblinear','dart']}
             }
         ],
-         "bernoullinb":[
+        "bernoullinb":[
             naive_bayes.BernoulliNB,
             {
                 "alpha":{'float':[1e-2,1.0]},
@@ -182,4 +182,11 @@ class classifier_config:
                 "epsilon":{'float':[1e-8, 0.1]},
             }
         ],
+        "categoricalnb":[
+            naive_bayes.CategoricalNB,
+            {
+                "alpha":{'float':[1e-2,1.0]},
+                "fit_prior":{'bool':[True,False],
+            }
+        ],
     }