drzewa decyzyjne
This commit is contained in:
parent
404586daf8
commit
4e78dfdbab
382
tree/data.csv
Normal file
382
tree/data.csv
Normal file
@ -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
|
|
31
tree/main.py
Normal file
31
tree/main.py
Normal file
@ -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
|
7
tree/test.csv
Normal file
7
tree/test.csv
Normal file
@ -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
|
|
Loading…
Reference in New Issue
Block a user