我正在使用scikit-learn机器学习库(Python)进行机器学习项目.我正在使用的算法之一是高斯朴素贝叶斯实现.GaussianNB()函数的一个属性如下:
class_prior_ : array, shape (n_classes,)
我想先手动更改类,因为我使用的数据非常偏斜,其中一个类的调用非常重要.通过为该类分配较高的先验概率,召回应该增加.
但是,我无法弄清楚如何正确设置属性.我已经阅读了以下主题,但他们的答案对我不起作用.
如何在scikit-learn中为Naive Bayes clf手动设置先验概率？
我怎么知道我给sci-kit学习的先前资料？(朴素贝叶斯分类器.)
这是我的代码:
gnb = GaussianNB()
gnb.class_prior_ = [0.1, 0.9]
gnb.fit(data.XTrain, yTrain)
yPredicted = gnb.predict(data.XTest)
我认为这是正确的语法,我可以通过播放值找出哪个类属于数组中的哪个位置,但结果保持不变.也没有给出任何错误.
从scikit-learn库设置GaussianNB算法属性的正确方法是什么？