iiitl / iiitl/Decision-Tree

Implement Gini Impurity and information gain from scratch

Open
#5 11 comments 0 reactions 0 assignees View on GitHub
medium track: scratch
Dominant language
Jupyter Notebook
Stars
0
Forks
16
PR merge metrics
No merged PRs in 30d

Description

Write pure NumPy functions to calculate Gini Impurity for a set of labels and evaluate the Information Gain for a proposed binary split on a continuous feature. Test it on a small subset of the data and verify it correctly identifies the best split point.

Contributor guide

Open the contributing guide

Research direction

The issue names no file or test; start by locating the notebook or entry point that contains the decision-tree work and inspect the small data subset used there. Implement the pure NumPy calculations described in the issue, then verify the selected binary split point against the expected best split.

Written by the indexing model from the issue text.

Assessment

Tech stack
numpy
Domain
machine-learning
Issue type
Feature
Difficulty
3/5
Estimated time
1-2 days
Activity status
Quiet
Clarity
Mostly clear
Newbie friendliness
55/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.