DeepBlockDeepak / kaggle_titanic

Titanic Survivor Predictor: A multiple machine learning model project to forecast survival outcomes of Titanic passengers. Engineered from historical data, refined with feature selection, tested with CI/CD, and outputs validation metrics.
https://www.kaggle.com/c/titanic
1 stars 0 forks source link

Hand-roll Decision Tree Classifier Module and add Notes to Wiki #3

Closed DeepBlockDeepak closed 8 months ago

DeepBlockDeepak commented 8 months ago
  1. Add separate module which uses a basic, hand-rolled version of a decision tree classifier. Will need following methods:

    • [x] split()
    • [x] gini()
    • [x] information_gain()
    • [x] find_best_split()
    • [x] build_tree() And the following classes:
    • [x] Leaf
    • [x] Internal_Node
  2. Add Math Notes to wiki for my own clarity on internal logic.

Important Features TODO: