From 4e78dfdbab94d27045518620e2274885e3134c98 Mon Sep 17 00:00:00 2001
From: s464840 <barign2@st.amu.edu.pl>
Date: Thu, 12 May 2022 20:35:45 +0200
Subject: [PATCH] drzewa decyzyjne

---
 tree/data.csv | 382 ++++++++++++++++++++++++++++++++++++++++++++++++++
 tree/main.py  |  31 ++++
 tree/test.csv |   7 +
 3 files changed, 420 insertions(+)
 create mode 100644 tree/data.csv
 create mode 100644 tree/main.py
 create mode 100644 tree/test.csv

diff --git a/tree/data.csv b/tree/data.csv
new file mode 100644
index 0000000..a2c385b
--- /dev/null
+++ b/tree/data.csv
@@ -0,0 +1,382 @@
+Y,typ,pora,zwierze,smieciarka_zapelnienie,smietnik_zapelnienie,czas_smieci,w_pracy
+1,1,1,0,0,0.51,1,1
+1,1,1,0,0,0.52,1,1
+1,1,1,0,0,0.53,1,1
+1,1,1,0,0,0.55,1,1
+1,1,1,0,0,0.56,1,1
+1,1,1,0,0,0.54,1,1
+1,1,1,0,0,0.57,1,1
+1,1,1,0,0,0.58,1,1
+1,1,1,0,0,0.59,1,1
+1,1,1,0,0,0.60,1,1
+1,1,1,0,0,0.61,1,1
+1,1,1,0,0,0.62,1,1
+1,1,1,0,0,0.63,1,1
+1,1,1,0,0,0.64,1,1
+1,1,1,0,0,0.65,1,1
+1,1,1,0,0,0.66,1,1
+1,1,1,0,0,0.67,1,1
+1,1,1,0,0,0.68,1,1
+1,1,1,0,0,0.69,1,1
+1,1,1,0,0,0.70,1,1
+1,1,1,0,0,0.71,1,1
+1,1,1,0,0,0.72,1,1
+1,1,1,0,0,0.73,1,1
+1,1,1,0,0,0.74,1,1
+1,1,1,0,0,0.75,1,1
+1,1,1,0,0,0.76,1,1
+1,1,1,0,0,0.77,1,1
+1,1,1,0,0,0.78,1,1
+1,1,1,0,0,0.79,1,1
+1,1,1,0,0,0.80,1,1
+1,1,1,0,0,0.81,1,1
+1,1,1,0,0,0.82,1,1
+1,1,1,0,0,0.83,1,1
+1,1,1,0,0,0.84,1,1
+1,1,1,0,0,0.85,1,1
+1,1,1,0,0,0.86,1,1
+1,2,1,0,0,0.51,1,2
+1,2,1,0,0,0.52,1,2
+1,2,1,0,0,0.53,1,2
+1,2,1,0,0,0.55,1,2
+1,2,1,0,0,0.56,1,2
+1,2,1,0,0,0.54,1,2
+1,2,1,0,0,0.57,1,2
+1,2,1,0,0,0.58,1,2
+1,2,1,0,0,0.59,1,2
+1,2,1,0,0,0.60,1,2
+1,2,1,0,0,0.61,1,2
+1,2,1,0,0,0.62,1,2
+1,2,1,0,0,0.63,1,2
+1,2,1,0,0,0.64,1,2
+1,2,1,0,0,0.65,1,2
+1,2,1,0,0,0.66,1,2
+1,2,1,0,0,0.67,1,2
+1,2,1,0,0,0.68,1,2
+1,2,1,0,0,0.69,1,2
+1,2,1,0,0,0.70,1,2
+1,2,1,0,0,0.71,1,2
+1,2,1,0,0,0.72,1,2
+1,2,1,0,0,0.73,1,2
+1,2,1,0,0,0.74,1,2
+1,2,1,0,0,0.75,1,2
+1,2,1,0,0,0.76,1,2
+1,2,1,0,0,0.77,1,2
+1,2,1,0,0,0.78,1,2
+1,2,1,0,0,0.79,1,2
+1,2,1,0,0,0.80,1,2
+1,2,1,0,0,0.81,1,2
+1,2,1,0,0,0.82,1,1
+1,2,1,0,0,0.83,1,2
+1,2,1,0,0,0.84,1,2
+1,2,1,0,0,0.85,1,2
+1,2,1,0,0,0.86,1,2
+1,3,1,0,0,0.51,1,3
+1,3,1,0,0,0.52,1,3
+1,3,1,0,0,0.53,1,3
+1,3,1,0,0,0.55,1,3
+1,3,1,0,0,0.56,1,3
+1,3,1,0,0,0.54,1,3
+1,3,1,0,0,0.57,1,3
+1,3,1,0,0,0.58,1,3
+1,3,1,0,0,0.59,1,3
+1,3,1,0,0,0.60,1,3
+1,3,1,0,0,0.61,1,3
+1,3,1,0,0,0.62,1,3
+1,3,1,0,0,0.63,1,3
+1,3,1,0,0,0.64,1,3
+1,3,1,0,0,0.65,1,3
+1,3,1,0,0,0.66,1,3
+1,3,1,0,0,0.67,1,3
+1,3,1,0,0,0.68,1,3
+1,3,1,0,0,0.69,1,3
+1,3,1,0,0,0.70,1,3
+1,3,1,0,0,0.71,1,3
+1,3,1,0,0,0.72,1,3
+1,3,1,0,0,0.73,1,3
+1,3,1,0,0,0.74,1,3
+1,3,1,0,0,0.75,1,3
+1,3,1,0,0,0.76,1,3
+1,3,1,0,0,0.77,1,3
+1,3,1,0,0,0.78,1,3
+1,3,1,0,0,0.79,1,3
+1,3,1,0,0,0.80,1,3
+1,3,1,0,0,0.81,1,3
+1,3,1,0,0,0.82,1,3
+1,3,1,0,0,0.83,1,3
+1,3,1,0,0,0.84,1,3
+1,3,1,0,0,0.85,1,3
+1,3,1,0,0,0.86,1,3
+0,1,1,1,1,0.12,1,0
+0,2,1,1,1,0.42,2,0
+0,3,1,1,1,0.01,4,0
+0,4,1,1,1,0.33,2,0
+0,0,1,1,1,0.72,3,0
+0,1,1,0,1,0.23,2,0
+0,2,1,0,1,0.13,0,0
+0,3,1,0,1,0.97,1,0
+0,4,1,0,1,0.42,6,0
+0,0,1,0,1,1,7,1
+0,1,2,1,1,0.24,3,1
+0,2,2,1,1,0.67,4,1
+0,3,2,1,1,0.78,2,1
+0,4,2,1,1,0.87,1,1
+0,0,2,1,1,0.88,6,1
+0,1,2,0,1,0.11,3,1
+0,2,2,0,1,0.02,2,2
+0,3,2,0,1,0.42,9,2
+0,4,2,0,1,0.44,1,2
+0,0,2,0,1,0.5,2,2
+0,1,3,1,1,0.51,7,2
+0,2,3,1,1,0.75,6,2
+0,3,3,1,1,0.99,3,2
+0,4,3,1,1,0,4,3
+0,0,3,1,1,0.1,8,3
+0,1,3,0,1,0.21,3,3
+0,2,3,0,1,0.34,2,3
+0,3,3,0,1,0.61,6,5
+0,4,3,0,1,0.87,0,5
+0,0,3,0,1,0.97,1,5
+0,1,4,1,1,0.90,3,5
+0,2,4,1,1,0.01,2,3
+0,3,4,1,1,0.55,7,3
+0,4,4,1,1,0.66,6,4
+0,0,4,1,1,0.77,4,4
+0,1,4,0,1,0.73,2,4
+0,2,4,0,1,0.83,4,4
+0,3,4,0,1,0.22,3,4
+0,4,4,0,1,0.17,2,5
+0,0,4,0,1,0.14,5,5
+0,1,1,1,0,0.12,1,0
+0,2,1,1,0,0.42,2,0
+0,3,1,1,0,0.01,4,0
+0,4,1,1,0,0.33,2,0
+0,0,1,1,0,0.72,3,0
+0,1,1,0,0,0.23,2,0
+0,2,1,0,0,0.13,0,0
+1,3,1,0,0,0.97,1,3
+0,4,1,0,0,0.42,6,0
+0,0,1,0,0,1,7,1
+0,1,2,1,0,0.24,3,1
+0,2,2,1,0,0.67,4,1
+0,3,2,1,0,0.78,2,1
+0,4,2,1,0,0.87,1,1
+0,0,2,1,0,0.88,6,1
+1,1,2,0,0,0.91,3,1
+1,2,2,0,0,0.52,2,2
+0,3,2,0,0,0.42,9,2
+0,4,2,0,0,0.44,1,2
+0,0,2,0,0,0.5,2,2
+0,1,3,1,0,0.51,7,2
+0,2,3,1,0,0.75,6,2
+0,3,3,1,0,0.99,3,2
+0,4,3,1,0,0,4,3
+0,0,3,1,0,0.1,8,3
+0,1,3,0,0,0.21,3,3
+0,2,3,0,0,0.34,2,3
+0,3,3,0,0,0.61,6,5
+0,4,3,0,0,0.87,0,5
+0,0,3,0,0,0.97,1,5
+0,1,4,1,0,0.90,3,5
+0,2,4,1,0,0.01,2,3
+0,3,4,1,0,0.55,7,3
+0,4,4,1,0,0.66,6,4
+0,0,4,1,0,0.77,4,4
+0,1,4,0,0,0.73,2,4
+0,2,4,0,0,0.83,4,4
+0,3,4,0,0,0.22,3,4
+0,4,4,0,0,0.17,2,5
+0,0,4,0,0,0.14,5,5
+1,1,3,0,0,0.67,1,1
+1,2,3,0,0,0.67,1,2
+1,3,3,0,0,0.67,1,3
+1,4,3,0,0,0.67,1,4
+1,0,3,0,0,0.67,1,0
+0,1,3,0,0,0.67,1,5
+0,2,3,0,0,0.67,1,5
+0,3,3,0,0,0.67,1,5
+0,4,3,0,0,0.67,1,5
+1,1,3,0,0,0.57,1,1
+0,1,3,0,0,0.47,1,1
+0,1,3,1,0,0.67,1,1
+0,1,3,1,0,0.27,1,1
+1,1,4,0,0,0.67,1,1
+1,2,4,0,0,0.67,1,2
+1,3,4,0,0,0.67,1,3
+1,4,4,0,0,0.67,1,4
+1,0,4,0,0,0.67,1,0
+0,1,4,0,0,0.67,1,5
+0,2,4,0,0,0.67,1,5
+0,3,4,0,0,0.67,1,5
+0,4,4,0,0,0.67,1,5
+1,1,4,0,0,0.57,1,1
+0,1,4,0,0,0.47,1,1
+0,1,4,1,0,0.67,1,1
+0,1,4,1,0,0.27,1,1
+1,1,2,0,0,0.67,1,1
+1,2,2,0,0,0.67,1,2
+1,3,2,0,0,0.67,1,3
+1,4,2,0,0,0.67,1,4
+1,0,2,0,0,0.67,1,0
+0,1,2,0,0,0.67,1,5
+0,2,2,0,0,0.67,1,5
+0,3,2,0,0,0.67,1,5
+0,4,2,0,0,0.67,1,5
+1,1,2,0,0,0.57,1,1
+0,1,2,0,0,0.47,1,1
+0,1,2,1,0,0.67,1,1
+0,1,2,1,0,0.27,1,1
+1,1,1,0,0,0.67,1,1
+1,2,1,0,0,0.67,1,2
+1,3,1,0,0,0.67,1,3
+1,4,1,0,0,0.67,1,4
+1,0,1,0,0,0.67,1,0
+0,1,1,0,0,0.67,1,5
+0,2,1,0,0,0.67,1,5
+0,3,1,0,0,0.67,1,5
+0,4,1,0,0,0.67,1,5
+1,1,1,0,0,0.57,1,1
+0,1,1,0,0,0.47,1,1
+0,1,1,1,0,0.67,1,1
+0,1,1,1,0,0.27,1,1
+1,1,1,0,0,0.67,3,1
+1,2,1,0,0,0.67,3,2
+1,3,1,0,0,0.67,3,3
+1,4,1,0,0,0.67,3,4
+1,0,1,0,0,0.67,3,0
+0,1,1,0,0,0.67,3,5
+0,2,1,0,0,0.67,3,5
+0,3,1,0,0,0.67,3,5
+0,4,1,0,0,0.67,3,5
+1,1,1,0,0,0.57,3,1
+1,1,1,0,0,0.47,3,1
+0,1,1,1,0,0.67,3,1
+0,1,1,1,0,0.27,3,1
+1,1,4,0,0,0.67,3,1
+1,2,4,0,0,0.67,3,2
+1,3,4,0,0,0.67,3,3
+1,4,4,0,0,0.67,3,4
+1,0,4,0,0,0.67,3,0
+0,1,4,0,0,0.67,3,5
+0,2,4,0,0,0.67,3,5
+0,3,4,0,0,0.67,3,5
+0,4,4,0,0,0.67,3,5
+1,1,4,0,0,0.57,3,1
+0,1,4,0,0,0.47,3,1
+0,1,4,1,0,0.67,3,1
+0,1,4,1,0,0.27,3,1
+1,1,4,0,0,0.7,3,1
+1,1,4,0,0,0.51,3,1
+1,1,4,0,0,0.52,3,1
+1,1,4,0,0,0.50,3,1
+0,1,4,0,0,0.49,3,1
+1,1,1,0,0,0.47,3,1
+0,1,1,0,0,0.47,0,1
+1,1,1,0,0,0.47,1,1
+1,1,1,0,0,0.01,3,1
+0,1,1,0,0,0.01,0,1
+1,1,1,0,0,0.99,0,1
+0,1,2,0,0,0.47,0,1
+1,1,2,0,0,0.47,1,1
+1,1,2,0,0,0.01,3,1
+0,1,2,0,0,0.01,0,1
+1,1,2,0,0,0.99,0,1
+0,1,3,0,0,0.47,0,1
+1,1,3,0,0,0.47,1,1
+0,1,3,0,0,0.01,3,1
+0,1,3,0,0,0.01,0,1
+1,1,3,0,0,0.99,0,1
+0,1,3,1,0,0.47,0,1
+0,1,3,1,0,0.47,1,1
+0,1,3,1,0,0.01,3,1
+0,1,3,1,0,0.01,0,1
+0,1,3,1,0,0.99,0,1
+0,1,2,1,0,0.47,0,1
+0,1,2,1,0,0.47,1,1
+0,1,2,1,0,0.01,3,1
+0,1,2,1,0,0.01,0,1
+0,1,2,1,0,0.99,0,1
+0,2,1,0,0,0.47,0,2
+1,2,1,0,0,0.47,1,2
+1,2,1,0,0,0.01,3,2
+0,2,1,0,0,0.01,0,2
+1,2,1,0,0,0.99,0,2
+0,2,2,0,0,0.47,0,2
+1,2,2,0,0,0.47,1,2
+1,2,2,0,0,0.01,3,2
+0,2,2,0,0,0.01,0,2
+1,2,2,0,0,0.99,0,2
+0,2,3,0,0,0.47,0,2
+1,2,3,0,0,0.47,1,2
+0,2,3,0,0,0.01,3,2
+0,2,3,0,0,0.01,0,2
+1,2,3,0,0,0.99,0,2
+0,3,1,0,0,0.47,0,3
+1,3,1,0,0,0.47,1,3
+1,3,1,0,0,0.01,3,3
+0,3,1,0,0,0.01,0,3
+1,3,1,0,0,0.99,0,3
+0,3,2,0,0,0.47,0,3
+1,3,2,0,0,0.47,1,3
+1,3,2,0,0,0.01,3,3
+0,3,2,0,0,0.01,0,3
+1,3,2,0,0,0.99,0,3
+0,3,3,0,0,0.47,0,3
+1,3,3,0,0,0.47,1,3
+0,3,3,0,0,0.01,3,3
+0,3,3,0,0,0.01,0,3
+1,3,3,0,0,0.99,0,3
+0,4,1,0,0,0.47,0,4
+1,4,1,0,0,0.47,1,4
+1,4,1,0,0,0.01,3,4
+0,4,1,0,0,0.01,0,4
+1,4,1,0,0,0.99,0,4
+0,4,2,0,0,0.47,0,4
+1,4,2,0,0,0.47,1,4
+1,4,2,0,0,0.01,3,4
+0,4,2,0,0,0.01,0,4
+1,4,2,0,0,0.99,0,4
+0,4,3,0,0,0.47,0,4
+1,4,3,0,0,0.47,1,4
+0,4,3,0,0,0.01,3,4
+0,4,3,0,0,0.01,0,4
+1,4,3,0,0,0.99,0,4
+0,0,1,0,0,0.47,0,0
+1,0,1,0,0,0.47,1,0
+1,0,1,0,0,0.01,3,0
+0,0,1,0,0,0.01,0,0
+1,0,1,0,0,0.99,0,0
+0,0,2,0,0,0.47,0,0
+1,0,2,0,0,0.47,1,0
+1,0,2,0,0,0.01,3,0
+0,0,2,0,0,0.01,0,0
+1,0,2,0,0,0.99,0,0
+0,0,3,0,0,0.47,0,0
+1,0,3,0,0,0.47,1,0
+0,0,3,0,0,0.01,3,0
+0,0,3,0,0,0.01,0,0
+1,0,3,0,0,0.99,0,0
+0,4,1,0,0,0.47,0,5
+0,4,1,0,0,0.47,1,5
+0,4,1,0,0,0.01,3,5
+0,4,1,0,0,0.01,0,5
+0,4,1,0,0,0.99,0,5
+0,4,2,0,0,0.47,0,5
+0,4,2,0,0,0.47,1,5
+0,4,2,0,0,0.01,3,5
+0,4,2,0,0,0.01,0,5
+0,4,2,0,0,0.99,0,5
+0,4,3,0,0,0.47,0,5
+0,4,3,0,0,0.47,1,5
+0,4,3,0,0,0.01,3,5
+0,4,3,0,0,0.01,0,5
+0,4,3,0,0,0.99,0,5
+1,1,3,0,0,0.77,8,1
+1,2,3,0,0,0.77,8,2
+1,2,1,0,0,0.43,8,2
+0,3,3,1,0,0.43,8,2
+0,1,1,1,0,0.67,1,1
+0,2,1,1,0,0.67,1,2
+0,3,1,1,0,0.67,1,3
+0,4,1,1,0,0.67,1,4
+0,0,1,1,0,0.67,1,0
\ No newline at end of file
diff --git a/tree/main.py b/tree/main.py
new file mode 100644
index 0000000..62cf72d
--- /dev/null
+++ b/tree/main.py
@@ -0,0 +1,31 @@
+import pandas as pd
+from sklearn import tree
+
+df = pd.read_csv('data.csv')
+print(df.head())
+target = df['Y']
+X = df.drop(columns=['Y'])
+clf = tree.DecisionTreeClassifier()
+clf.fit(X, target)
+
+
+df_test = pd.read_csv('test.csv')
+print(df_test.head())
+target_test = df_test['Y']
+X_test = df_test.drop(columns=['Y'])
+pred = clf.predict(X_test)
+print(pred)
+
+print('Jedynki: ', len(df[df['Y'] == 1]))
+print('Zera: ', len(df[df['Y'] == 0]))
+
+
+#Legenda
+#czy wywiezc zmieci	1 tak 0 nie
+#typ smiecia		0-zmieszane 1-bio 2-szklo 3-plastik 4-papier
+#pora roku		1-wiosna 2-lato 3-jesien 4-zima
+#zwierze			1 jest 0 nie ma
+#smieciarka pełna	1-pełna 0-nie pełna
+#śmietnik pełny		0.XX
+#czas smieci 		liczba to ilosc dni
+#w pracy			0-zmieszane 1-bio 2-szklo 3-plastik 4-papier 5-wolne
\ No newline at end of file
diff --git a/tree/test.csv b/tree/test.csv
new file mode 100644
index 0000000..b45f65c
--- /dev/null
+++ b/tree/test.csv
@@ -0,0 +1,7 @@
+Y,typ,pora,zwierze,smieciarka_zapelnienie,smietnik_zapelnienie,czas_smieci,w_pracy
+1,4,3,0,0,0.77,8,4
+1,0,3,0,0,0.77,8,0
+0,3,3,1,0,0.43,8,2
+1,3,1,0,0,0.43,8,3
+0,2,1,1,1,0.88,1,2
+0,3,1,1,0,0.73,1,3