add dataset for garbage classification

decision_tree/tree_as_txt.txt

@@ -16,28 +16,6 @@
 |   |   |   |   |   |--- feature_1 <= 2.50
 |   |   |   |   |   |   |--- feature_0 <= 2.50
 |   |   |   |   |   |   |   |--- feature_1 <= 1.50
-|   |   |   |   |   |   |   |   |--- feature_4 <= 2.50
-|   |   |   |   |   |   |   |   |   |--- class: 1
-|   |   |   |   |   |   |   |   |--- feature_4 >  2.50
-|   |   |   |   |   |   |   |   |   |--- feature_2 <= 2.00
-|   |   |   |   |   |   |   |   |   |   |--- class: 1
-|   |   |   |   |   |   |   |   |   |--- feature_2 >  2.00
-|   |   |   |   |   |   |   |   |   |   |--- class: 0
-|   |   |   |   |   |   |   |--- feature_1 >  1.50
-|   |   |   |   |   |   |   |   |--- class: 0
-|   |   |   |   |   |   |--- feature_0 >  2.50
-|   |   |   |   |   |   |   |--- feature_2 <= 2.50
-|   |   |   |   |   |   |   |   |--- class: 1
-|   |   |   |   |   |   |   |--- feature_2 >  2.50
-|   |   |   |   |   |   |   |   |--- feature_4 <= 2.50
-|   |   |   |   |   |   |   |   |   |--- class: 1
-|   |   |   |   |   |   |   |   |--- feature_4 >  2.50
-|   |   |   |   |   |   |   |   |   |--- class: 0
-|   |   |   |   |   |--- feature_1 >  2.50
-|   |   |   |   |   |   |--- feature_0 <= 3.50
-|   |   |   |   |   |   |   |--- class: 0
-|   |   |   |   |   |   |--- feature_0 >  3.50
-|   |   |   |   |   |   |   |--- feature_1 <= 3.50
 |   |   |   |   |   |   |   |   |--- feature_2 <= 2.50
 |   |   |   |   |   |   |   |   |   |--- class: 1
 |   |   |   |   |   |   |   |   |--- feature_2 >  2.50
@@ -45,6 +23,28 @@
 |   |   |   |   |   |   |   |   |   |   |--- class: 1
 |   |   |   |   |   |   |   |   |   |--- feature_4 >  2.00
 |   |   |   |   |   |   |   |   |   |   |--- class: 0
+|   |   |   |   |   |   |   |--- feature_1 >  1.50
+|   |   |   |   |   |   |   |   |--- class: 0
+|   |   |   |   |   |   |--- feature_0 >  2.50
+|   |   |   |   |   |   |   |--- feature_4 <= 2.50
+|   |   |   |   |   |   |   |   |--- class: 1
+|   |   |   |   |   |   |   |--- feature_4 >  2.50
+|   |   |   |   |   |   |   |   |--- feature_2 <= 2.50
+|   |   |   |   |   |   |   |   |   |--- class: 1
+|   |   |   |   |   |   |   |   |--- feature_2 >  2.50
+|   |   |   |   |   |   |   |   |   |--- class: 0
+|   |   |   |   |   |--- feature_1 >  2.50
+|   |   |   |   |   |   |--- feature_0 <= 3.50
+|   |   |   |   |   |   |   |--- class: 0
+|   |   |   |   |   |   |--- feature_0 >  3.50
+|   |   |   |   |   |   |   |--- feature_1 <= 3.50
+|   |   |   |   |   |   |   |   |--- feature_4 <= 2.50
+|   |   |   |   |   |   |   |   |   |--- class: 1
+|   |   |   |   |   |   |   |   |--- feature_4 >  2.50
+|   |   |   |   |   |   |   |   |   |--- feature_2 <= 2.00
+|   |   |   |   |   |   |   |   |   |   |--- class: 1
+|   |   |   |   |   |   |   |   |   |--- feature_2 >  2.00
+|   |   |   |   |   |   |   |   |   |   |--- class: 0
 |   |   |   |   |   |   |   |--- feature_1 >  3.50
 |   |   |   |   |   |   |   |   |--- class: 0
 |   |   |   |   |--- feature_3 >  4.50
@@ -59,13 +59,13 @@
 |   |   |   |   |   |   |--- class: 1
 |   |   |   |   |--- feature_3 >  3.50
 |   |   |   |   |   |--- feature_1 <= 2.50
-|   |   |   |   |   |   |--- feature_3 <= 4.50
-|   |   |   |   |   |   |   |--- feature_0 <= 2.50
-|   |   |   |   |   |   |   |   |--- class: 0
-|   |   |   |   |   |   |   |--- feature_0 >  2.50
-|   |   |   |   |   |   |   |   |--- class: 1
-|   |   |   |   |   |   |--- feature_3 >  4.50
+|   |   |   |   |   |   |--- feature_0 <= 2.50
 |   |   |   |   |   |   |   |--- class: 0
+|   |   |   |   |   |   |--- feature_0 >  2.50
+|   |   |   |   |   |   |   |--- feature_3 <= 4.50
+|   |   |   |   |   |   |   |   |--- class: 1
+|   |   |   |   |   |   |   |--- feature_3 >  4.50
+|   |   |   |   |   |   |   |   |--- class: 0
 |   |   |   |   |   |--- feature_1 >  2.50
 |   |   |   |   |   |   |--- class: 0
 |   |   |   |--- feature_4 >  4.50

main.py

@@ -118,7 +118,7 @@ class Game():
         #     print('----')
 
         print('positive actions')
-        print(len(self.positive_actions))
+        print(len(self.positive_decision))
         for i in self.positive_decision:
             # print(i.get_coords())
             trash_x, trash_y = i.get_coords()