From 9715493351f3a9bd5e46cad1319821d23bf1fcfc Mon Sep 17 00:00:00 2001
From: Jakub Chmielecki <jakchm4@st.amu.edu.pl>
Date: Mon, 22 May 2023 09:32:17 +0200
Subject: [PATCH] Adding examples and attributes

---
 main.py | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/main.py b/main.py
index 7bdb0e21..368a82cc 100644
--- a/main.py
+++ b/main.py
@@ -8,6 +8,22 @@ import blocks
 from pygame.locals import *
 from datetime import datetime
 
+examples = [
+    ['piasek', 'sucha', 'jalowa', 'żółty'],
+    ['czarnoziem', 'wilgotna', 'bogata', 'brazowa'],
+    ['rzedzina', 'wilgotna', 'bogata', 'zielona'],
+    ['gleby murszowe', 'wilgotna', 'bogata', 'szara'],
+    ['pustynne gleby', 'sucha', 'jalowa', 'pomarańczowa'],
+    ['torfowiska', 'sucha', 'jalowa', 'czerwona']
+]
+
+attributes = ['typ_gleby', 'wilgotność', 'zawartość_składników', 'kolor']
+
+# Tworzenie obiektu TreeLearn i nauka drzewa decyzyjnego
+tree_learner = TreeLearn()
+default_class = 'nieznane'
+
+tree_learner.train(examples, attributes, default_class)
 
 class TreeLearn:
     def __init__(self):