如何解决为什么在使用 Naive-Bayes 中的 CategoricalNB 训练模型时会出现 IndexError?
我正在尝试使用以下代码训练 CategoricalNB 模型。我发现所有变量都是不言自明的。
from sklearn.naive_bayes import CategoricalNB
cnb = CategoricalNB()
cnb.fit(train_split,train_split_label)
cnb_preds = cnb.predict(test_split)
print('test_CNB: ',accuracy_score(test_split_label,cnb_preds))
但是,在预测时,我收到以下错误:
/usr/local/lib/python3.6/dist-packages/sklearn/naive_bayes.py in _joint_log_likelihood(self,X)
1217 for i in range(self.n_features_):
1218 indices = X[:,i]
-> 1219 jll += self.feature_log_prob_[i][:,indices].T
1220 total_ll = jll + self.class_log_prior_
1221 return total_ll
IndexError: index 1 is out of bounds for axis 1 with size 1
如果您需要,我可以提供有关 DataFrames(train_split、test_split 等...)的任何形状信息,但它们都是匹配的。你能帮我找到这个问题的根源吗?
提前致谢。
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。