ID3 evolution part2

main.py

@@ -7,6 +7,8 @@ from src.Tractor import Tractor
 from src.Plant import Plant
 from src.bfs import Astar
 from src.Field import Field
+import pickle
+import os
 
 
 # pygame initialization
@@ -37,6 +39,9 @@ plant_group = pygame.sprite.Group()
 plant_group = seedForFirstTime()
 fields = return_fields_list()
 
+#ID3 TREE
+tree = pickle.load(open(os.path.join('src','tree.plk'),'rb'))
+
 #
 tractor_move = pygame.USEREVENT + 1
 pygame.time.set_timer(tractor_move, 800)
@@ -50,6 +55,7 @@ mx=int((mx+18)/36)
 my=int((my+18)/36)
 print("Destination: ", mx,my)
 tmp = WORLD_MATRIX[mx][my]
+print(tmp)
 
 moves = goal_astar.search(
     [tractor.rect.x, tractor.rect.y, directions[tractor.rotation]], destination)

src/ID3.py

@@ -1,14 +1,14 @@
 import pandas as pd
 from sklearn.tree import DecisionTreeClassifier
-import pickle
-import os
 from sklearn import tree
+import os
+import pickle
 import pydotplus
 import matplotlib.image as pltimg
 import matplotlib.pyplot as plt
 
 # Read the CSV file
-train = pd.read_csv("train_2.csv", delimiter=";")
+train = pd.read_csv("train_3.csv", delimiter=";")
 
 x_train = train.drop('czy_zebrac',axis=1)
 y_train = train['czy_zebrac']
@@ -16,21 +16,16 @@ y_train = train['czy_zebrac']
 d_tree = DecisionTreeClassifier()
 d_tree = d_tree.fit(x_train,y_train)
 
-# Save the decision tree model as a pickle file in the script's folder
-pickle.dump(d_tree, open('tree.plk', 'wb'))
-
-# Export the decision tree as DOT data
-data = tree.export_graphviz(d_tree, out_file=None)
-
-# Create a graph from the DOT data
+pickle.dump(d_tree, open(os.path.join('.','tree.plk'),'wb'))
+data = tree.export_graphviz(d_tree, out_file=None, feature_names=['Wzrost','wilgotnosc','dni_od_nawiezienia','aktualna_pogoda','czy_roslina_robaczywa','paliwo','pojemnosc_ekwipunku','cena_sprzedarzy'])
 graph = pydotplus.graph_from_dot_data(data)
 
 # Save the graph as a PNG image in the script's folder
 graph.write_png(os.path.join('.', 'mytree.png'))
 
 # Read the PNG image
-img = pltimg.imread(os.path.join('.', 'mytree.png'))
+# img = pltimg.imread(os.path.join('.', 'mytree.png'))
 
 # Display the image
-imgplot = plt.imshow(img)
-plt.show()
+# imgplot = plt.imshow(img)
+# plt.show()

src/Tractor.py

@@ -1,6 +1,9 @@
 import pygame
 from settings import block_size, tile
 
+mx=0
+my=0
+
 class Tractor(pygame.sprite.Sprite):
     def __init__(self, engine, transmission, fuel, fertilizer, capacity):
         super().__init__()
@@ -24,6 +27,8 @@ class Tractor(pygame.sprite.Sprite):
         self.transmission = transmission
         self.fuel = fuel
         self.fertilizer = fertilizer
+        
+        
 
     def movement_using_keys(self):
         keys = pygame.key.get_pressed()
@@ -47,6 +52,7 @@ class Tractor(pygame.sprite.Sprite):
             self.rect.y += block_size
         if self.rect.x > 0 and self.rotation == 270:
             self.rect.x -= block_size
+        
     
     def move_left(self):
         self.rotation -= 90
@@ -69,6 +75,7 @@ class Tractor(pygame.sprite.Sprite):
             self.image = self.left
             
     def movement(self, direction):
