Skip to content

Instantly share code, notes, and snippets.

@sararob

sararob/dnnestimator.py Secret

Created August 14, 2018 18:25
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save sararob/959dc89bddada72c795d61a348366bb7 to your computer and use it in GitHub Desktop.
Save sararob/959dc89bddada72c795d61a348366bb7 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
dnnestimator.py
estimator = tf.contrib.estimator.DNNEstimator(
head=multi_label_head,
hidden_units=[64,10],
feature_columns=[description_embeddings]
)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment