Vidhi_Khaitan
Databricks Employee
Databricks Employee

Hi Mauricen,

[JAVA_GATEWAY_EXITED] Java gateway process exited before sending its port number.
...
[wgkkr] JAVA_HOME is not set


This suggests that the serving environment cannot spin up a JVM, which is required to run Spark models (like pyspark.ml.classification.LogisticRegressionModel) because:
Spark models need a JVM to work (they rely on the Spark engine).
Databricks Model Serving (serverless or classic) does not support Spark MLlib models.

You can switching to sklearn for the model type or use the model in batch inference pipelines only, not via Model Serving UI.