EpistasisLab / EpistasisLab/tpot

How TPOT specify categorical feature by lgbm?

Open
#771 1 comment 1 reaction 0 assignees View on GitHub
enhancement need contributor
Dominant language
Jupyter Notebook
Stars
10.1k
Forks
1.6k
PR merge metrics
No merged PRs in 30d

Description

Hi all,

As we know, lgbm is capable of specifying the categorical features through the **categorical_feature** parameter of the **fit** function. As follows:
`model = lgb.LGBMRegressor()
model .fit(X_train, y_train,feature_name=list(X_train.columns),
categorical_feature=list(['col_1', 'col_2']))`

I know TPOT added **OneHotEncoder** to handle Categorical features. However, the idea of ​​lgbm processing categories is not just based on one-hot. So, how can i use TPOT_CONFIG to specify the categorical feature of lgbm?

Thank you very much and best regards!

Contributor guide

No contributing guide indexed for this repository

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.