User Defined Function In Databricks Training