提问者:小点点

来自scikit学习的朴素贝叶斯模型


两个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)


共1个答案

匿名用户

我看到两个区别:

  1. 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()