dotnet / dotnet/machinelearning

Specify Categorical Features in LightGBM

Open
#7,137 0 comments 0 reactions 0 assignees View on GitHub
area-Trees enhancement untriaged
Dominant language
C#
Stars
9.4k
Forks
2k
Avg merge
2d 20h
Merged PRs (30d)
11

Description

I'm trying to use LightGBM algorithm in ML.NET to train a model using a dataset.

The dataset has a number of categorical features such as "gender", "race", "country_of_birth".

I can convert/encode these categorical data as integers.

My question is how I can specify that these features are categorical and not (continuous variables), so that LightGBM can handle them as categorical data.

I can use OneHotEncoding/OneHotHashEncoding, but I believe it's not the same.

TIA

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.