Disease Classifier

Using a Naive Bayes Classifier gets possible diseases from symptoms

Example Input:

python GuessDisease.py "agitation,exhaustion"

Example Output:

('bipolar disorder', 1.2303930712136743e-07)
('delusion', 5.93049460324991e-14)
('hypothyroidism', 2.965247301624956e-14)
('paranoia', 2.965247301624956e-14)
('malignant neoplasm of lung carcinoma of lung', 2.965247301624956e-14)
('schizophrenia', 2.965247301624956e-14)
('psychotic disorder', 2.965247301624956e-14)
('confusion', 2.965247301624956e-14)
('anxiety state', 2.965247301624956e-14)
('dependence', 2.965247301624956e-14)

DATA USED

http://people.dbmi.columbia.edu/~friedma/Projects/DiseaseSymptomKB/index.html

(Dataset.csv is ready to use)(Contains 180 Diferent Diseases and their symptoms)

Requsites

sudo pip install naiveBayesClassifier
(changed some code..)

Thanks to:

https://github.com/muatik/naive-bayes-classifier

see the screenshot for example input and output