dotnet / dotnet/machinelearning

Does GetFeatureWeights support categorical splits?

Open
#3,766 6 comments 0 reactions 0 assignees View on GitHub
area-Trees bug Priority:1
Dominant language
C#
Stars
9.4k
Forks
2k
Avg merge
2d 20h
Merged PRs (30d)
11

Description

Version: 1.0

I'm training my multi class LightGBM with mostly categorical features (which works great). But when I want to get the `GetFeatureWeights` from the binary predictors I get huge values for feature of index `-1`. Though, just inspecting the models in the debugger the trees almost exclusively use categorical features for the splits. It seems that the `GainMap` doesn't actually consider any categorical splits and just assigns all those gains to the index `-1` which makes the feature weights vector completely useless in my case.

Is this something that will be supported? Or am I wrong here?

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.