Merge pull request 'This: -small refactor in print' (#21) from tree_tractor_move into tree

Reviewed-on: #21
This commit is contained in:
s481825 2024-06-04 11:32:36 +02:00
commit 6959afe81a

View File

@ -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):