+        print(int((self.rect.x-18)/36),';',int((self.rect.y-18)/36))
         if direction == 'F':
             self.move_forward()
         elif direction == 'L':

src/train_2.csv

@@ -1,202 +0,0 @@
-wzrost;wilgotnosc;dni_od_nawiezienia;aktualna_pogoda;czy_roslina_robaczywa;typ_rosliny;pojemnosc_ekwipunku;cena_sprzedarzy;czy_zebrac
-54;19;15;0;0;0;90;66;0
-64;63;5;0;0;0;16;4;0
-93;0;29;0;1;0;77;73;1
-30;43;23;0;1;0;74;75;1
-48;30;10;0;0;0;39;23;0
-44;86;2;0;1;0;41;64;0
-99;74;8;0;1;0;39;37;0
-70;80;25;0;1;0;11;90;1
-62;35;2;0;1;0;53;57;0
-32;71;29;0;1;0;21;54;1
-59;27;11;0;0;0;71;68;0
-43;97;24;0;0;0;82;70;0
-24;49;1;0;0;0;22;40;0
-60;59;18;0;1;0;29;99;1
-100;87;23;0;1;0;69;55;1
-5;88;24;0;1;0;54;87;1
-35;92;17;0;1;0;30;100;1
-9;89;29;0;0;0;35;24;0
-58;7;11;0;1;0;6;62;1
-98;88;1;0;1;0;100;88;0
-14;74;7;0;1;0;71;24;0
-6;67;18;0;0;0;72;12;0
-53;41;25;0;0;0;45;56;0
-35;40;1;0;0;0;61;36;0
-93;17;15;0;1;0;63;0;1
-20;70;19;0;1;0;14;88;1
-12;57;24;0;1;0;10;90;1
-59;67;7;0;0;0;35;90;0
-79;29;8;0;1;0;59;37;0
-48;81;14;0;1;0;85;78;1
-41;32;8;0;0;0;5;26;0
-6;62;1;0;0;0;75;22;0
-18;74;5;0;1;0;9;32;0
-62;44;7;0;0;0;0;87;0
-34;19;14;0;0;0;83;92;0
-84;39;15;0;1;0;8;86;1
-55;39;4;0;1;0;31;63;0
-17;58;2;0;1;0;33;22;0
-95;12;17;0;0;0;51;16;0
-85;91;5;0;1;0;19;8;0
-62;87;1;0;1;0;22;48;0
-50;21;15;0;0;0;76;55;0
-21;84;8;0;0;0;63;87;0
-87;36;9;0;0;0;62;69;0
-60;85;22;0;0;0;98;24;0
-72;0;18;0;1;0;31;36;1
-60;41;29;0;1;0;11;60;1
-43;50;20;0;1;0;53;66;1
-56;24;10;0;1;0;91;87;1
-30;75;29;0;0;0;83;36;0
-86;62;13;0;1;0;97;26;1
-69;35;10;0;1;0;25;100;1
-98;55;11;0;0;0;66;18;0
-71;23;26;0;1;0;47;88;1
-97;88;0;0;1;0;37;55;0
-89;90;17;0;0;0;50;47;0
-84;44;1;0;0;0;99;86;0
-57;72;12;0;1;0;85;44;1
-89;65;20;0;0;0;39;16;0
-31;100;7;0;0;0;64;66;0
-45;47;3;0;1;0;76;35;0
-21;52;27;0;1;0;90;15;1
-7;30;8;0;0;0;39;70;0
-42;67;27;0;0;0;72;82;0
-88;69;28;0;1;0;85;63;1
-84;62;21;0;1;0;43;16;1
-41;62;25;0;1;0;26;24;1
-18;39;9;0;1;0;70;96;0
-75;60;16;0;1;0;37;64;1
-66;9;29;0;1;0;3;84;1
-2;92;25;0;0;0;2;19;0
-100;20;11;0;0;0;45;65;0
-21;19;8;0;1;0;82;2;0
-93;37;10;0;0;0;57;3;0
-35;12;5;0;1;0;48;96;0
-62;12;4;0;0;0;94;1;0
-82;73;12;0;1;0;66;81;1
-43;74;15;0;1;0;43;70;1
-54;76;11;0;1;0;80;88;1
-51;64;26;0;1;0;31;21;1
-14;12;12;0;1;0;97;17;1
-58;59;22;0;0;0;79;36;0
-17;48;17;0;0;0;81;72;0
-47;2;12;0;1;0;3;22;1
-38;52;13;0;0;0;31;46;0
-32;19;4;0;1;0;30;85;0
-9;28;29;0;0;0;20;46;0
-98;0;15;0;1;0;72;15;1
-95;33;0;0;1;0;33;11;0
-55;56;10;0;0;0;28;56;0
-18;22;13;0;0;0;94;90;0
-81;28;17;0;1;0;95;29;1
-96;6;6;0;0;0;14;89;0
-58;26;24;0;0;0;70;76;0
-83;33;16;0;1;0;31;54;1
-90;48;29;0;1;0;55;40;1
-65;72;4;0;1;0;70;10;0
-74;100;14;0;1;0;44;24;1
-56;25;30;0;1;0;84;16;1
-14;75;29;0;1;0;42;9;1
-90;51;2;0;1;0;86;79;0
-60;31;25;0;0;0;43;54;0
-74;1;2;0;0;0;17;76;0
-86;72;16;0;1;0;38;53;1
-70;69;19;0;0;0;40;63;0
-37;18;24;0;1;0;84;78;1
-48;77;7;0;0;0;99;42;0
-90;92;5;0;0;0;80;48;0
-99;74;9;0;1;0;41;68;0
-58;18;5;0;1;0;53;94;0
-42;22;10;0;1;0;24;17;1
-44;34;29;0;0;0;86;8;0
-79;13;0;0;0;0;31;43;0
-16;54;28;0;0;0;26;84;0
-44;44;12;0;0;0;60;94;0
-36;6;9;0;0;0;90;90;0
-33;43;5;0;1;0;5;86;0
-24;5;0;0;0;0;47;41;0
-87;67;11;0;1;0;70;32;1
-62;76;6;0;1;0;33;96;0
-48;85;25;0;1;0;41;12;1
-31;15;13;0;1;0;87;18;1
-27;69;2;0;0;0;38;2;0
-4;98;24;0;1;0;22;22;1
-61;50;20;0;0;0;14;0;0
-2;31;29;0;1;0;40;0;1
-53;19;29;0;0;0;34;100;0
-64;71;13;0;1;0;65;66;1
-86;67;15;0;1;0;69;34;1
-33;63;4;0;0;0;87;24;0
-22;84;8;0;0;0;73;13;0
-54;64;10;0;1;0;13;65;1
-51;25;4;0;1;0;26;0;0
-1;95;14;0;0;0;83;64;0
-8;32;19;0;1;0;96;77;1
-47;97;10;0;1;0;74;16;1
-30;65;24;0;0;0;83;89;0
-76;58;7;0;0;0;73;86;0
-84;41;17;0;1;0;68;78;1
-50;61;9;0;0;0;6;72;0
-37;12;8;0;1;0;11;46;0
-7;91;29;0;0;0;1;7;0
-64;69;4;0;1;0;57;28;0
-21;71;3;0;0;0;53;23;0
-58;54;13;0;0;0;19;1;0
-96;11;22;0;0;0;46;12;0
-85;41;2;0;1;0;28;24;0
-58;11;13;0;0;0;90;41;0
-49;8;30;0;0;0;63;39;0
-70;72;9;0;0;0;56;65;0
-82;39;20;0;1;0;98;55;1
-43;67;28;0;0;0;33;27;0
-70;48;6;0;1;0;58;2;0
-10;89;0;0;0;0;46;52;0
-73;18;24;0;0;0;65;40;0
-84;29;26;0;1;0;80;30;1
-32;83;0;0;0;0;68;100;0
-17;40;14;0;1;0;47;44;1
-5;77;30;0;1;0;44;65;1
-27;95;29;0;1;0;71;81;1
-27;39;15;0;1;0;22;13;1
-81;96;23;0;1;0;61;89;1
-84;58;19;0;1;0;84;62;1
-23;40;21;0;0;0;64;13;0
-57;61;20;0;0;0;84;10;0
-31;55;3;0;0;0;98;26;0
-25;33;2;0;1;0;68;75;0
-56;15;6;0;1;0;34;51;0
-45;60;20;0;0;0;9;51;0
-14;42;19;0;0;0;49;46;0
-41;100;17;0;0;0;71;18;0
-38;38;20;0;1;0;41;5;1
-73;99;17;0;0;0;99;45;0
-43;62;21;0;1;0;57;58;1
-99;31;18;0;1;0;51;52;1
-83;4;9;0;1;0;3;65;0
-53;71;26;0;0;0;93;78;0
-18;28;13;0;1;0;34;76;1
-22;29;29;0;1;0;47;68;1
-96;40;9;0;1;0;91;51;0
-24;74;9;0;1;0;57;4;0
-58;56;1;0;1;0;17;40;0
-43;39;20;0;0;0;81;77;0
-24;6;9;0;1;0;90;42;0
-32;100;16;0;0;0;56;71;0
-75;25;6;0;0;0;48;18;0
-80;96;27;0;0;0;62;84;0
-57;90;19;0;0;0;81;26;0
-1;18;12;0;0;0;4;0;0
-28;79;11;0;1;0;95;19;1
-20;4;11;0;0;0;96;64;0
-37;80;14;0;1;0;97;64;1
-2;24;23;0;0;0;63;52;0
-54;97;18;0;1;0;20;65;1
-42;44;20;0;0;0;33;20;0
-68;51;12;0;1;0;76;90;1
-93;77;24;0;1;0;41;59;1
-74;94;21;0;1;0;17;39;1
-63;63;26;0;1;0;67;2;1
-65;73;1;0;0;0;74;40;0
-100;39;19;0;0;0;41;9;0

