Skip to content

In this case study, a decision tree is build to predict the income of a given population, which is labelled as <= 50๐พ๐‘Ž๐‘›๐‘‘> 50K on the basis of various attributes (predictors) like age, working class type, marital status, gender, race etc.

Notifications You must be signed in to change notification settings

Kunika01-dev/Decision_Tree

Folders and files

NameName
Last commit message
Last commit date

Latest commit

ย 

History

3 Commits
ย 
ย 
ย 
ย 
ย 
ย 

Repository files navigation

Problem Statement :Income Prediction

In decision analysis, a decision tree can be used to visually and explicitly represent decisions and decision making.
As the name goes, it uses a tree-like model of decisions.


A decision tree is a flowchart like tree structure, where each internal node denotes a test on an attribute, each branch represents an outcome of the test, and each leaf node (terminal node) holds a class label.

Below are the important sections:

โœ”๏ธBuilding a decision tree to predict the income of a given population, which is labelled as <= 50๐พ๐‘Ž๐‘›๐‘‘> 50K.
โœ”๏ธThe attributes (predictors) are age, working class type, marital status, gender, race etc.
โœ”๏ธData cleaning and Data modeling
โœ”๏ธBuilding a decision tree with default hyperparameters
โœ”๏ธConsidering all the hyperparameters that can be tuned
โœ”๏ธChooseing the optimal hyperparameters using grid search cross-validation.
โœ”๏ธPruning of decision tree for considering relevant split/nodes.

About

In this case study, a decision tree is build to predict the income of a given population, which is labelled as <= 50๐พ๐‘Ž๐‘›๐‘‘> 50K on the basis of various attributes (predictors) like age, working class type, marital status, gender, race etc.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published