Skip to content

Tune criterion for RF / XT #203

@dholzmueller

Description

@dholzmueller

For classification with RF/XT, we could optimize the criterion in ['gini', 'entropy']. In my tests, one could just always use 'entropy' if logloss is the target metric, but I suspect that 'gini' might be better for other target metrics.
Since the criterion needs to be different for classification and regression, one could add a hyperparameter classification_criterion which is tuned in ['gini', 'entropy'] and ignored for regression. This might require changing AutoGluon code, though.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions