diff --git a/decision_tree/decisionTree.py b/decision_tree/decisionTree.py index 3986b0e..f445e90 100644 --- a/decision_tree/decisionTree.py +++ b/decision_tree/decisionTree.py @@ -9,7 +9,7 @@ attributes = ["season", "trash_type", "mass", "space", "trash_mass"] # return tree made from attributes def tree(): - dataset = pandas.read_csv('/Users/mac/Desktop/tree_dataset.csv', sep=";") + dataset = pandas.read_csv('decision_tree/tree_dataset.csv', sep=";") x = dataset[attributes] y = dataset[decisions] diff --git a/decision_tree/tree_as_txt.txt b/decision_tree/tree_as_txt.txt index af74d02..519d17e 100644 --- a/decision_tree/tree_as_txt.txt +++ b/decision_tree/tree_as_txt.txt @@ -4,12 +4,12 @@ | | | |--- class: 0 | | |--- feature_0 > 1.50 | | | |--- feature_3 <= 3.50 -| | | | |--- feature_2 <= 2.50 +| | | | |--- feature_4 <= 2.50 | | | | | |--- class: 1 -| | | | |--- feature_2 > 2.50 -| | | | | |--- feature_4 <= 2.50 +| | | | |--- feature_4 > 2.50 +| | | | | |--- feature_2 <= 2.50 | | | | | | |--- class: 1 -| | | | | |--- feature_4 > 2.50 +| | | | | |--- feature_2 > 2.50 | | | | | | |--- class: 0 | | | |--- feature_3 > 3.50 | | | | |--- feature_3 <= 4.50 @@ -26,12 +26,12 @@ | | | | | | | |--- feature_1 > 1.50 | | | | | | | | |--- class: 0 | | | | | | |--- feature_0 > 2.50 -| | | | | | | |--- feature_4 <= 2.50 +| | | | | | | |--- feature_2 <= 2.50 | | | | | | | | |--- class: 1 -| | | | | | | |--- feature_4 > 2.50 -| | | | | | | | |--- feature_2 <= 2.50 +| | | | | | | |--- feature_2 > 2.50 +| | | | | | | | |--- feature_4 <= 2.50 | | | | | | | | | |--- class: 1 -| | | | | | | | |--- feature_2 > 2.50 +| | | | | | | | |--- feature_4 > 2.50 | | | | | | | | | |--- class: 0 | | | | | |--- feature_1 > 2.50 | | | | | | |--- feature_0 <= 3.50 @@ -73,8 +73,8 @@ | | |--- feature_2 > 1.50 | | | |--- class: 0 |--- feature_2 > 3.50 -| |--- feature_1 <= 1.50 -| | |--- feature_4 <= 1.50 +| |--- feature_4 <= 1.50 +| | |--- feature_1 <= 1.50 | | | |--- feature_2 <= 4.50 | | | | |--- feature_3 <= 4.50 | | | | | |--- feature_0 <= 1.50 @@ -85,7 +85,7 @@ | | | | | |--- class: 0 | | | |--- feature_2 > 4.50 | | | | |--- class: 0 -| | |--- feature_4 > 1.50 +| | |--- feature_1 > 1.50 | | | |--- class: 0 -| |--- feature_1 > 1.50 +| |--- feature_4 > 1.50 | | |--- class: 0 diff --git a/decision_tree/tree_dataset.csv b/decision_tree/tree_dataset.csv new file mode 100644 index 0000000..d3cefbc --- /dev/null +++ b/decision_tree/tree_dataset.csv @@ -0,0 +1,1251 @@ +season;trash_type;mass;space;trash_mass;decision +1;1;1;1;1;0 +1;1;1;1;3;0 +1;1;1;1;5;0 +1;1;1;2;2;0 +1;1;1;2;4;0 +1;1;1;3;1;0 +1;1;1;3;3;0 +1;1;1;3;5;0 +1;1;1;4;2;0 +1;1;1;4;4;0 +1;1;1;5;1;0 +1;1;1;5;3;0 +1;1;1;5;5;0 +1;1;2;1;2;0 +1;1;2;1;4;0 +1;1;2;2;1;0 +1;1;2;2;3;0 +1;1;2;2;5;0 +1;1;2;3;2;0 +1;1;2;3;4;0 +1;1;2;4;1;0 +1;1;2;4;3;0 +1;1;2;4;5;0 +1;1;2;5;2;0 +1;1;2;5;4;0 +1;1;3;1;1;0 +1;1;3;1;3;0 +1;1;3;1;5;0 +1;1;3;2;2;0 +1;1;3;2;4;0 +1;1;3;3;1;0 +1;1;3;3;3;0 +1;1;3;3;5;0 +1;1;3;4;2;0 +1;1;3;4;4;0 +1;1;3;5;1;0 +1;1;3;5;3;0 +1;1;3;5;5;0 +1;1;4;1;2;0 +1;1;4;1;4;0 +1;1;4;2;1;0 +1;1;4;2;3;0 +1;1;4;2;5;0 +1;1;4;3;2;0 +1;1;4;3;4;0 +1;1;4;4;1;0 +1;1;4;4;3;0 +1;1;4;4;5;0 +1;1;4;5;2;0 +1;1;4;5;4;0 +1;1;5;1;1;0 +1;1;5;1;3;0 +1;1;5;1;5;0 +1;1;5;2;2;0 +1;1;5;2;4;0 +1;1;5;3;1;0 +1;1;5;3;3;0 +1;1;5;3;5;0 +1;1;5;4;2;0 +1;1;5;4;4;0 +1;1;5;5;1;0 +1;1;5;5;3;0 +1;1;5;5;5;0 +1;2;1;1;2;0 +1;2;1;1;4;0 +1;2;1;2;1;0 +1;2;1;2;3;0 +1;2;1;2;5;0 +1;2;1;3;2;0 +1;2;1;3;4;0 +1;2;1;4;1;0 +1;2;1;4;3;0 +1;2;1;4;5;0 +1;2;1;5;2;0 +1;2;1;5;4;0 +1;2;2;1;1;0 +1;2;2;1;3;0 +1;2;2;1;5;0 +1;2;2;2;2;0 +1;2;2;2;4;0 +1;2;2;3;1;0 +1;2;2;3;3;0 +1;2;2;3;5;0 +1;2;2;4;2;0 +1;2;2;4;4;0 +1;2;2;5;1;0 +1;2;2;5;3;0 +1;2;2;5;5;0 +1;2;3;1;2;0 +1;2;3;1;4;0 +1;2;3;2;1;0 +1;2;3;2;3;0 +1;2;3;2;5;0 +1;2;3;3;2;0 +1;2;3;3;4;0 +1;2;3;4;1;0 +1;2;3;4;3;0 +1;2;3;4;5;0 +1;2;3;5;2;0 +1;2;3;5;4;0 +1;2;4;1;1;0 +1;2;4;1;3;0 +1;2;4;1;5;0 +1;2;4;2;2;0 +1;2;4;2;4;0 +1;2;4;3;1;0 +1;2;4;3;3;0 +1;2;4;3;5;0 +1;2;4;4;2;0 +1;2;4;4;4;0 +1;2;4;5;1;0 +1;2;4;5;3;0 +1;2;4;5;5;0 +1;2;5;1;2;0 +1;2;5;1;4;0 +1;2;5;2;1;0 +1;2;5;2;3;0 +1;2;5;2;5;0 +1;2;5;3;2;0 +1;2;5;3;4;0 +1;2;5;4;1;0 +1;2;5;4;3;0 +1;2;5;4;5;0 +1;2;5;5;2;0 +1;2;5;5;4;0 +1;3;1;1;1;0 +1;3;1;1;3;0 +1;3;1;1;5;0 +1;3;1;2;2;0 +1;3;1;2;4;0 +1;3;1;3;1;0 +1;3;1;3;3;0 +1;3;1;3;5;0 +1;3;1;4;2;0 +1;3;1;4;4;0 +1;3;1;5;1;0 +1;3;1;5;3;0 +1;3;1;5;5;0 +1;3;2;1;2;0 +1;3;2;1;4;0 +1;3;2;2;1;0 +1;3;2;2;3;0 +1;3;2;2;5;0 +1;3;2;3;2;0 +1;3;2;3;4;0 +1;3;2;4;1;0 +1;3;2;4;3;0 +1;3;2;4;5;0 +1;3;2;5;2;0 +1;3;2;5;4;0 +1;3;3;1;1;0 +1;3;3;1;3;0 +1;3;3;1;5;0 +1;3;3;2;2;0 +1;3;3;2;4;0 +1;3;3;3;1;0 +1;3;3;3;3;0 +1;3;3;3;5;0 +1;3;3;4;2;0 +1;3;3;4;4;0 +1;3;3;5;1;0 +1;3;3;5;3;0 +1;3;3;5;5;0 +1;3;4;1;2;0 +1;3;4;1;4;0 +1;3;4;2;1;0 +1;3;4;2;3;0 +1;3;4;2;5;0 +1;3;4;3;2;0 +1;3;4;3;4;0 +1;3;4;4;1;0 +1;3;4;4;3;0 +1;3;4;4;5;0 +1;3;4;5;2;0 +1;3;4;5;4;0 +1;3;5;1;1;0 +1;3;5;1;3;0 +1;3;5;1;5;0 +1;3;5;2;2;0 +1;3;5;2;4;0 +1;3;5;3;1;0 +1;3;5;3;3;0 +1;3;5;3;5;0 +1;3;5;4;2;0 +1;3;5;4;4;0 +1;3;5;5;1;0 +1;3;5;5;3;0 +1;3;5;5;5;0 +1;4;1;1;2;0 +1;4;1;1;4;0 +1;4;1;2;1;0 +1;4;1;2;3;0 +1;4;1;2;5;0 +1;4;1;3;2;0 +1;4;1;3;4;0 +1;4;1;4;1;0 +1;4;1;4;3;0 +1;4;1;4;5;0 +1;4;1;5;2;0 +1;4;1;5;4;0 +1;4;2;1;1;0 +1;4;2;1;3;0 +1;4;2;1;5;0 +1;4;2;2;2;0 +1;4;2;2;4;0 +1;4;2;3;1;0 +1;4;2;3;3;0 +1;4;2;3;5;0 +1;4;2;4;2;0 +1;4;2;4;4;0 +1;4;2;5;1;0 +1;4;2;5;3;0 +1;4;2;5;5;0 +1;4;3;1;2;0 +1;4;3;1;4;0 +1;4;3;2;1;0 +1;4;3;2;3;0 +1;4;3;2;5;0 +1;4;3;3;2;0 +1;4;3;3;4;0 +1;4;3;4;1;0 +1;4;3;4;3;0 +1;4;3;4;5;0 +1;4;3;5;2;0 +1;4;3;5;4;0 +1;4;4;1;1;0 +1;4;4;1;3;0 +1;4;4;1;5;0 +1;4;4;2;2;0 +1;4;4;2;4;0 +1;4;4;3;1;0 +1;4;4;3;3;0 +1;4;4;3;5;0 +1;4;4;4;2;0 +1;4;4;4;4;0 +1;4;4;5;1;0 +1;4;4;5;3;0 +1;4;4;5;5;0 +1;4;5;1;2;0 +1;4;5;1;4;0 +1;4;5;2;1;0 +1;4;5;2;3;0 +1;4;5;2;5;0 +1;4;5;3;2;0 +1;4;5;3;4;0 +1;4;5;4;1;0 +1;4;5;4;3;0 +1;4;5;4;5;0 +1;4;5;5;2;0 +1;4;5;5;4;0 +1;5;1;1;1;0 +1;5;1;1;3;0 +1;5;1;1;5;0 +1;5;1;2;2;0 +1;5;1;2;4;0 +1;5;1;3;1;0 +1;5;1;3;3;0 +1;5;1;3;5;0 +1;5;1;4;2;0 +1;5;1;4;4;0 +1;5;1;5;1;0 +1;5;1;5;3;0 +1;5;1;5;5;0 +1;5;2;1;2;0 +1;5;2;1;4;0 +1;5;2;2;1;0 +1;5;2;2;3;0 +1;5;2;2;5;0 +1;5;2;3;2;0 +1;5;2;3;4;0 +1;5;2;4;1;0 +1;5;2;4;3;0 +1;5;2;4;5;0 +1;5;2;5;2;0 +1;5;2;5;4;0 +1;5;3;1;1;0 +1;5;3;1;3;0 +1;5;3;1;5;0 +1;5;3;2;2;0 +1;5;3;2;4;0 +1;5;3;3;1;0 +1;5;3;3;3;0 +1;5;3;3;5;0 +1;5;3;4;2;0 +1;5;3;4;4;0 +1;5;3;5;1;0 +1;5;3;5;3;0 +1;5;3;5;5;0 +1;5;4;1;2;0 +1;5;4;1;4;0 +1;5;4;2;1;0 +1;5;4;2;3;0 +1;5;4;2;5;0 +1;5;4;3;2;0 +1;5;4;3;4;0 +1;5;4;4;1;0 +1;5;4;4;3;0 +1;5;4;4;5;0 +1;5;4;5;2;0 +1;5;4;5;4;0 +1;5;5;1;1;0 +1;5;5;1;3;0 +1;5;5;1;5;0 +1;5;5;2;2;0 +1;5;5;2;4;0 +1;5;5;3;1;0 +1;5;5;3;3;0 +1;5;5;3;5;0 +1;5;5;4;2;0 +1;5;5;4;4;0 +1;5;5;5;1;0 +1;5;5;5;3;0 +1;5;5;5;5;0 +2;1;1;1;2;1 +2;1;1;1;4;1 +2;1;1;2;1;1 +2;1;1;2;3;1 +2;1;1;2;5;0 +2;1;1;3;2;1 +2;1;1;3;4;1 +2;1;1;4;1;1 +2;1;1;4;3;1 +2;1;1;4;5;0 +2;1;1;5;2;0 +2;1;1;5;4;0 +2;1;2;1;1;1 +2;1;2;1;3;1 +2;1;2;1;5;0 +2;1;2;2;2;1 +2;1;2;2;4;0 +2;1;2;3;1;1 +2;1;2;3;3;1 +2;1;2;3;5;0 +2;1;2;4;2;1 +2;1;2;4;4;0 +2;1;2;5;1;0 +2;1;2;5;3;0 +2;1;2;5;5;0 +2;1;3;1;2;1 +2;1;3;1;4;0 +2;1;3;2;1;1 +2;1;3;2;3;0 +2;1;3;2;5;0 +2;1;3;3;2;1 +2;1;3;3;4;0 +2;1;3;4;1;1 +2;1;3;4;3;0 +2;1;3;4;5;0 +2;1;3;5;2;0 +2;1;3;5;4;0 +2;1;4;1;1;1 +2;1;4;1;3;0 +2;1;4;1;5;0 +2;1;4;2;2;0 +2;1;4;2;4;0 +2;1;4;3;1;1 +2;1;4;3;3;0 +2;1;4;3;5;0 +2;1;4;4;2;0 +2;1;4;4;4;0 +2;1;4;5;1;0 +2;1;4;5;3;0 +2;1;4;5;5;0 +2;1;5;1;2;0 +2;1;5;1;4;0 +2;1;5;2;1;0 +2;1;5;2;3;0 +2;1;5;2;5;0 +2;1;5;3;2;0 +2;1;5;3;4;0 +2;1;5;4;1;0 +2;1;5;4;3;0 +2;1;5;4;5;0 +2;1;5;5;2;0 +2;1;5;5;4;0 +2;2;1;1;1;1 +2;2;1;1;3;1 +2;2;1;1;5;0 +2;2;1;2;2;1 +2;2;1;2;4;1 +2;2;1;3;1;1 +2;2;1;3;3;1 +2;2;1;3;5;0 +2;2;1;4;2;0 +2;2;1;4;4;0 +2;2;1;5;1;0 +2;2;1;5;3;0 +2;2;1;5;5;0 +2;2;2;1;2;1 +2;2;2;1;4;0 +2;2;2;2;1;1 +2;2;2;2;3;1 +2;2;2;2;5;0 +2;2;2;3;2;1 +2;2;2;3;4;0 +2;2;2;4;1;0 +2;2;2;4;3;0 +2;2;2;4;5;0 +2;2;2;5;2;0 +2;2;2;5;4;0 +2;2;3;1;1;1 +2;2;3;1;3;0 +2;2;3;1;5;0 +2;2;3;2;2;1 +2;2;3;2;4;0 +2;2;3;3;1;1 +2;2;3;3;3;0 +2;2;3;3;5;0 +2;2;3;4;2;0 +2;2;3;4;4;0 +2;2;3;5;1;0 +2;2;3;5;3;0 +2;2;3;5;5;0 +2;2;4;1;2;0 +2;2;4;1;4;0 +2;2;4;2;1;0 +2;2;4;2;3;0 +2;2;4;2;5;0 +2;2;4;3;2;0 +2;2;4;3;4;0 +2;2;4;4;1;0 +2;2;4;4;3;0 +2;2;4;4;5;0 +2;2;4;5;2;0 +2;2;4;5;4;0 +2;2;5;1;1;0 +2;2;5;1;3;0 +2;2;5;1;5;0 +2;2;5;2;2;0 +2;2;5;2;4;0 +2;2;5;3;1;0 +2;2;5;3;3;0 +2;2;5;3;5;0 +2;2;5;4;2;0 +2;2;5;4;4;0 +2;2;5;5;1;0 +2;2;5;5;3;0 +2;2;5;5;5;0 +2;3;1;1;2;1 +2;3;1;1;4;1 +2;3;1;2;1;1 +2;3;1;2;3;1 +2;3;1;2;5;0 +2;3;1;3;2;1 +2;3;1;3;4;1 +2;3;1;4;1;0 +2;3;1;4;3;0 +2;3;1;4;5;0 +2;3;1;5;2;0 +2;3;1;5;4;0 +2;3;2;1;1;1 +2;3;2;1;3;1 +2;3;2;1;5;0 +2;3;2;2;2;1 +2;3;2;2;4;0 +2;3;2;3;1;1 +2;3;2;3;3;1 +2;3;2;3;5;0 +2;3;2;4;2;0 +2;3;2;4;4;0 +2;3;2;5;1;0 +2;3;2;5;3;0 +2;3;2;5;5;0 +2;3;3;1;2;1 +2;3;3;1;4;0 +2;3;3;2;1;1 +2;3;3;2;3;0 +2;3;3;2;5;0 +2;3;3;3;2;1 +2;3;3;3;4;0 +2;3;3;4;1;0 +2;3;3;4;3;0 +2;3;3;4;5;0 +2;3;3;5;2;0 +2;3;3;5;4;0 +2;3;4;1;1;0 +2;3;4;1;3;0 +2;3;4;1;5;0 +2;3;4;2;2;0 +2;3;4;2;4;0 +2;3;4;3;1;0 +2;3;4;3;3;0 +2;3;4;3;5;0 +2;3;4;4;2;0 +2;3;4;4;4;0 +2;3;4;5;1;0 +2;3;4;5;3;0 +2;3;4;5;5;0 +2;3;5;1;2;0 +2;3;5;1;4;0 +2;3;5;2;1;0 +2;3;5;2;3;0 +2;3;5;2;5;0 +2;3;5;3;2;0 +2;3;5;3;4;0 +2;3;5;4;1;0 +2;3;5;4;3;0 +2;3;5;4;5;0 +2;3;5;5;2;0 +2;3;5;5;4;0 +2;4;1;1;1;1 +2;4;1;1;3;1 +2;4;1;1;5;0 +2;4;1;2;2;1 +2;4;1;2;4;1 +2;4;1;3;1;1 +2;4;1;3;3;1 +2;4;1;3;5;0 +2;4;1;4;2;0 +2;4;1;4;4;0 +2;4;1;5;1;0 +2;4;1;5;3;0 +2;4;1;5;5;0 +2;4;2;1;2;1 +2;4;2;1;4;0 +2;4;2;2;1;1 +2;4;2;2;3;1 +2;4;2;2;5;0 +2;4;2;3;2;1 +2;4;2;3;4;0 +2;4;2;4;1;0 +2;4;2;4;3;0 +2;4;2;4;5;0 +2;4;2;5;2;0 +2;4;2;5;4;0 +2;4;3;1;1;1 +2;4;3;1;3;0 +2;4;3;1;5;0 +2;4;3;2;2;1 +2;4;3;2;4;0 +2;4;3;3;1;1 +2;4;3;3;3;0 +2;4;3;3;5;0 +2;4;3;4;2;0 +2;4;3;4;4;0 +2;4;3;5;1;0 +2;4;3;5;3;0 +2;4;3;5;5;0 +2;4;4;1;2;0 +2;4;4;1;4;0 +2;4;4;2;1;0 +2;4;4;2;3;0 +2;4;4;2;5;0 +2;4;4;3;2;0 +2;4;4;3;4;0 +2;4;4;4;1;0 +2;4;4;4;3;0 +2;4;4;4;5;0 +2;4;4;5;2;0 +2;4;4;5;4;0 +2;4;5;1;1;0 +2;4;5;1;3;0 +2;4;5;1;5;0 +2;4;5;2;2;0 +2;4;5;2;4;0 +2;4;5;3;1;0 +2;4;5;3;3;0 +2;4;5;3;5;0 +2;4;5;4;2;0 +2;4;5;4;4;0 +2;4;5;5;1;0 +2;4;5;5;3;0 +2;4;5;5;5;0 +2;5;1;1;2;1 +2;5;1;1;4;1 +2;5;1;2;1;1 +2;5;1;2;3;1 +2;5;1;2;5;0 +2;5;1;3;2;1 +2;5;1;3;4;1 +2;5;1;4;1;0 +2;5;1;4;3;0 +2;5;1;4;5;0 +2;5;1;5;2;0 +2;5;1;5;4;0 +2;5;2;1;1;1 +2;5;2;1;3;1 +2;5;2;1;5;0 +2;5;2;2;2;1 +2;5;2;2;4;0 +2;5;2;3;1;1 +2;5;2;3;3;1 +2;5;2;3;5;0 +2;5;2;4;2;0 +2;5;2;4;4;0 +2;5;2;5;1;0 +2;5;2;5;3;0 +2;5;2;5;5;0 +2;5;3;1;2;1 +2;5;3;1;4;0 +2;5;3;2;1;1 +2;5;3;2;3;0 +2;5;3;2;5;0 +2;5;3;3;2;1 +2;5;3;3;4;0 +2;5;3;4;1;0 +2;5;3;4;3;0 +2;5;3;4;5;0 +2;5;3;5;2;0 +2;5;3;5;4;0 +2;5;4;1;1;0 +2;5;4;1;3;0 +2;5;4;1;5;0 +2;5;4;2;2;0 +2;5;4;2;4;0 +2;5;4;3;1;0 +2;5;4;3;3;0 +2;5;4;3;5;0 +2;5;4;4;2;0 +2;5;4;4;4;0 +2;5;4;5;1;0 +2;5;4;5;3;0 +2;5;4;5;5;0 +2;5;5;1;2;0 +2;5;5;1;4;0 +2;5;5;2;1;0 +2;5;5;2;3;0 +2;5;5;2;5;0 +2;5;5;3;2;0 +2;5;5;3;4;0 +2;5;5;4;1;0 +2;5;5;4;3;0 +2;5;5;4;5;0 +2;5;5;5;2;0 +2;5;5;5;4;0 +3;1;1;1;1;1 +3;1;1;1;3;1 +3;1;1;1;5;0 +3;1;1;2;2;1 +3;1;1;2;4;1 +3;1;1;3;1;1 +3;1;1;3;3;1 +3;1;1;3;5;0 +3;1;1;4;2;1 +3;1;1;4;4;1 +3;1;1;5;1;0 +3;1;1;5;3;0 +3;1;1;5;5;0 +3;1;2;1;2;1 +3;1;2;1;4;0 +3;1;2;2;1;1 +3;1;2;2;3;1 +3;1;2;2;5;0 +3;1;2;3;2;1 +3;1;2;3;4;0 +3;1;2;4;1;1 +3;1;2;4;3;1 +3;1;2;4;5;0 +3;1;2;5;2;0 +3;1;2;5;4;0 +3;1;3;1;1;1 +3;1;3;1;3;0 +3;1;3;1;5;0 +3;1;3;2;2;1 +3;1;3;2;4;0 +3;1;3;3;1;1 +3;1;3;3;3;0 +3;1;3;3;5;0 +3;1;3;4;2;1 +3;1;3;4;4;0 +3;1;3;5;1;0 +3;1;3;5;3;0 +3;1;3;5;5;0 +3;1;4;1;2;0 +3;1;4;1;4;0 +3;1;4;2;1;1 +3;1;4;2;3;0 +3;1;4;2;5;0 +3;1;4;3;2;0 +3;1;4;3;4;0 +3;1;4;4;1;1 +3;1;4;4;3;0 +3;1;4;4;5;0 +3;1;4;5;2;0 +3;1;4;5;4;0 +3;1;5;1;1;0 +3;1;5;1;3;0 +3;1;5;1;5;0 +3;1;5;2;2;0 +3;1;5;2;4;0 +3;1;5;3;1;0 +3;1;5;3;3;0 +3;1;5;3;5;0 +3;1;5;4;2;0 +3;1;5;4;4;0 +3;1;5;5;1;0 +3;1;5;5;3;0 +3;1;5;5;5;0 +3;2;1;1;2;1 +3;2;1;1;4;1 +3;2;1;2;1;1 +3;2;1;2;3;1 +3;2;1;2;5;0 +3;2;1;3;2;1 +3;2;1;3;4;1 +3;2;1;4;1;1 +3;2;1;4;3;1 +3;2;1;4;5;0 +3;2;1;5;2;0 +3;2;1;5;4;0 +3;2;2;1;1;1 +3;2;2;1;3;1 +3;2;2;1;5;0 +3;2;2;2;2;1 +3;2;2;2;4;0 +3;2;2;3;1;1 +3;2;2;3;3;1 +3;2;2;3;5;0 +3;2;2;4;2;1 +3;2;2;4;4;0 +3;2;2;5;1;0 +3;2;2;5;3;0 +3;2;2;5;5;0 +3;2;3;1;2;1 +3;2;3;1;4;0 +3;2;3;2;1;1 +3;2;3;2;3;0 +3;2;3;2;5;0 +3;2;3;3;2;1 +3;2;3;3;4;0 +3;2;3;4;1;1 +3;2;3;4;3;0 +3;2;3;4;5;0 +3;2;3;5;2;0 +3;2;3;5;4;0 +3;2;4;1;1;0 +3;2;4;1;3;0 +3;2;4;1;5;0 +3;2;4;2;2;0 +3;2;4;2;4;0 +3;2;4;3;1;0 +3;2;4;3;3;0 +3;2;4;3;5;0 +3;2;4;4;2;0 +3;2;4;4;4;0 +3;2;4;5;1;0 +3;2;4;5;3;0 +3;2;4;5;5;0 +3;2;5;1;2;0 +3;2;5;1;4;0 +3;2;5;2;1;0 +3;2;5;2;3;0 +3;2;5;2;5;0 +3;2;5;3;2;0 +3;2;5;3;4;0 +3;2;5;4;1;0 +3;2;5;4;3;0 +3;2;5;4;5;0 +3;2;5;5;2;0 +3;2;5;5;4;0 +3;3;1;1;1;1 +3;3;1;1;3;1 +3;3;1;1;5;0 +3;3;1;2;2;1 +3;3;1;2;4;1 +3;3;1;3;1;1 +3;3;1;3;3;1 +3;3;1;3;5;0 +3;3;1;4;2;0 +3;3;1;4;4;0 +3;3;1;5;1;0 +3;3;1;5;3;0 +3;3;1;5;5;0 +3;3;2;1;2;1 +3;3;2;1;4;0 +3;3;2;2;1;1 +3;3;2;2;3;1 +3;3;2;2;5;0 +3;3;2;3;2;1 +3;3;2;3;4;0 +3;3;2;4;1;0 +3;3;2;4;3;0 +3;3;2;4;5;0 +3;3;2;5;2;0 +3;3;2;5;4;0 +3;3;3;1;1;1 +3;3;3;1;3;0 +3;3;3;1;5;0 +3;3;3;2;2;1 +3;3;3;2;4;0 +3;3;3;3;1;1 +3;3;3;3;3;0 +3;3;3;3;5;0 +3;3;3;4;2;0 +3;3;3;4;4;0 +3;3;3;5;1;0 +3;3;3;5;3;0 +3;3;3;5;5;0 +3;3;4;1;2;0 +3;3;4;1;4;0 +3;3;4;2;1;0 +3;3;4;2;3;0 +3;3;4;2;5;0 +3;3;4;3;2;0 +3;3;4;3;4;0 +3;3;4;4;1;0 +3;3;4;4;3;0 +3;3;4;4;5;0 +3;3;4;5;2;0 +3;3;4;5;4;0 +3;3;5;1;1;0 +3;3;5;1;3;0 +3;3;5;1;5;0 +3;3;5;2;2;0 +3;3;5;2;4;0 +3;3;5;3;1;0 +3;3;5;3;3;0 +3;3;5;3;5;0 +3;3;5;4;2;0 +3;3;5;4;4;0 +3;3;5;5;1;0 +3;3;5;5;3;0 +3;3;5;5;5;0 +3;4;1;1;2;1 +3;4;1;1;4;1 +3;4;1;2;1;1 +3;4;1;2;3;1 +3;4;1;2;5;0 +3;4;1;3;2;1 +3;4;1;3;4;1 +3;4;1;4;1;0 +3;4;1;4;3;0 +3;4;1;4;5;0 +3;4;1;5;2;0 +3;4;1;5;4;0 +3;4;2;1;1;1 +3;4;2;1;3;1 +3;4;2;1;5;0 +3;4;2;2;2;1 +3;4;2;2;4;0 +3;4;2;3;1;1 +3;4;2;3;3;1 +3;4;2;3;5;0 +3;4;2;4;2;0 +3;4;2;4;4;0 +3;4;2;5;1;0 +3;4;2;5;3;0 +3;4;2;5;5;0 +3;4;3;1;2;1 +3;4;3;1;4;0 +3;4;3;2;1;1 +3;4;3;2;3;0 +3;4;3;2;5;0 +3;4;3;3;2;1 +3;4;3;3;4;0 +3;4;3;4;1;0 +3;4;3;4;3;0 +3;4;3;4;5;0 +3;4;3;5;2;0 +3;4;3;5;4;0 +3;4;4;1;1;0 +3;4;4;1;3;0 +3;4;4;1;5;0 +3;4;4;2;2;0 +3;4;4;2;4;0 +3;4;4;3;1;0 +3;4;4;3;3;0 +3;4;4;3;5;0 +3;4;4;4;2;0 +3;4;4;4;4;0 +3;4;4;5;1;0 +3;4;4;5;3;0 +3;4;4;5;5;0 +3;4;5;1;2;0 +3;4;5;1;4;0 +3;4;5;2;1;0 +3;4;5;2;3;0 +3;4;5;2;5;0 +3;4;5;3;2;0 +3;4;5;3;4;0 +3;4;5;4;1;0 +3;4;5;4;3;0 +3;4;5;4;5;0 +3;4;5;5;2;0 +3;4;5;5;4;0 +3;5;1;1;1;1 +3;5;1;1;3;1 +3;5;1;1;5;0 +3;5;1;2;2;1 +3;5;1;2;4;1 +3;5;1;3;1;1 +3;5;1;3;3;1 +3;5;1;3;5;0 +3;5;1;4;2;0 +3;5;1;4;4;0 +3;5;1;5;1;0 +3;5;1;5;3;0 +3;5;1;5;5;0 +3;5;2;1;2;1 +3;5;2;1;4;0 +3;5;2;2;1;1 +3;5;2;2;3;1 +3;5;2;2;5;0 +3;5;2;3;2;1 +3;5;2;3;4;0 +3;5;2;4;1;0 +3;5;2;4;3;0 +3;5;2;4;5;0 +3;5;2;5;2;0 +3;5;2;5;4;0 +3;5;3;1;1;1 +3;5;3;1;3;0 +3;5;3;1;5;0 +3;5;3;2;2;1 +3;5;3;2;4;0 +3;5;3;3;1;1 +3;5;3;3;3;0 +3;5;3;3;5;0 +3;5;3;4;2;0 +3;5;3;4;4;0 +3;5;3;5;1;0 +3;5;3;5;3;0 +3;5;3;5;5;0 +3;5;4;1;2;0 +3;5;4;1;4;0 +3;5;4;2;1;0 +3;5;4;2;3;0 +3;5;4;2;5;0 +3;5;4;3;2;0 +3;5;4;3;4;0 +3;5;4;4;1;0 +3;5;4;4;3;0 +3;5;4;4;5;0 +3;5;4;5;2;0 +3;5;4;5;4;0 +3;5;5;1;1;0 +3;5;5;1;3;0 +3;5;5;1;5;0 +3;5;5;2;2;0 +3;5;5;2;4;0 +3;5;5;3;1;0 +3;5;5;3;3;0 +3;5;5;3;5;0 +3;5;5;4;2;0 +3;5;5;4;4;0 +3;5;5;5;1;0 +3;5;5;5;3;0 +3;5;5;5;5;0 +4;1;1;1;2;1 +4;1;1;1;4;1 +4;1;1;2;1;1 +4;1;1;2;3;1 +4;1;1;2;5;0 +4;1;1;3;2;1 +4;1;1;3;4;1 +4;1;1;4;1;1 +4;1;1;4;3;1 +4;1;1;4;5;0 +4;1;1;5;2;0 +4;1;1;5;4;0 +4;1;2;1;1;1 +4;1;2;1;3;1 +4;1;2;1;5;0 +4;1;2;2;2;1 +4;1;2;2;4;0 +4;1;2;3;1;1 +4;1;2;3;3;1 +4;1;2;3;5;0 +4;1;2;4;2;1 +4;1;2;4;4;0 +4;1;2;5;1;0 +4;1;2;5;3;0 +4;1;2;5;5;0 +4;1;3;1;2;1 +4;1;3;1;4;0 +4;1;3;2;1;1 +4;1;3;2;3;0 +4;1;3;2;5;0 +4;1;3;3;2;1 +4;1;3;3;4;0 +4;1;3;4;1;1 +4;1;3;4;3;0 +4;1;3;4;5;0 +4;1;3;5;2;0 +4;1;3;5;4;0 +4;1;4;1;1;1 +4;1;4;1;3;0 +4;1;4;1;5;0 +4;1;4;2;2;0 +4;1;4;2;4;0 +4;1;4;3;1;1 +4;1;4;3;3;0 +4;1;4;3;5;0 +4;1;4;4;2;0 +4;1;4;4;4;0 +4;1;4;5;1;0 +4;1;4;5;3;0 +4;1;4;5;5;0 +4;1;5;1;2;0 +4;1;5;1;4;0 +4;1;5;2;1;0 +4;1;5;2;3;0 +4;1;5;2;5;0 +4;1;5;3;2;0 +4;1;5;3;4;0 +4;1;5;4;1;0 +4;1;5;4;3;0 +4;1;5;4;5;0 +4;1;5;5;2;0 +4;1;5;5;4;0 +4;2;1;1;1;1 +4;2;1;1;3;1 +4;2;1;1;5;0 +4;2;1;2;2;1 +4;2;1;2;4;1 +4;2;1;3;1;1 +4;2;1;3;3;1 +4;2;1;3;5;0 +4;2;1;4;2;1 +4;2;1;4;4;1 +4;2;1;5;1;0 +4;2;1;5;3;0 +4;2;1;5;5;0 +4;2;2;1;2;1 +4;2;2;1;4;0 +4;2;2;2;1;1 +4;2;2;2;3;1 +4;2;2;2;5;0 +4;2;2;3;2;1 +4;2;2;3;4;0 +4;2;2;4;1;1 +4;2;2;4;3;1 +4;2;2;4;5;0 +4;2;2;5;2;0 +4;2;2;5;4;0 +4;2;3;1;1;1 +4;2;3;1;3;0 +4;2;3;1;5;0 +4;2;3;2;2;1 +4;2;3;2;4;0 +4;2;3;3;1;1 +4;2;3;3;3;0 +4;2;3;3;5;0 +4;2;3;4;2;1 +4;2;3;4;4;0 +4;2;3;5;1;0 +4;2;3;5;3;0 +4;2;3;5;5;0 +4;2;4;1;2;0 +4;2;4;1;4;0 +4;2;4;2;1;0 +4;2;4;2;3;0 +4;2;4;2;5;0 +4;2;4;3;2;0 +4;2;4;3;4;0 +4;2;4;4;1;0 +4;2;4;4;3;0 +4;2;4;4;5;0 +4;2;4;5;2;0 +4;2;4;5;4;0 +4;2;5;1;1;0 +4;2;5;1;3;0 +4;2;5;1;5;0 +4;2;5;2;2;0 +4;2;5;2;4;0 +4;2;5;3;1;0 +4;2;5;3;3;0 +4;2;5;3;5;0 +4;2;5;4;2;0 +4;2;5;4;4;0 +4;2;5;5;1;0 +4;2;5;5;3;0 +4;2;5;5;5;0 +4;3;1;1;2;1 +4;3;1;1;4;1 +4;3;1;2;1;1 +4;3;1;2;3;1 +4;3;1;2;5;0 +4;3;1;3;2;1 +4;3;1;3;4;1 +4;3;1;4;1;1 +4;3;1;4;3;1 +4;3;1;4;5;0 +4;3;1;5;2;0 +4;3;1;5;4;0 +4;3;2;1;1;1 +4;3;2;1;3;1 +4;3;2;1;5;0 +4;3;2;2;2;1 +4;3;2;2;4;0 +4;3;2;3;1;1 +4;3;2;3;3;1 +4;3;2;3;5;0 +4;3;2;4;2;1 +4;3;2;4;4;0 +4;3;2;5;1;0 +4;3;2;5;3;0 +4;3;2;5;5;0 +4;3;3;1;2;1 +4;3;3;1;4;0 +4;3;3;2;1;1 +4;3;3;2;3;0 +4;3;3;2;5;0 +4;3;3;3;2;1 +4;3;3;3;4;0 +4;3;3;4;1;1 +4;3;3;4;3;0 +4;3;3;4;5;0 +4;3;3;5;2;0 +4;3;3;5;4;0 +4;3;4;1;1;0 +4;3;4;1;3;0 +4;3;4;1;5;0 +4;3;4;2;2;0 +4;3;4;2;4;0 +4;3;4;3;1;0 +4;3;4;3;3;0 +4;3;4;3;5;0 +4;3;4;4;2;0 +4;3;4;4;4;0 +4;3;4;5;1;0 +4;3;4;5;3;0 +4;3;4;5;5;0 +4;3;5;1;2;0 +4;3;5;1;4;0 +4;3;5;2;1;0 +4;3;5;2;3;0 +4;3;5;2;5;0 +4;3;5;3;2;0 +4;3;5;3;4;0 +4;3;5;4;1;0 +4;3;5;4;3;0 +4;3;5;4;5;0 +4;3;5;5;2;0 +4;3;5;5;4;0 +4;4;1;1;1;1 +4;4;1;1;3;1 +4;4;1;1;5;0 +4;4;1;2;2;1 +4;4;1;2;4;1 +4;4;1;3;1;1 +4;4;1;3;3;1 +4;4;1;3;5;0 +4;4;1;4;2;0 +4;4;1;4;4;0 +4;4;1;5;1;0 +4;4;1;5;3;0 +4;4;1;5;5;0 +4;4;2;1;2;1 +4;4;2;1;4;0 +4;4;2;2;1;1 +4;4;2;2;3;1 +4;4;2;2;5;0 +4;4;2;3;2;1 +4;4;2;3;4;0 +4;4;2;4;1;0 +4;4;2;4;3;0 +4;4;2;4;5;0 +4;4;2;5;2;0 +4;4;2;5;4;0 +4;4;3;1;1;1 +4;4;3;1;3;0 +4;4;3;1;5;0 +4;4;3;2;2;1 +4;4;3;2;4;0 +4;4;3;3;1;1 +4;4;3;3;3;0 +4;4;3;3;5;0 +4;4;3;4;2;0 +4;4;3;4;4;0 +4;4;3;5;1;0 +4;4;3;5;3;0 +4;4;3;5;5;0 +4;4;4;1;2;0 +4;4;4;1;4;0 +4;4;4;2;1;0 +4;4;4;2;3;0 +4;4;4;2;5;0 +4;4;4;3;2;0 +4;4;4;3;4;0 +4;4;4;4;1;0 +4;4;4;4;3;0 +4;4;4;4;5;0 +4;4;4;5;2;0 +4;4;4;5;4;0 +4;4;5;1;1;0 +4;4;5;1;3;0 +4;4;5;1;5;0 +4;4;5;2;2;0 +4;4;5;2;4;0 +4;4;5;3;1;0 +4;4;5;3;3;0 +4;4;5;3;5;0 +4;4;5;4;2;0 +4;4;5;4;4;0 +4;4;5;5;1;0 +4;4;5;5;3;0 +4;4;5;5;5;0 +4;5;1;1;2;1 +4;5;1;1;4;1 +4;5;1;2;1;1 +4;5;1;2;3;1 +4;5;1;2;5;0 +4;5;1;3;2;1 +4;5;1;3;4;1 +4;5;1;4;1;0 +4;5;1;4;3;0 +4;5;1;4;5;0 +4;5;1;5;2;0 +4;5;1;5;4;0 +4;5;2;1;1;1 +4;5;2;1;3;1 +4;5;2;1;5;0 +4;5;2;2;2;1 +4;5;2;2;4;0 +4;5;2;3;1;1 +4;5;2;3;3;1 +4;5;2;3;5;0 +4;5;2;4;2;0 +4;5;2;4;4;0 +4;5;2;5;1;0 +4;5;2;5;3;0 +4;5;2;5;5;0 +4;5;3;1;2;1 +4;5;3;1;4;0 +4;5;3;2;1;1 +4;5;3;2;3;0 +4;5;3;2;5;0 +4;5;3;3;2;1 +4;5;3;3;4;0 +4;5;3;4;1;0 +4;5;3;4;3;0 +4;5;3;4;5;0 +4;5;3;5;2;0 +4;5;3;5;4;0 +4;5;4;1;1;0 +4;5;4;1;3;0 +4;5;4;1;5;0 +4;5;4;2;2;0 +4;5;4;2;4;0 +4;5;4;3;1;0 +4;5;4;3;3;0 +4;5;4;3;5;0 +4;5;4;4;2;0 +4;5;4;4;4;0 +4;5;4;5;1;0 +4;5;4;5;3;0 +4;5;4;5;5;0 +4;5;5;1;2;0 +4;5;5;1;4;0 +4;5;5;2;1;0 +4;5;5;2;3;0 +4;5;5;2;5;0 +4;5;5;3;2;0 +4;5;5;3;4;0 +4;5;5;4;1;0 +4;5;5;4;3;0 +4;5;5;4;5;0 +4;5;5;5;2;0 +4;5;5;5;4;0 \ No newline at end of file diff --git a/decision_tree/tree_model b/decision_tree/tree_model index b28c9a3..f91985d 100644 Binary files a/decision_tree/tree_model and b/decision_tree/tree_model differ diff --git a/game_objects/aiPlayer.py b/game_objects/aiPlayer.py index e79482e..f4b87f8 100644 --- a/game_objects/aiPlayer.py +++ b/game_objects/aiPlayer.py @@ -18,6 +18,7 @@ class aiPlayer(): self.player.update() self.game.draw() # print(self.player.get_actual_coords()) + def turn_left(self): change = int(self.player.rotation()) - 1 diff --git a/game_objects/trashbin.py b/game_objects/trashbin.py index f6f37da..d54f00e 100644 --- a/game_objects/trashbin.py +++ b/game_objects/trashbin.py @@ -13,7 +13,7 @@ class Trashbin(Tile): self.trash_type = randrange(5) self.mass = randrange(5) self.space = randrange(5) - self.trash_mass = randrange(5) + self.trash_mass = randrange(5) def get_coords(self): diff --git a/main.py b/main.py index fe582db..518652f 100644 --- a/main.py +++ b/main.py @@ -113,17 +113,18 @@ class Game(): self.negative_decision.append(i) self.negative_actions.append(i) - # j = 0 - # for i in self.positive_actions: - - # print(f"step {j} actions is : {i}") - # j+=1 + for i in self.positive_actions: + print('----') + print(i) + print('----') # vec = pg.math.Vector2 # for i in self.positive_actions: # self.t.startAiController(i) + # self.player.update() # self.player.pos = vec(32, 32) - + # self.player.update() + self.t.startAiController(self.positive_actions[0])