machine-learning - 如何调整 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_