I have the following tensorflow 1.0 code:
import tensorflow as tf feature_cols = tf.contrib.learn.infer_real_valued_columns_from_input(X_train) dnn_clf = tf.