2024-06-04 11:36:14 +02:00
1 changed files with 2 additions and 2 deletions
--- a/Drzewo.py
+++ b/Drzewo.py
@ -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')