From 94b4cbef58e2e130bc58276f94c28a1a4ae2567f Mon Sep 17 00:00:00 2001 From: AleksandraMuczynska Date: Sat, 21 May 2022 14:31:02 +0200 Subject: [PATCH] fixed hp directions --- learning/decision_tree.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/learning/decision_tree.py b/learning/decision_tree.py index 18c1c47..7f92344 100644 --- a/learning/decision_tree.py +++ b/learning/decision_tree.py @@ -47,7 +47,7 @@ class DecisionTree: opponents_parsed = [] for opponent in opponents: opponents_parsed.append( - (manhattan_distance(current_knight.position, opponent.position), parse_hp(opponent.current_hp))) + (manhattan_distance(current_knight.position, opponent.position), parse_hp(opponent.health_bar.current_hp))) prediction = self.get_prediction(tower_dist=distance_to_castle, tower_hp=castle.current_hp, mob1_dist=monsters_parsed[0][0], mob1_hp=monsters_parsed[0][1], @@ -56,7 +56,7 @@ class DecisionTree: opp2_dist=opponents_parsed[1][0], opp2_hp=opponents_parsed[1][1], opp3_dist=opponents_parsed[2][0], opp3_hp=opponents_parsed[2][1], opp4_dist=opponents_parsed[3][0], opp4_hp=opponents_parsed[3][1], - agent_hp=current_knight.current_hp) + agent_hp=current_knight.health_bar.current_hp) print(prediction) if prediction == 'tower': # castle... return castle_neighbors(grid, castle_bottom_right_row=castle.position[0],