Skip to content

Instantly share code, notes, and snippets.

@HarshSingh16

HarshSingh16/feature.py

Created February 25, 2019 06:48
Show Gist options
  • Save HarshSingh16/cc361e4f257d67f6f4cfeeb1b0666850 to your computer and use it in GitHub Desktop.
Save HarshSingh16/cc361e4f257d67f6f4cfeeb1b0666850 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
feature.py
#Creating feature columns from our categorical data
education1=tf.feature_column.categorical_column_with_hash_bucket("education",hash_bucket_size=16)
workclass1=tf.feature_column.categorical_column_with_hash_bucket("workclass",hash_bucket_size=10)
martial1=tf.feature_column.categorical_column_with_hash_bucket("marital_status",hash_bucket_size=7)
occupation1=tf.feature_column.categorical_column_with_hash_bucket("occupation",hash_bucket_size=14)
relationship1=tf.feature_column.categorical_column_with_hash_bucket("relationship",hash_bucket_size=6)
race1=tf.feature_column.categorical_column_with_hash_bucket("race",hash_bucket_size=5)
gender1=tf.feature_column.categorical_column_with_hash_bucket("gender",hash_bucket_size=2)
native_country1=tf.feature_column.categorical_column_with_hash_bucket("native_country",hash_bucket_size=60)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment