back button

The Custom Decision Tree Algorithm

Source: https://github.com/keaganderson/decision_tree

Description: A project done for the class TCSS 455, Intro to Machine Learning. This program takes in the Titanic Dataset, splits it into test and train datasets, and constructs a custom Decision Tree machine learning classifier for the data categories out of the train dataset. It then tests this classifier, on the test dataset calculating the probability of survival of the test people given the attributes of the person. It then compares these calculations to the actual results. Then a decision tree made with the SKLearn library is implemented for comparison.

decision tree