Dan_Z
Databricks Employee
Databricks Employee

To my knowledge, if you write a custom model's predict() function, you can do any arbitrary operations in it (log inputs or outputs somewhere).