Trying to compare various models on a test set and I need the model name displayed on the chart where each step on x-axis is a particular model name/checkpoint and y-axis is its respective accuracy metric value on a test set. So far what I’ve done is to log the following for each model eval
”””
run.log({
"model_path": *weights*,
“model_date”: os.path.getmtime(weights),
“precision_M”: metrics.results_dict[‘metrics/precision(M)’],
“recall_M”: metrics.results_dict[‘metrics/recall(M)’],
})
I need the model name to be shown when I hover a particular data point. I’ve gotten a model name to be displayed but it remains the same when I click on any checkpoint. Can anyone help?