两个gnb之间的区别是什么。fit()?
from sklearn.naive_bayes import GaussianNB
gnb = GaussianNB()
gnb.fit(X_train, y_train.ravel())
gnb.fit(X_train, y_train).predict(X_test)
我看到两个区别:
np。ravel
展平阵列,例如x = np.array([[1, 2, 3], [4, 5, 6]])
np.ravel(x)
array([1, 2, 3, 4, 5, 6])
如果您的y\u列车是一维的,那么无论您是否调用
gbn,您都将得到相同的模型。安装(x,y)
或gbn。拟合(x,y.ravel())
因为y==y.ravel()