石榴中的贝叶斯网络：ValueError：示例的维数与模型的维数不同

Question

我正在尝试使用Pomegranate包在python中建立贝叶斯网络模型。应该从数据中学习网络。所以我正在使用.from_samples方法。但是我在使用方法.predict_proba（）时遇到了麻烦，它给了我错误。

这就是我建立模型的方式：

model = BayesianNetwork.from_samples(X_train, algorithm='chow-liu')

这就是我的预测方式：

model.predict_proba(X_train)

这是我得到的错误：

ValueError：样本的维数与模型不相同。非常感谢您的帮助。

Answer 1

我得到了答案：您应该在调用from_samples方法时定义您的state_names。

另一个问题是我们如何使用该模型进行分类？

Answer 2

您应该使用predict（）方法来预测未值节点的状态。

检查documentation以获得更多详细信息。 另外，在repository中，您可以找到一些有趣的教程来帮助您。

Answer 3

请在要传递的样本周围添加[]。