diff --git a/Condition.py b/Condition.py index 07e321e..52594ac 100644 --- a/Condition.py +++ b/Condition.py @@ -26,7 +26,7 @@ class Condition: return random.randint(0,max-1) def cycle(self): - if(self.clock==12): + if(self.clock==11): self.currentTime=0 self.rain=self.setRandomRain() self.temperature=self.setRandomTemperature() diff --git a/Drzewo.py b/Drzewo.py index 9e36877..192e2f6 100644 --- a/Drzewo.py +++ b/Drzewo.py @@ -16,7 +16,7 @@ class Drzewo: self.tree=self.tree.fit(x.values,decision) def plotTree(self): - plt.figure(figsize=(10,15)) + plt.figure(figsize=(20,30)) skltree.plot_tree(self.tree,filled=True,feature_names=atributes) plt.title("Drzewo decyzyjne wytrenowane na przygotowanych danych") plt.savefig('tree.png')