вывод исключения переменных. ошибка map_query для байесовской модели

Question

Я пытался вычислить MAP-запрос по переменным, имеющим свидетельства.

from pgmpy.inference import VariableElimination
from pgmpy.models import BayesianModel
import numpy as np
import pandas as pd
values = pd.DataFrame(np.random.randint(low=0, high=2, size=(1000, 5)),
                       columns=['A', 'B', 'C', 'D', 'E'])
model = BayesianModel([('A', 'B'), ('C', 'B'), ('C', 'D'), ('B', 'E')])
model.fit(values)
inference = VariableElimination(model)
phi_query = inference.map_query(['A', 'B'], evidence= {'B':1})

что дает мне ошибку:

Finding Elimination Order: : 100%|██████████| 3/3 [00:00<00:00, 651.66it/s]
Eliminating: E: 100%|██████████| 3/3 [00:00<00:00, 309.08it/s]
---------------------------------------------------------------------------
KeyError                                  Traceback (most recent call last)
<ipython-input-22-0e47cda916c1> in <module>()
      8 model.fit(values)
      9 inference = VariableElimination(model)
---> 10 phi_query = inference.map_query(['A', 'B'], evidence= {'B':1})

/usr/local/lib/python3.6/dist-packages/pgmpy/inference/ExactInference.py in map_query(self, variables, evidence, elimination_order, show_progress)
    360             return_dict = {}
    361             for var in variables:
--> 362                 return_dict[var] = map_query_results[var]
    363             return return_dict
    364 

KeyError: 'B'

Согласно документации:

Параметры переменных (список) - список переменных, по которым мы хотим вычислить max-marginal.

доказательство (dict) - ключ dict, пара значений как {var: state_of_var_observed} Нет, если нет доказательств

elimination_order (list) - порядок исключения переменных (если ничего не указано) порядок вычисляется автоматически

Итак, где я ошибаюсь, почему я получаю эту ошибку?

РЕДАКТИРОВАТЬ: версия pgmpy: 0.1.9

Answer 1

Вероятно, это из-за используемой вами версии pgmpy. Попробуй это

!pip install pgmpy==0.1.6

а затем запустите код.

Answer 2

Проблема в том, что вы передаете переменную доказательства также в переменных запроса, и нет никаких проверок для правильной обработки этого случая. Вы уже знаете состояние дляB в качестве 1 поскольку это свидетельство, вам нужно просто запросить A в качестве:

>>> phi_query = inference.map_query(['A'], evidence= {'B':1})
>>> print(phi_query)
{'A': 1}