Relate coefficients under linear feature transformation | Databricks Interview Question