Weights & Biases - AI

Initialize WandB

import wandb
from wandb.keras import WandbCallback

run = wandb.init()
config = run.config
        

Specify input type for visualization and classification

model.fit(
    X_train,
    is_five_train,
    epochs=config.epochs,
    validation_data=(X_test, is_five_test),
    callbacks=[WandbCallback(labels=labels, input_type="image")]
)