search

tarrade / proj_multilingual_text_classification

Explore multilingal text classification using embedding, bert and deep learning architecture
Apache License 2.0
5 stars 2 forks source link

TensorBoard: display plot as a function of nb epoch and as nb of steps #8

Closed tarrade closed 4 years ago

tarrade commented 4 years ago

display plot as a function of nb epoch (this is wrong, we have 2 epochs here!) image

this display only the first value (acc, loss) -> wrong Epoch 1/10 2/2 [==============================] - 70s 35s/step - loss: 0.6915 - accuracy: 0.5469

and as nb of steps (as for tf.estimator)

tarrade commented 4 years ago

just need to use the right parameters in TensorBoard:

tf.keras.callbacks.TensorBoard(log_dir=log_dir, histogram_freq=1, embeddings_freq=1, write_graph=True, update_freq='batch')