Drzewo.py

@@ -15,11 +15,11 @@ class Drzewo:
         self.tree=self.tree.fit(x.values,decision)
 
     def plotTree(self):
-        plt.figure()
+        plt.figure(figsize=(10,15))
         skltree.plot_tree(self.tree,filled=True,feature_names=atributes)
         plt.title("Drzewo decyzyjne wytrenowane na przygotowanych danych")
         plt.savefig('tree.png')
-        # plt.show()
+        #plt.show()
     def makeDecision(self,values):
         action=self.tree.predict([values]) #0- nie podlewac, 1-podlewac
         if(action==[0]):