src/train_3.csv

@@ -0,0 +1,202 @@
+Wzrost;wilgotnosc;dni_od_nawiezienia;aktualna_pogoda;czy_roslina_robaczywa;paliwo;pojemnosc_ekwipunku;cena_sprzedarzy;czy_zebrac
+23;78;12;2;1;78;88;91;1
+39;31;15;4;1;82;37;87;0
+12;41;21;2;1;62;28;33;1
+2;42;11;1;1;68;96;40;1
+43;16;20;3;0;91;23;85;0
+9;8;8;1;1;78;28;16;0
+47;0;23;2;0;94;46;45;0
+70;85;12;3;1;65;14;54;1
+34;76;15;2;0;80;13;84;0
+22;24;6;3;0;17;19;62;0
+79;9;17;4;0;94;75;80;0
+56;30;0;3;1;43;55;6;0
+45;34;12;3;1;93;80;97;1
+49;15;26;1;1;48;76;14;1
+69;16;13;4;1;10;13;28;0
+74;97;9;1;1;97;25;47;0
+37;40;19;4;0;65;81;24;0
+83;26;3;2;0;91;62;16;0
+52;37;0;3;0;93;4;89;0
+67;68;30;4;0;34;3;62;0
+96;93;18;4;0;42;5;45;0
+90;22;11;3;1;59;69;0;1
+38;10;17;4;1;57;33;1;0
+62;51;23;3;1;55;60;76;1
+64;44;5;2;1;92;85;22;0
+93;87;25;2;0;61;41;44;0
+46;45;16;4;1;85;46;1;0
+12;41;28;4;0;26;42;74;0
+40;27;23;3;0;66;100;3;0
+0;53;20;1;1;52;61;11;1
+31;10;3;1;1;97;87;99;0
+40;72;0;2;1;39;39;72;0
+19;19;27;3;1;25;68;91;1
+44;85;11;4;1;49;78;35;0
+83;60;11;3;1;56;68;83;1
+6;85;2;2;1;33;46;0;0
+99;83;18;4;0;37;81;21;0
+87;54;13;2;1;84;26;43;1
+27;96;0;1;1;20;55;80;0
+44;79;30;1;1;77;43;12;1
+14;84;26;4;0;13;13;38;0
+11;84;28;1;1;43;55;75;1
+75;10;19;3;0;94;22;37;0
+88;52;20;2;1;17;87;2;1
+38;81;14;1;0;86;91;2;0
+35;34;12;1;1;36;69;16;1
+87;19;16;3;0;66;36;56;0
+23;87;5;1;1;69;54;29;0
+66;98;17;3;1;81;82;20;1
+39;32;29;4;0;39;48;96;0
+29;7;15;4;0;57;93;38;0
+80;54;7;2;0;37;22;73;0
+33;70;19;2;1;85;80;3;1
+81;55;2;2;1;20;10;70;0
+77;9;29;2;0;10;21;91;0
+34;20;4;4;0;54;41;36;0
+16;52;28;1;0;18;33;92;0
+41;47;17;1;0;42;36;38;0
+46;25;0;3;0;19;52;7;0
+89;4;13;3;1;10;14;17;1
+88;6;26;2;0;70;27;64;0
+87;4;20;2;0;28;22;44;0
+93;78;21;2;1;94;6;70;1
+51;58;20;3;0;82;40;78;0
+99;12;21;4;1;83;18;90;0
+6;17;11;2;1;100;91;29;1
+96;50;24;3;0;39;58;91;0
+94;98;14;2;0;29;11;65;0
+79;13;21;2;1;76;85;49;1
+56;71;28;2;1;33;89;91;1
+46;19;26;4;0;52;58;3;0
+27;81;28;2;1;77;71;99;1
+6;35;22;2;1;76;60;58;1
+99;80;29;1;0;13;19;9;0
+24;23;22;2;1;10;66;80;1
+48;91;26;4;0;61;41;66;0
+67;71;21;3;1;100;99;87;1
+60;29;25;3;0;20;79;92;0
+34;65;0;4;0;76;29;62;0
+6;56;2;2;1;63;25;92;0
+77;31;27;2;1;42;51;34;1
+65;28;0;2;0;29;69;30;0
+84;75;0;2;0;23;36;59;0
+10;13;14;2;1;57;2;36;1
+48;58;8;2;1;69;56;37;0
+71;16;27;3;1;100;56;34;1
+43;97;29;2;1;30;37;5;1
+2;91;29;2;0;88;60;99;0
+70;67;19;1;0;100;55;93;0
+63;74;10;2;1;92;69;63;1
+88;69;13;4;0;27;38;45;0
+78;67;27;4;0;59;28;7;0
+58;69;28;3;0;52;23;7;0
+79;60;30;4;1;24;38;76;0
+88;98;10;1;0;71;30;37;0
+90;89;28;1;0;23;17;92;0
+31;32;29;4;1;82;45;23;0
+57;50;20;1;0;12;94;45;0
+20;88;3;2;1;44;94;63;0
+24;71;3;1;0;40;19;23;0
+50;46;2;2;0;70;96;52;0
+78;46;11;2;1;99;49;12;1
+43;98;10;3;0;12;87;85;0
+23;41;12;1;0;94;29;84;0
+98;37;8;3;0;49;99;44;0
+84;82;24;3;0;36;79;40;0
+56;78;16;3;0;34;89;87;0
+79;39;1;4;1;62;54;40;0
+92;31;17;1;1;55;15;56;1
+4;32;15;3;1;95;60;62;1
+74;28;16;1;0;52;70;54;0
+12;2;28;4;1;21;54;18;0
+43;51;16;3;0;67;45;32;0
+30;36;14;3;1;63;15;49;1
+55;46;19;4;1;81;57;18;0
+74;95;5;2;1;52;29;78;0
+66;64;13;2;0;19;84;83;0
+60;73;11;4;0;41;3;65;0
+100;90;7;3;1;54;3;80;0
+84;90;15;2;1;59;16;27;1
+74;28;20;2;0;17;95;10;0
+7;63;11;2;0;46;5;22;0
+61;79;16;3;1;84;56;31;1
+13;4;0;1;0;73;51;92;0
+94;32;3;1;1;13;4;32;0
+85;72;14;3;0;10;56;78;0
+62;70;24;1;1;44;45;9;1
+26;11;9;1;0;92;10;34;0
+78;56;1;1;0;10;99;95;0
+85;55;23;4;0;89;83;84;0
+34;83;11;1;1;30;74;100;1
+69;34;14;4;0;57;60;1;0
+91;12;23;1;0;32;43;80;0
+23;54;27;2;1;53;54;30;1
+47;80;24;1;0;23;33;74;0
+79;40;27;3;0;13;78;55;0
+29;40;11;3;0;96;48;92;0
+29;15;22;1;0;22;93;8;0
+59;28;1;2;1;22;63;51;0
+33;55;5;1;0;32;29;90;0
+88;67;2;2;1;61;93;34;0
+75;41;19;3;1;47;64;0;1
+48;53;21;3;1;39;2;73;1
+1;96;9;2;0;98;13;76;0
+10;90;22;2;0;23;12;68;0
+58;70;6;4;0;96;83;60;0
+67;71;19;3;1;90;93;62;1
+2;78;8;1;1;62;48;42;0
+97;12;19;4;1;97;70;74;0
+83;93;5;2;1;73;12;63;0
+42;49;4;3;0;39;72;80;0
+69;52;29;3;1;94;26;25;1
+60;84;0;2;0;43;70;92;0
+15;27;7;2;1;92;6;42;0
+7;54;13;3;1;68;70;63;1
+34;38;7;4;0;86;58;68;0
+43;40;15;4;1;61;83;63;0
+96;66;22;1;1;94;43;10;1
+9;26;18;2;1;41;5;11;1
+94;96;30;4;0;26;47;33;0
+41;80;8;4;0;34;1;29;0
+34;3;3;2;0;33;48;93;0
+47;73;23;3;0;53;53;29;0
+19;26;25;3;0;27;45;22;0
+43;52;14;1;1;90;32;84;1
+1;0;12;1;1;94;11;89;1
+52;22;28;2;1;100;24;55;1
+24;25;13;2;0;97;0;43;0
+68;8;1;3;1;39;71;54;0
+91;12;13;2;0;15;39;14;0
+53;55;12;1;1;53;17;34;1
+70;91;26;1;1;33;24;19;1
+1;77;14;4;1;88;96;62;0
+93;59;5;2;0;43;28;19;0
+71;27;27;4;1;33;17;80;0
+86;72;5;1;0;34;38;23;0
+15;84;29;4;0;66;71;81;0
+1;16;30;1;0;18;6;87;0
+38;34;28;3;0;60;72;15;0
+43;94;20;3;1;73;84;89;1
+28;52;5;3;0;50;0;48;0
+4;16;27;1;1;87;72;10;1
+60;1;19;2;1;41;31;69;1
+91;80;18;1;1;63;11;92;1
+59;58;5;2;1;24;71;8;0
+21;24;14;3;1;17;75;71;1
+35;61;12;3;1;31;51;53;1
+73;6;14;1;1;45;10;45;1
+17;79;12;1;0;23;35;72;0
+18;6;17;4;0;30;97;63;0
+58;81;18;3;0;14;53;97;0
+24;18;28;3;0;54;39;7;0
+92;63;8;4;0;65;38;49;0
+1;46;6;4;0;29;14;49;0
+83;95;7;1;1;65;99;52;0
+77;25;24;1;1;25;41;56;1
+6;96;7;2;1;91;24;84;0
+17;13;24;4;1;84;99;71;0
+6;60;11;1;0;25;6;95;0
+72;14;0;1;0;82;66;47;0
+30;39;21;3;1;100;48;28;1