code: def test_embeddings_image(log_dir):
      iris = datasets.load_iris()
      X = iris.data
      Y = iris.target
  
      n_epochs = 2
      model = Ivis(
          epochs=n_epochs,
          k=15,
          batch_size=16,
          callbacks=[
              TensorBoardEmbeddingsImage(
                  X, Y, log_dir, epoch_interval=1
              )
          ],
      )
  
      model.fit_transform(X)
      assert os.path.exists(os.path.join(log_dir, "embeddings"))
  
file_path: ivis/tests/nn/test_callbacks.py
imports: import os
  import tempfile
  import pytest
  import numpy
  from sklearn import datasets
  from ivis.nn.callbacks import ModelCheckpoint, EmbeddingsLogging, EmbeddingsImage, TensorBoardEmbeddingsImage
  from ivis import Ivis
label_desc: Use image node_id 5 for calling a global function with example usage: test_embeddings_image(log_dir) without return types