Azure Databricks gives the capability to handle a lot of big data use cases and machine learning use cases, but machine learning use cases need quite a lot of compute power, and that is where the cost spikes up.
At this point, I cannot comment on the cost being ideal; it is on the higher side, but in the cloud-based environment, compared to on-premise, it could be far lesser in cost.