Comment classes

src/AI/DecisionTrees/projectSpecificClasses/SurvivalAttributesDefinitions.py

@@ -4,6 +4,13 @@ from src.AI.DecisionTrees.projectSpecificClasses.PlayerStatsValue import PlayerS
 
 
 class SurvivalAttributesDefinitions:
+    """
+    This class contains all attributes definitions which will be used to create decision tree examples
+    and decision tree learning.
+
+    There are definitions of attributes like hunger, thirst, etc...
+    """
+
     attrsDefinitionsCount = 6
 
     hungerAttrDef = AttributeDefinition(0, "Hunger", [PlayerStatsValue.ZERO_TO_QUARTER,

src/AI/DecisionTrees/projectSpecificClasses/SurvivalDTExample.py

@@ -7,6 +7,10 @@ from src.AI.DecisionTrees.projectSpecificClasses.SurvivalClassification import S
 
 
 class SurvivalDTExample(DecisionTreeExample):
+    """
+    This class will be used to create examples for decision trees with project specific attributes like hunger,
+    distance from food, etc.
+    """
     hungerVal: PlayerStatsValue
     thirstVal: PlayerStatsValue
     staminaVal: PlayerStatsValue