Skip to content

Add classification decision tree with support for entropy and gini index #14466

@anikchand461

Description

@anikchand461

Feature description

The existing decision tree implementation supports only regression using mean squared error. This PR introduces a classification decision tree that supports multiple splitting criteria, including entropy (information gain) and gini index. This enhances flexibility and aligns with standard decision tree implementations. The code will include type hints, doctests, and proper validation.

Metadata

Metadata

Assignees

No one assigned

    Labels

    enhancementThis PR modified some existing files

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions