autogluon / autogluon/tabarena

[TabArena-v0.X] Try better fillna / cat encoding strategies for RF/XT/KNN

Open
#124 3 comments 0 reactions 0 assignees View on GitHub
TabArena-v0.X
Dominant language
Python
Stars
303
Forks
69
Avg merge
1d 4h
Merged PRs (30d)
49

Description

Try better fillna / cat encoding strategies for RF/XT/KNN

Currently ordinal encoding for RF/XT and fillna 0 for RF/XT/KNN.
KNN ignores categorical features.

Figure out what missing categories are assigned to for RF/XT.

Maybe try mean fillna + indicator for numeric.

Contributor guide

Open the contributing guide

Research direction

Start by tracing the RF, XT, and KNN preprocessing entry points to identify where ordinal encoding and fillna 0 are applied. Compare alternative categorical and numeric missing-value strategies in the TabArena benchmark, verify how missing categories are assigned for RF/XT, and document the resulting model comparisons.

Written by the indexing model from the issue text.

Assessment

Tech stack
python
Domain
machine-learning
Issue type
Feature
Difficulty
5/5
Estimated time
Over a week
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
25/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.