This: -made graph bigger

This commit is contained in:
jakzar 2024-05-12 13:58:57 +02:00
parent 7a14a94b1e
commit 3713ccc8ff

View File

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