Merge pull request 'This: -small refactor in print' (#21) from tree_tractor_move into tree
Reviewed-on: #21
This commit is contained in:
commit
6959afe81a
@ -18,7 +18,7 @@ class Drzewo:
|
|||||||
def plotTree(self):
|
def plotTree(self):
|
||||||
plt.figure(figsize=(20,30))
|
plt.figure(figsize=(20,30))
|
||||||
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):
|
||||||
|
Loading…
Reference in New Issue
Block a user