import markov
# Your model definition
# model = ...
# create recorder
recorder = markov.ExperimentRecorder(
name="MarkovML experiment #1"
)
# The register method registers this recorder with the MarkovML backend.
# You can record data with the backend only through a registered recorder.
recorder.register()
with recorder:
for epoch in range(500):
pred = model(x)
# Calculate and record loss
loss = loss_function(pred, y)
recorder.add_record({"loss": loss})
# calculate and record accuracy
accuracy = accuracy_function(pred, y)
recorder.add_record({"accuracy": accuracy})
# REST OF THE MODEL TRAINING CODE GOES HERE
recorder.summary.add_training_loss(value=str(loss))
recorder.summary.add_training_accuracy(value=str(accuracy))