From 3fafade1e91e0712eb5d4ce051cb4d6fc59de027 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Micha=C5=82=20Czeka=C5=84ski?= Date: Sun, 24 May 2020 00:38:54 +0200 Subject: [PATCH] Add examples class which will contain learning examples for decision tree --- .../projectSpecificClasses/Examples.py | 23 +++++++++++++++++++ 1 file changed, 23 insertions(+) create mode 100644 src/AI/DecisionTrees/projectSpecificClasses/Examples.py 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