tensorboard works

Author: whyboris

.gitignore

@@ -4,3 +4,4 @@ __pycache__
 output
 venv
 *.h5
+logs

notes.txt

@@ -11,3 +11,7 @@ pip install -r requirements.txt
 jupyter notebook 
 
 jupyter lab
+
+tensorboard --logdir=logs
+http://localhost:6006/
+

requirements.txt

@@ -6,4 +6,6 @@ matplotlib
 numpy==1.13.0
 opencv-python
 pylint
-tensorflow
+tensorflow
+# html5lib newer version needed to get tensorboard to work
+html5lib==1.0.1

tensorboard.py

@@ -25,6 +25,21 @@
 
 model.summary()
 
-model.compile(optimiser='rmsprop', loss='binary_crossentropy', metrics=['acc'])
+model.compile(optimizer='rmsprop', loss='binary_crossentropy', metrics=['acc'])
+
+callbacks = [
+  keras.callbacks.TensorBoard(
+    log_dir = './logs',
+    histogram_freq=1,
+    embeddings_freq=1,
+  )  
+]
+
+history = model.fit(x_train, y_train, 
+                    epochs=20,
+                    batch_size=128,
+                    validation_split=0.2,
+                    callbacks=callbacks)
+