Merge old branches to refactor due to merge it with master #22
@ -15,7 +15,7 @@ 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')
|
||||||
|
Loading…
Reference in New Issue
Block a user