Skip to main content

AutoKeras IMDB

The autokeras_imdb_example.py example script demonstrates the integration of ClearML into code, which uses autokeras.

The example does the following:

  • Trains text classification networks on the Keras built-in IMDB dataset, using the autokeras TextClassifier class, and searches for the best model.
  • Uses two TensorBoard callbacks, one for training and one for testing.
  • ClearML automatically logs everything the code sends to TensorBoard.
  • Creates an experiment named autokeras imdb example with scalars, which is associated with the autokeras project.

Scalars

The loss and accuracy metric scalar plots appear in SCALARS, along with the resource utilization plots, which are titled :monitor: machine.

image

Hyperparameters

ClearML automatically logs TensorFlow Definitions. They appear in CONFIGURATION > HYPER PARAMETERS > TF_DEFINE.

image

Console

Text printed to the console for training progress, as well as all other console output, appear in CONSOLE.

image

Artifacts

Models created by the experiment appear in the experiment’s ARTIFACTS tab.

image

Clicking on the model's name takes you to the model’s page, where you can view the model’s details and access the model.

image