From 4e27052ebfb20789c3d2928e8eb0e7f462aa3349 Mon Sep 17 00:00:00 2001 From: ZhuangYumin Date: Wed, 3 Jul 2024 01:27:56 +0000 Subject: [PATCH] bayes q1 --- tracking/inference.py | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) diff --git a/tracking/inference.py b/tracking/inference.py index fc3cd0b..a5afcd6 100644 --- a/tracking/inference.py +++ b/tracking/inference.py @@ -60,9 +60,15 @@ def constructBayesNet(gameState: hunters.GameState): edges = [] variableDomainsDict = {} - "*** YOUR CODE HERE ***" - raiseNotDefined() - "*** END YOUR CODE HERE ***" + variables = [PAC, GHOST0, GHOST1, OBS0, OBS1] + edges = [(PAC, OBS0), (GHOST0, OBS0), (PAC, OBS1), (GHOST1, OBS1)] + variableDomainsDict[PAC] = [(x, y) for x in range(X_RANGE) for y in range(Y_RANGE)] + variableDomainsDict[GHOST0] = [(x, y) for x in range(X_RANGE) for y in range(Y_RANGE)] + variableDomainsDict[GHOST1] = [(x, y) for x in range(X_RANGE) for y in range(Y_RANGE)] + max_manhattan = X_RANGE + Y_RANGE - 2 + max_obs = max_manhattan + MAX_NOISE + variableDomainsDict[OBS0] = list(range(max_obs + 1)) + variableDomainsDict[OBS1] = list(range(max_obs + 1)) net = bn.constructEmptyBayesNet(variables, edges, variableDomainsDict) return net