Skip to content

Instantly share code, notes, and snippets.

@devforfu

devforfu/tf_oop_init.py

Created April 18, 2018 04:41
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 devforfu/aa6910d874aaae43d83b53e826d45b2d to your computer and use it in GitHub Desktop.
Save devforfu/aa6910d874aaae43d83b53e826d45b2d to your computer and use it in GitHub Desktop.
Download ZIP
DNNClassifier Init
Raw
tf_oop_init.py
class DNNClassifier:
def __init__(self, hidden_layers, n_classes, activation=tf.nn.elu):
self.hidden_layers = hidden_layers
self.n_classes = n_classes
self.activation = activation
self._graph = None
self._session = None
self._saver = None
self.stop_training = False
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment