Skip to content

Instantly share code, notes, and snippets.

@NMZivkovic

NMZivkovic/building_classifier.py

Created February 3, 2018 15:01
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 NMZivkovic/26866cd1018b437fdc287e61fc6e2751 to your computer and use it in GitHub Desktop.
Save NMZivkovic/26866cd1018b437fdc287e61fc6e2751 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
building_classifier.py
# Build Neural Network - Classifier
classifier = tf.estimator.DNNClassifier(
feature_columns=columns_feat,
# Two hidden layers of 10 nodes each.
hidden_units=[10, 10],
# The model is classifying 3 classes
n_classes=3)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment