From 8d4cde8eb0bc8f6d2e93a46232ad37ea2fcf7d50 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Micha=C5=82=20Czeka=C5=84ski?= Date: Sat, 23 May 2020 17:15:44 +0200 Subject: [PATCH] Add enum for classifying dt survival examples. This enum will be used in decision tree learning. For example food classification will mean that player should go for food. --- .../projectSpecificClasses/SurvivalClassification.py | 11 +++++++++++ 1 file changed, 11 insertions(+) create mode 100644 src/AI/DecisionTrees/projectSpecificClasses/SurvivalClassification.py diff --git a/src/AI/DecisionTrees/projectSpecificClasses/SurvivalClassification.py b/src/AI/DecisionTrees/projectSpecificClasses/SurvivalClassification.py new file mode 100644 index 0000000..6d813e3 --- /dev/null +++ b/src/AI/DecisionTrees/projectSpecificClasses/SurvivalClassification.py @@ -0,0 +1,11 @@ +from enum import Enum + + +class SurvivalClassification(Enum): + """ + For example food classification means that player should go for food. + """ + + FOOD = 0, + WATER = 1, + REST = 2