Skip to content

Instantly share code, notes, and snippets.

@securetorobert

securetorobert/tf_estimator_dnnclassifier_iris_train.py

Created July 15, 2018 10:38
Show Gist options
  • Save securetorobert/52c47a2c32fcb7a567b22db8ea176bec to your computer and use it in GitHub Desktop.
Save securetorobert/52c47a2c32fcb7a567b22db8ea176bec to your computer and use it in GitHub Desktop.
Download ZIP
Training a DNNClassifier on the Iris Dataset
Raw
tf_estimator_dnnclassifier_iris_train.py
# Train our model, use the previously function my_input_fn
# Input to training is a file with training example
# Stop training after 8 iterations of train data (epochs)
classifier.train(
input_fn=lambda: my_input_fn(FILE_TRAIN, True, 8))
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment