如何解决Python石榴Bayes网络声明错误
import csv
import pandas
from pomegranate import *
csvFile = pandas.read_csv('data.csv')
layer1= DiscreteDistribution({0: 1./3,0.5: 1./3,1.0: 1./3})
layer2 = DiscreteDistribution({0: 1./5,0.25: 1./3,0.5: 1./5,0.75: 1./5,1.0: 1./5})
node1 = Node(layer1,name="node1")
node2 = Node(layer1,name="node2")
node3 = Node(layer1,name="node3")
node4 = Node(layer2,name="node4")
node5 = Node(layer2,name="node5")
node6 = Node(layer2,name="node6")
model = BayesianNetwork("Bayes Net")
model.add_states(node1,node2,node3,node4,node5,node6)
model.add_edge(node1,node4)
model.add_edge(node1,node5)
model.add_edge(node1,node6)
model.add_edge(node2,node4)
model.add_edge(node2,node5)
model.add_edge(node2,node6)
model.add_edge(node3,node4)
model.add_edge(node3,node5)
model.add_edge(node3,node6)
model.bake()
model.fit(csvFile)
print(model.predict([[0,1,0.5,None,None]]))
我正在尝试使用石榴训练贝叶斯网。网络有两层,三个节点的输入层和三个节点的输出层。每个第一层节点连接到第二层节点。运行此代码时出现以下错误:
File "pomegranate\BayesianNetwork.pyx",line 525,in pomegranate.BayesianNetwork.BayesianNetwork.predict
File "pomegranate\BayesianNetwork.pyx",line 621,in pomegranate.BayesianNetwork.BayesianNetwork.predict_proba
File "pomegranate\BayesianNetwork.pyx",line 616,in pomegranate.BayesianNetwork.BayesianNetwork.predict_proba
File "pomegranate\FactorGraph.pyx",line 325,in pomegranate.FactorGraph.FactorGraph.predict_proba
File "pomegranate\distributions\DiscreteDistribution.pyx",line 95,in pomegranate.distributions.DiscreteDistribution.DiscreteDistribution.__mul__
AssertionError
是什么原因造成的?为什么模型无法预测?
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。