Skip to content

Instantly share code, notes, and snippets.

@ntakouris

ntakouris/types.py

Created July 8, 2020 12:52
Show Gist options
  • Save ntakouris/2c310dde9366aeb77fd36ef811203bc0 to your computer and use it in GitHub Desktop.
Save ntakouris/2c310dde9366aeb77fd36ef811203bc0 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
types.py
CATEGORICAL_FEATURE_KEYS = [
'workclass',
'education',
'marital-status',
'occupation',
'relationship',
'...'
NUMERIC_FEATURE_KEYS = [
'age',
'capital-gain',
'capital-loss',
'hours-per-week',
]
OPTIONAL_NUMERIC_FEATURE_KEYS = [
'education-num',
]
LABEL_KEY = 'label'
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment