s444409/DSZI_Survival
s444409/DSZI_Survival
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Add printing tree decision to console

Browse Source
This commit is contained in:
Michał Czekański 2020-06-15 14:59:37 +02:00
parent c5497ff291
commit 5aaba17b2e
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/game/Game.py
View File

@ -197,6 +197,7 @@ class Game:
# Choose target for player using decision tree # Choose target for player using decision tree
if self.player.movementTarget is None: if self.player.movementTarget is None:
pickedEntity = survivalDecisionTree.pickEntity(self.player, self.map) pickedEntity = survivalDecisionTree.pickEntity(self.player, self.map)
self.screen.ui.console.printToConsole("I've decided to go for: {}".format(pickedEntity.classifier.name))
if pickedEntity.classifier == Classifiers.FOOD: if pickedEntity.classifier == Classifiers.FOOD:
result = image_predictor.predict_image() result = image_predictor.predict_image()
self.screen.ui.console.printToConsole("I think it is a: " + result[1]) self.screen.ui.console.printToConsole("I think it is a: " + result[1])
@ -439,8 +440,10 @@ class Game:
# Choose target for player using decision tree # Choose target for player using decision tree
if self.player.movementTarget is None: if self.player.movementTarget is None:
self.player.gotoToTarget(survivalDecisionTree.pickEntity(self.player, self.map), self.map) pickedEntity = survivalDecisionTree.pickEntity(self.player, self.map)
self.player.gotoToTarget(pickedEntity, self.map)
decisionsMade += 1 decisionsMade += 1
self.screen.ui.console.printToConsole("I've decided to go for: {}".format(pickedEntity.classifier.name))
if pauseAfterDecision: if pauseAfterDecision:
pause = True pause = True