| # Setup feature columns | |
| columns_feat = [ | |
| tf.feature_column.numeric_column(key='SepalLength'), | |
| tf.feature_column.numeric_column(key='SepalWidth'), | |
| tf.feature_column.numeric_column(key='PetalLength'), | |
| tf.feature_column.numeric_column(key='PetalWidth') | |
| ] |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment