Przewidywanie choroby serca


Klasyfikacja binarna w oparciu o drzewo decyzyjne. Program w zależności od wybranych zmiennych, tj: spoczynkowe ciśnienie krwi, cholesterol i cukier we krwi przewiduje czy wystąpi ryzyko zapadnięcia na chorobę serca.

Podany program powstał jedynie w celach edukacyjnych.

Do stworzenia logiki drzewa decyzyjnego została użyta biblioteka Scikit-learn. W podejmowaniu decyzji biorą udział takie dane jak: wiek, płeć, rodzaj bólu klatki piersiowej, spoczynkowe ciśnienie krwi, cholesterol, cukier we krwi na czczo, spoczynkowe EKG, maksymalne tętno oraz czy występuje duszność spowodowana wysiłkiem.

Następnie drzewo o głębokości równej trzy przewiduje ryzyko wystąpienia choroby.


Komentarze