diff --git a/dataset.csv b/dataset.csv index 181b121..d70b343 100644 --- a/dataset.csv +++ b/dataset.csv @@ -1,201 +1,201 @@ plec;wiek;czas_w_pom;temp_w_pom;poziom_kurzu;poziom_oswietlenia;niebezp_towary;decyzja -2;18;40;0;2;1;0;1 -2;54;40;0;2;1;0;1 -2;25;40;0;2;1;0;1 -2;41;40;0;2;1;0;1 -2;25;41;0;2;1;0;1 -2;18;40;0;1;1;1;1 -2;54;40;0;1;1;1;1 -2;25;40;0;1;1;1;1 -2;41;40;0;1;1;1;1 -2;25;41;0;1;1;1;1 -2;18;40;0;1;1;0;0 -2;54;40;0;1;1;0;0 -2;25;40;0;1;1;0;0 -2;41;40;0;1;1;0;0 -2;25;41;0;1;1;0;0 -2;18;40;0;1;2;0;0 -2;54;40;0;1;2;0;0 -2;25;40;0;1;2;0;0 -2;41;40;0;1;2;0;0 -2;25;41;0;1;2;0;0 -1;18;40;0;1;1;0;1 -1;54;40;0;1;1;0;1 -1;25;40;0;1;1;0;1 -1;41;40;0;1;1;0;1 -1;25;41;0;1;1;0;1 -2;18;39;0;2;1;1;1 -2;54;39;0;2;1;1;1 -2;25;39;0;2;1;1;1 -2;41;39;0;2;1;1;1 -2;25;20;0;2;1;1;1 -2;18;39;0;2;2;1;1 -2;54;39;0;2;2;1;1 -2;25;39;0;2;2;1;1 -2;41;39;0;2;2;1;1 -2;25;20;0;2;2;1;1 -2;18;39;0;1;1;1;0 -2;54;39;0;1;1;1;0 -2;25;39;0;1;1;1;0 -2;41;39;0;1;1;1;0 -2;25;20;0;1;1;1;0 -2;18;39;0;1;2;1;0 -2;54;39;0;1;2;1;0 -2;25;39;0;1;2;1;0 -2;41;39;0;1;2;1;0 -2;25;20;0;1;2;1;0 -1;18;39;0;1;1;1;1 -1;54;39;0;1;1;1;1 -1;25;39;0;1;1;1;1 -1;41;39;0;1;1;1;1 -1;25;20;0;1;1;1;1 -1;18;39;0;2;1;0;0 -1;54;39;0;2;1;0;0 -1;25;39;0;2;1;0;0 -1;41;39;0;2;1;0;0 -1;25;20;0;2;1;0;0 -2;18;40;1;2;1;1;1 -2;54;40;1;2;1;1;1 -2;25;40;1;2;1;1;1 -2;41;40;1;2;1;1;1 -2;25;41;1;2;1;1;1 -2;18;40;1;2;2;1;1 -2;54;40;1;2;2;1;1 -2;25;40;1;2;2;1;1 -2;41;40;1;2;2;1;1 -2;25;41;1;2;2;1;1 -2;18;40;1;1;1;1;0 -2;54;40;1;1;1;1;0 -2;25;40;1;1;1;1;0 -2;41;40;1;1;1;1;0 -2;25;41;1;1;1;1;0 -1;18;40;1;1;1;1;1 -1;54;40;1;1;1;1;1 -1;25;40;1;1;1;1;1 -1;41;40;1;1;1;1;1 -1;25;41;1;1;1;1;1 -1;18;40;1;1;2;1;1 -1;54;40;1;1;2;1;1 -1;25;40;1;1;2;1;1 -1;41;40;1;1;2;1;1 -1;25;41;1;1;2;1;1 -1;18;40;1;2;1;0;0 -1;54;40;1;2;1;0;0 -1;25;40;1;2;1;0;0 -1;41;40;1;2;1;0;0 -1;25;41;1;2;1;0;0 -2;18;39;1;2;1;1;0 -2;54;39;1;2;1;1;0 -2;25;39;1;2;1;1;0 -2;41;39;1;2;1;1;0 -2;25;20;1;2;1;1;0 -1;18;39;1;2;1;1;0 -1;54;39;1;2;1;1;0 -1;25;39;1;2;1;1;0 -1;41;39;1;2;1;1;0 -1;25;20;1;2;1;1;0 -1;18;19;0;2;1;1;0 -1;54;19;0;2;1;1;0 -1;25;19;0;2;1;1;0 -1;41;19;0;2;1;1;0 -1;25;19;0;2;1;1;0 -2;55;40;0;2;1;0;1 -2;63;40;0;2;1;0;1 -2;57;40;0;2;1;0;1 -2;72;40;0;2;1;0;1 -2;57;41;0;2;1;0;1 -2;55;40;0;2;2;1;1 -2;63;40;0;2;2;1;1 -2;57;40;0;2;2;1;1 -2;72;40;0;2;2;1;1 -2;57;41;0;2;2;1;1 -2;55;40;0;2;2;0;0 -2;63;40;0;2;2;0;0 -2;57;40;0;2;2;0;0 -2;72;40;0;2;2;0;0 -2;57;41;0;2;2;0;0 -2;55;40;0;1;2;0;0 -2;63;40;0;1;2;0;0 -2;57;40;0;1;2;0;0 -2;72;40;0;1;2;0;0 -2;57;41;0;1;2;0;0 -1;55;40;0;2;2;0;1 -1;63;40;0;2;2;0;1 -1;57;40;0;2;2;0;1 -1;72;40;0;2;2;0;1 -1;57;41;0;2;2;0;1 -2;55;39;0;2;1;1;1 -2;63;39;0;2;1;1;1 -2;57;39;0;2;1;1;1 -2;72;39;0;2;1;1;1 -2;57;20;0;2;1;1;1 -2;55;39;0;1;1;1;1 -2;63;39;0;1;1;1;1 -2;57;39;0;1;1;1;1 -2;72;39;0;1;1;1;1 -2;57;20;0;1;1;1;1 -2;55;39;0;2;2;1;0 -2;63;39;0;2;2;1;0 -2;57;39;0;2;2;1;0 -2;72;39;0;2;2;1;0 -2;57;20;0;2;2;1;0 -2;55;39;0;1;2;1;0 -2;63;39;0;1;2;1;0 -2;57;39;0;1;2;1;0 -2;72;39;0;1;2;1;0 -2;57;20;0;1;2;1;0 -1;55;39;0;2;2;1;1 -1;63;39;0;2;2;1;1 -1;57;39;0;2;2;1;1 -1;72;39;0;2;2;1;1 -1;57;20;0;2;2;1;1 -1;55;39;0;2;1;0;0 -1;63;39;0;2;1;0;0 -1;57;39;0;2;1;0;0 -1;72;39;0;2;1;0;0 -1;57;20;0;2;1;0;0 -2;55;40;1;2;1;1;1 -2;63;40;1;2;1;1;1 -2;57;40;1;2;1;1;1 -2;72;40;1;2;1;1;1 -2;57;41;1;2;1;1;1 -2;55;40;1;1;1;1;1 -2;63;40;1;1;1;1;1 -2;57;40;1;1;1;1;1 -2;72;40;1;1;1;1;1 -2;57;41;1;1;1;1;1 -2;55;40;1;2;2;1;0 -2;63;40;1;2;2;1;0 -2;57;40;1;2;2;1;0 -2;72;40;1;2;2;1;0 -2;57;41;1;2;2;1;0 -1;55;40;1;2;2;1;1 -1;63;40;1;2;2;1;1 -1;57;40;1;2;2;1;1 -1;72;40;1;2;2;1;1 -1;57;41;1;2;2;1;1 -1;55;40;1;1;2;1;1 -1;63;40;1;1;2;1;1 -1;57;40;1;1;2;1;1 -1;72;40;1;1;2;1;1 -1;57;41;1;1;2;1;1 -1;55;40;1;2;1;0;0 -1;63;40;1;2;1;0;0 -1;57;40;1;2;1;0;0 -1;72;40;1;2;1;0;0 -1;57;41;1;2;1;0;0 -2;55;39;1;2;1;1;0 -2;63;39;1;2;1;1;0 -2;57;39;1;2;1;1;0 -2;72;39;1;2;1;1;0 -2;57;20;1;2;1;1;0 -1;55;39;1;2;1;1;0 -1;63;39;1;2;1;1;0 -1;57;39;1;2;1;1;0 -1;72;39;1;2;1;1;0 -1;57;20;1;2;1;1;0 -1;55;19;0;2;1;1;0 -1;63;19;0;2;1;1;0 -1;57;19;0;2;1;1;0 -1;72;19;0;2;1;1;0 -1;57;19;0;2;1;1;0 +2;54;40;0;60;59;0;1 +2;55;40;0;60;59;0;1 +2;25;40;0;60;59;0;1 +2;60;40;0;60;59;0;1 +2;25;50;0;60;59;0;1 +2;55;50;0;60;59;0;1 +2;54;40;0;87;32;0;1 +2;55;40;0;87;32;0;1 +2;19;51;0;71;48;0;1 +2;68;51;0;62;48;0;1 +2;54;40;0;59;59;1;1 +2;55;40;0;60;60;1;1 +2;25;40;0;59;59;1;1 +2;60;40;0;60;60;1;1 +2;25;50;0;45;45;1;1 +2;55;50;0;71;71;1;1 +2;54;40;0;33;33;1;1 +2;55;40;0;87;87;1;1 +2;19;51;0;50;50;1;1 +2;68;51;0;62;62;1;1 +2;54;40;0;59;59;0;0 +2;55;40;0;60;60;0;0 +2;25;40;0;59;59;0;0 +2;60;40;0;60;60;0;0 +2;25;50;0;45;45;0;0 +2;55;50;0;71;71;0;0 +2;54;40;0;33;33;0;0 +2;55;40;0;87;87;0;0 +2;19;51;0;50;50;0;0 +2;68;51;0;62;62;0;0 +2;54;40;0;59;60;0;0 +2;55;40;0;59;60;0;0 +2;25;40;0;59;60;0;0 +2;60;40;0;59;60;0;0 +2;25;50;0;59;60;0;0 +2;55;50;0;59;60;0;0 +2;54;40;0;32;87;0;0 +2;55;40;0;32;87;0;0 +2;19;51;0;45;71;0;0 +2;68;51;0;47;62;0;0 +1;54;40;0;59;59;0;1 +1;55;40;0;60;60;0;1 +1;25;40;0;59;59;0;1 +1;60;40;0;60;60;0;1 +1;25;50;0;45;45;0;1 +1;55;50;0;71;71;0;1 +1;54;40;0;33;33;0;1 +1;55;40;0;87;87;0;1 +1;19;51;0;50;50;0;1 +1;68;51;0;62;62;0;1 +2;54;39;0;60;59;1;1 +2;55;39;0;60;59;1;1 +2;25;39;0;60;59;1;1 +2;60;39;0;60;59;1;1 +2;25;20;0;60;59;1;1 +2;55;20;0;60;59;1;1 +2;54;39;0;87;32;1;1 +2;55;39;0;87;32;1;1 +2;19;24;0;71;48;1;1 +2;68;24;0;62;48;1;1 +2;54;39;0;60;60;1;1 +2;55;39;0;59;59;1;1 +2;25;39;0;60;60;1;1 +2;60;39;0;59;59;1;1 +2;25;20;0;71;71;1;1 +2;55;20;0;45;45;1;1 +2;54;39;0;87;87;1;1 +2;55;39;0;33;33;1;1 +2;19;24;0;62;62;1;1 +2;68;24;0;50;50;1;1 +2;54;39;0;59;59;1;0 +2;55;39;0;60;60;1;0 +2;25;39;0;59;59;1;0 +2;60;39;0;60;60;1;0 +2;25;20;0;45;45;1;0 +2;55;20;0;71;71;1;0 +2;54;39;0;33;33;1;0 +2;55;39;0;87;87;1;0 +2;19;24;0;50;50;1;0 +2;68;24;0;62;62;1;0 +2;54;39;0;59;60;1;0 +2;55;39;0;59;60;1;0 +2;25;39;0;59;60;1;0 +2;60;39;0;59;60;1;0 +2;25;20;0;59;60;1;0 +2;55;20;0;59;60;1;0 +2;54;39;0;32;87;1;0 +2;55;39;0;32;87;1;0 +2;19;24;0;45;71;1;0 +2;68;24;0;47;62;1;0 +1;54;39;0;59;59;1;1 +1;55;39;0;60;60;1;1 +1;25;39;0;59;59;1;1 +1;60;39;0;60;60;1;1 +1;25;20;0;45;45;1;1 +1;55;20;0;71;71;1;1 +1;54;39;0;33;33;1;1 +1;55;39;0;87;87;1;1 +1;19;24;0;50;50;1;1 +1;68;24;0;62;62;1;1 +1;54;39;0;60;59;0;0 +1;55;39;0;60;59;0;0 +1;25;39;0;60;59;0;0 +1;60;39;0;60;59;0;0 +1;25;20;0;60;59;0;0 +1;55;20;0;60;59;0;0 +1;54;39;0;87;32;0;0 +1;55;39;0;87;32;0;0 +1;19;24;0;71;48;0;0 +1;68;24;0;62;48;0;0 +2;54;40;1;60;59;1;1 +2;55;40;1;60;59;1;1 +2;25;40;1;60;59;1;1 +2;60;40;1;60;59;1;1 +2;25;50;1;60;59;1;1 +2;55;50;1;60;59;1;1 +2;54;40;1;87;32;1;1 +2;55;40;1;87;32;1;1 +2;19;51;1;71;48;1;1 +2;68;51;1;62;48;1;1 +2;54;40;1;60;60;1;1 +2;55;40;1;59;59;1;1 +2;25;40;1;60;60;1;1 +2;60;40;1;59;59;1;1 +2;25;50;1;71;71;1;1 +2;55;50;1;45;45;1;1 +2;54;40;1;87;87;1;1 +2;55;40;1;33;33;1;1 +2;19;51;1;62;62;1;1 +2;68;51;1;50;50;1;1 +2;54;40;1;59;59;1;0 +2;55;40;1;60;60;1;0 +2;25;40;1;59;59;1;0 +2;60;40;1;60;60;1;0 +2;25;50;1;45;45;1;0 +2;55;50;1;71;71;1;0 +2;54;40;1;33;33;1;0 +2;55;40;1;87;87;1;0 +2;19;51;1;50;50;1;0 +2;68;51;1;62;62;1;0 +1;54;40;1;59;59;1;1 +1;55;40;1;60;60;1;1 +1;25;40;1;59;59;1;1 +1;60;40;1;60;60;1;1 +1;25;50;1;45;45;1;1 +1;55;50;1;71;71;1;1 +1;54;40;1;33;33;1;1 +1;55;40;1;87;87;1;1 +1;19;51;1;50;50;1;1 +1;68;51;1;62;62;1;1 +1;54;40;1;59;60;1;1 +1;55;40;1;59;60;1;1 +1;25;40;1;59;60;1;1 +1;60;40;1;59;60;1;1 +1;25;50;1;59;60;1;1 +1;55;50;1;59;60;1;1 +1;54;40;1;32;87;1;1 +1;55;40;1;32;87;1;1 +1;19;51;1;45;71;1;1 +1;68;51;1;47;62;1;1 +1;54;40;1;60;59;0;0 +1;55;40;1;60;59;0;0 +1;25;40;1;60;59;0;0 +1;60;40;1;60;59;0;0 +1;25;50;1;60;59;0;0 +1;55;50;1;60;59;0;0 +1;54;40;1;87;32;0;0 +1;55;40;1;87;32;0;0 +1;19;51;1;71;48;0;0 +1;68;51;1;62;48;0;0 +2;54;39;1;60;59;1;0 +2;55;39;1;60;59;1;0 +2;25;39;1;60;59;1;0 +2;60;39;1;60;59;1;0 +2;25;20;1;60;59;1;0 +2;55;20;1;60;59;1;0 +2;54;39;1;87;32;1;0 +2;55;39;1;87;32;1;0 +2;19;24;1;71;48;1;0 +2;68;24;1;62;48;1;0 +1;54;39;1;60;59;1;0 +1;55;39;1;60;59;1;0 +1;25;39;1;60;59;1;0 +1;60;39;1;60;59;1;0 +1;25;20;1;60;59;1;0 +1;55;20;1;60;59;1;0 +1;54;39;1;87;32;1;0 +1;55;39;1;87;32;1;0 +1;19;24;1;71;48;1;0 +1;68;24;1;62;48;1;0 +1;54;19;0;60;59;1;0 +1;55;19;0;60;59;1;0 +1;25;19;0;60;59;1;0 +1;60;5;0;60;59;1;0 +1;25;5;0;60;59;1;0 +1;55;10;0;60;59;1;0 +1;54;18;0;87;32;1;0 +1;55;18;0;87;32;1;0 +1;19;3;0;71;48;1;0 +1;68;12;0;62;48;1;0 diff --git a/drzewo.png b/drzewo.png index e1c70a7..59239ef 100644 Binary files a/drzewo.png and b/drzewo.png differ diff --git a/rescue.py b/rescue.py index bb5aa6b..313e02e 100644 --- a/rescue.py +++ b/rescue.py @@ -32,8 +32,8 @@ def decyzja_osoba(osoba: PoleKraty, clf: DecisionTreeClassifier): z.append(0) # zimne pomieszczenie else: z.append(1) # normalne pomieszczenie - z.extend(random.choices([1, 2], weights=[7, 3], k=1)) # poziom kurzu, 2 to największy/najbardziej niebezpieczny - z.extend(random.choices([1, 2], weights=[4, 6], k=1)) # poziom oświetlenia, 2 to najlepsze oświetlenie + z.append(random.randint(20, 100)) # poziom kurzu + z.append(random.randint(20, 100)) # poziom oświetlenia if (0<=osoba.wiersz or osoba.wiersz<=13) and (17<=osoba.kolumna or osoba.kolumna<=19): #obok szafki z niebezpiecznymi towarami z.append(1) else: