diff --git a/src/AI/DecisionTrees/projectSpecificClasses/Examples.py b/src/AI/DecisionTrees/projectSpecificClasses/Examples.py new file mode 100644 index 0000000..2b6dcf8 --- /dev/null +++ b/src/AI/DecisionTrees/projectSpecificClasses/Examples.py @@ -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), + +] \ No newline at end of file