如何调整 GaussianNB？

Question

尝试拟合数据GaussianNB()给我的准确度得分很低。

我想尝试网格搜索，但似乎无法设置参数sigma和theta。反正有调GausssianNB吗？

Answer 1

您可以像这样调整“ var_smoothing ”参数：

nb_classifier = GaussianNB()

params_NB = {'var_smoothing': np.logspace(0,-9, num=100)}
gs_NB = GridSearchCV(estimator=nb_classifier, 
                 param_grid=params_NB, 
                 cv=cv_method,   # use any cross validation technique 
                 verbose=1, 
                 scoring='accuracy') 
gs_NB.fit(x_train, y_train)

gs_NB.best_params_