Skip to content
New issue

Have a question about this project? # for a free GitHub account to open an issue and contact its maintainers and the community.

#

By clicking “#”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? # to your account

Jump to bottom

Splitting into Categorical features #17

Open
harshraj22 opened this issue May 11, 2022 · 0 comments
Open

Splitting into Categorical features #17

harshraj22 opened this issue May 11, 2022 · 0 comments
Assignees
@harshraj22

Comments

@harshraj22
Copy link
Owner

DeepLearning/implementations/TabTransformer/data_loader/datasets.py

Line 21 in 60b90a6

CATEGORICAL_COLUMNS = ['gender', 'SeniorCitizen', 'Partner', 'Dependents', 'PhoneService', 'MultipleLines', 'InternetService', 'OnlineSecurity', 'OnlineBackup', 'DeviceProtection', 'TechSupport', 'StreamingTV', 'StreamingMovies', 'Contract', 'PaperlessBilling', 'PaymentMethod']

Since the dataset is specific, it seems ok to use variables to store categorical and continuous variables. For generic case, you may want to use fastai’s cont_cat_split
@harshraj22 harshraj22 self-assigned this May 13, 2022
# for free to join this conversation on GitHub. Already have an account? # to comment
Assignees

@harshraj22 harshraj22

Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
1 participant
@harshraj22