diff --git a/Drzewo.py b/Drzewo.py index 706312b..5cc95ca 100644 --- a/Drzewo.py +++ b/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]):