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

Add function testing decision tree and game mode to use it

Browse Source
This commit is contained in:
Michał Czekański 2020-05-25 02:21:18 +02:00
parent c1b12883f8
commit 63b1d36c36
2 changed files with 16 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
DecisionTree.md Normal file
View File

@ -0,0 +1 @@
# DSZI_Survival - Drzewa Decyzyjne

15
src/game/Game.py
View File

@ -8,6 +8,7 @@ import pygame
from src.AI.Affinities import Affinities from src.AI.Affinities import Affinities
from src.AI.DecisionTrees.DecisionTree import DecisionTree from src.AI.DecisionTrees.DecisionTree import DecisionTree
from src.AI.DecisionTrees.ExamplesManager import ExamplesManager from src.AI.DecisionTrees.ExamplesManager import ExamplesManager
from src.AI.DecisionTrees.TestDecisionTree import testDecisionTree
from src.AI.DecisionTrees.projectSpecificClasses.SurvivalClassification import SurvivalClassification from src.AI.DecisionTrees.projectSpecificClasses.SurvivalClassification import SurvivalClassification
from src.AI.GA import geneticAlgorithm from src.AI.GA import geneticAlgorithm
from src.entities.Player import Player from src.entities.Player import Player
@ -67,6 +68,9 @@ class Game:
if argv[2] == "-p": if argv[2] == "-p":
print("Running Decision Tree in pause mode.") print("Running Decision Tree in pause mode.")
self.dtRun(filesPath, True) self.dtRun(filesPath, True)
elif argv[2] == "-test":
print("Testing Decision Tree.")
self.dtTestRun(filesPath, int(argv[3]))
else: else:
print("Running Decision Tree.") print("Running Decision Tree.")
self.dtRun(filesPath) self.dtRun(filesPath)
@ -360,3 +364,14 @@ class Game:
geneticAlgorithmWithDecisionTree(self.map, iter, 10, dtExamples, 0.1) geneticAlgorithmWithDecisionTree(self.map, iter, 10, dtExamples, 0.1)
print("Time elapsed: ", self.pgTimer.tick() // 1000) print("Time elapsed: ", self.pgTimer.tick() // 1000)
def dtTestRun(self, filesPath, iterations):
# Read examples for decision tree testing
examplesFilePath = str(
filesPath) + os.sep + "data" + os.sep + "AI_data" + os.sep + "dt_exmpls" + os.sep + "dt_examples"
examplesManager = ExamplesManager(examplesFilePath)
examples = examplesManager.readExamples()
scores = testDecisionTree(examples, iterations)
avg = sum(scores) / iterations
print("Average: {}".format(str(avg)))