Add examples class which will contain learning examples for decision tree

This commit is contained in:
Michał Czekański 2020-05-24 00:38:54 +02:00
parent 0d6b3e4e84
commit 3fafade1e9

View File

@ -0,0 +1,23 @@
from src.AI.DecisionTrees.projectSpecificClasses.SurvivalClassification import SurvivalClassification
from src.AI.DecisionTrees.projectSpecificClasses.SurvivalDTExample import SurvivalDTExample
from src.AI.DecisionTrees.projectSpecificClasses.PlayerStatsValue import PlayerStatsValue as Stats
from src.AI.DecisionTrees.projectSpecificClasses.DistFromObject import DistFromObject as Dist
examples = [
SurvivalDTExample(SurvivalClassification.WATER,
Stats.HALF_TO_THREE_QUARTERS,
Stats.QUARTER_TO_HALF,
Stats.QUARTER_TO_HALF,
Dist.GE_3_LT_8,
Dist.GE_8,
Dist.LT_3),
SurvivalDTExample(SurvivalClassification.FOOD,
Stats.HALF_TO_THREE_QUARTERS,
Stats.QUARTER_TO_HALF,
Stats.QUARTER_TO_HALF,
Dist.LT_3,
Dist.GE_8,
Dist.GE_8),
]