python - 使用贝叶斯模型预测值并得到错误

Question

数据集文件我在 python 中使用贝叶斯模型预测 no_of_people 值。我正在使用图片中给出的这个数据集。我的错误也附在图片中

#My code is here 
import numpy as np
from pgmpy.models import BayesianModel
import pandas as pd
data = pd.read_csv('C:/Users/hp/Desktop/file3.csv')
train = data[:40]
test = data[40:].drop('no_of_people', axis=1)
restaurant_model = BayesianModel(
[('location', 'cost'),
('quality', 'cost'),
('location', 'no_of_people'),
('cost', 'no_of_people')])
restaurant_model.fit(train)
restaurant_model.get_cpds()
restaurant_model.predict(test).values.ravel()

请帮我解决这个问题。我怎样才能完美地预测这些值。