This: -made graph bigger
This commit is contained in:
parent
7a14a94b1e
commit
3713ccc8ff
@ -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]):
|
||||
|
Loading…
Reference in New Issue
Block a user