Decision tree for HIV data using sklearn