I am using MLflow in databricks and have been able to train and track model runs in the Experiments section very easily. However, once I have trained a model, I want to load the model in a new Databricks notebook, predict on new data and log the metrics. This works great too but when I look at the runs on the Experiment page, the "Models" column is not populated, see picture below. The models column is however populated when the model is trained and the first time. Is it possible to populate the Models column when loading an already trained model?
