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

Add survival dt example class

Browse Source 
It will be used to create decision tree examples with project specific attributes like hunger, distance from food etc...
This commit is contained in:
Michał Czekański 2020-05-23 17:20:18 +02:00
parent 8bc292f06d
commit 052516540f
1 changed files with 34 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
src/AI/DecisionTrees/projectSpecificClasses/SurvivalDTExample.py Normal file
View File

@ -0,0 +1,34 @@
from src.AI.DecisionTrees.Attribute import Attribute
from src.AI.DecisionTrees.DecisionTreeExample import DecisionTreeExample
from src.AI.DecisionTrees.projectSpecificClasses.DistFromObject import DistFromObject
from src.AI.DecisionTrees.projectSpecificClasses.PlayerStatsValue import PlayerStatsValue
from src.AI.DecisionTrees.projectSpecificClasses.SurvivalAttributesDefinitions import SurvivalAttributesDefinitions
from src.AI.DecisionTrees.projectSpecificClasses.SurvivalClassification import SurvivalClassification
class SurvivalDTExample(DecisionTreeExample):
hungerVal: PlayerStatsValue
thirstVal: PlayerStatsValue
staminaVal: PlayerStatsValue
distFromFood: DistFromObject
distFromWater: DistFromObject
distFromRestPlace: DistFromObject
def __init__(self, classification: SurvivalClassification, hungerVal: PlayerStatsValue, thirstVal: PlayerStatsValue,
staminaVal: PlayerStatsValue, distFromFood: DistFromObject, distFromWater: DistFromObject,
distFromRestPlace: DistFromObject):
self.hungerVal = hungerVal
self.thirstVal = thirstVal
self.staminaVal = staminaVal
self.distFromFood = distFromFood
self.distFromWater = distFromWater
self.distFromRestPlace = distFromRestPlace
attributes = [Attribute(SurvivalAttributesDefinitions.hungerAttrDef, hungerVal),
Attribute(SurvivalAttributesDefinitions.thirstAttrDef, thirstVal),
Attribute(SurvivalAttributesDefinitions.staminaAttrDef, staminaVal),
Attribute(SurvivalAttributesDefinitions.foodDistanceAttrDef, distFromFood),
Attribute(SurvivalAttributesDefinitions.waterDistanceAttrDef, distFromWater),
Attribute(SurvivalAttributesDefinitions.restDistanceAttrDef, distFromRestPlace)]
super().__init__(classification, attributes)