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