Skip to content

Instantly share code, notes, and snippets.

@netskink

netskink/test_1_works.py

Last active August 27, 2018 18:47
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save netskink/23973388bb708ab75922cd587e4194d3 to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save netskink/23973388bb708ab75922cd587e4194d3 to your computer and use it in GitHub Desktop.
Download ZIP
make feature list with list comprehension
Raw
test_1_works.py
INPUT_COLUMNS = [
# Define features
tf.feature_column.categorical_column_with_vocabulary_list('dayofweek', vocabulary_list = ['Sun', 'Mon', 'Tues', 'Wed', 'Thu', 'Fri', 'Sat']),
tf.feature_column.categorical_column_with_identity('hourofday', num_buckets = 24),
]
Raw
test_2_works.py
INPUT_COLUMNS = [
# Numeric Columns
tf.feature_column.numeric_column(column) for column in CSV_COLUMNS[2:]
]
Raw
test_3_fails.py
INPUT_COLUMNS = [
# Define features
tf.feature_column.categorical_column_with_vocabulary_list('dayofweek', vocabulary_list = ['Sun', 'Mon', 'Tues', 'Wed', 'Thu', 'Fri', 'Sat']),
tf.feature_column.categorical_column_with_identity('hourofday', num_buckets = 24),
# Numeric Columns
tf.feature_column.numeric_column(column) for column in CSV_COLUMNS[2:]
]
Raw
test_4_fails.py
INPUT_COLUMNS = [
# Use a list comprehension and concatenation
# Define features
[
tf.feature_column.categorical_column_with_vocabulary_list('dayofweek', vocabulary_list = ['Sun', 'Mon', 'Tues', 'Wed', 'Thu', 'Fri', 'Sat']),
tf.feature_column.categorical_column_with_identity('hourofday', num_buckets = 24)
] +
# Numeric Columns
tf.feature_column.numeric_column(column) for column in CSV_COLUMNS[2:]
]
Raw
test_5_works.py
INPUT_COLUMNS_1 = [
# Define features
tf.feature_column.categorical_column_with_vocabulary_list('dayofweek', vocabulary_list = ['Sun', 'Mon', 'Tues', 'Wed', 'Thu', 'Fri', 'Sat']),
tf.feature_column.categorical_column_with_identity('hourofday', num_buckets = 24)
]
INPUT_COLUMNS_2 = [
# Numeric Columns
tf.feature_column.numeric_column(column) for column in CSV_COLUMNS[2:]
]
INPUT_COLUMNS = INPUT_COLUMNS_1 + INPUT_COLUMNS_2
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment