Skip to content

Instantly share code, notes, and snippets.

@securetorobert

securetorobert/tf_l1_reg_feature_columns.py

Created July 15, 2018 21:43
Show Gist options
  • Save securetorobert/a9fd1c2de39921e00ae3e9f669920b34 to your computer and use it in GitHub Desktop.
Save securetorobert/a9fd1c2de39921e00ae3e9f669920b34 to your computer and use it in GitHub Desktop.
Download ZIP
Create feature columns for l1 reg in TensorFlow
Raw
tf_l1_reg_feature_columns.py
X.info()
feature_columns = [
tf.feature_column.numeric_column('crim', dtype=tf.float64),
tf.feature_column.numeric_column('zn', dtype=tf.float64),
tf.feature_column.numeric_column('indus', dtype=tf.float64),
tf.feature_column.numeric_column('chas', dtype=tf.int64),
tf.feature_column.numeric_column('nox', dtype=tf.float64),
tf.feature_column.numeric_column('rm', dtype=tf.float64),
tf.feature_column.numeric_column('age', dtype=tf.float64),
tf.feature_column.numeric_column('dis', dtype=tf.float64),
tf.feature_column.numeric_column('rad', dtype=tf.int64),
tf.feature_column.numeric_column('tax', dtype=tf.int64),
tf.feature_column.numeric_column('ptratio', dtype=tf.float64),
tf.feature_column.numeric_column('black', dtype=tf.float64),
tf.feature_column.numeric_column('lstat', dtype=tf.float64),
tf.feature_column.numeric_column('crim_2', dtype=tf.float64),
tf.feature_column.numeric_column('zn_2', dtype=tf.float64),
tf.feature_column.numeric_column('indus_2', dtype=tf.float64),
tf.feature_column.numeric_column('chas_2', dtype=tf.int64),
tf.feature_column.numeric_column('nox_2', dtype=tf.float64),
tf.feature_column.numeric_column('rm_2', dtype=tf.float64),
tf.feature_column.numeric_column('age_2', dtype=tf.float64),
tf.feature_column.numeric_column('dis_2', dtype=tf.float64),
tf.feature_column.numeric_column('rad_2', dtype=tf.int64),
tf.feature_column.numeric_column('tax_2', dtype=tf.int64),
tf.feature_column.numeric_column('ptratio_2', dtype=tf.float64),
tf.feature_column.numeric_column('black_2', dtype=tf.float64),
tf.feature_column.numeric_column('lstat_2', dtype=tf.float64)
]
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment