Skip to content

Instantly share code, notes, and snippets.

@Tathagatd96

Tathagatd96/gist:da596e9ced3d7dec0fdead72f59c24de

Created June 15, 2018 06:18
Show Gist options
  • Save Tathagatd96/da596e9ced3d7dec0fdead72f59c24de to your computer and use it in GitHub Desktop.
Save Tathagatd96/da596e9ced3d7dec0fdead72f59c24de to your computer and use it in GitHub Desktop.
Download ZIP
Raw
gistfile1.txt
#The estimator model
model=tf.estimator.DNNRegressor(hidden_units=[6,10,6],feature_columns=feat_cols)
#the input function
input_func=tf.estimator.inputs.pandas_input_fn(X_train,y_train,batch_size=10,num_epochs=1000,shuffle=True)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment