Error using score_batch for batch inference

[Ariane]

Hey everybody,

I have been learning to use the Databricks feature store and I was trying to train the model using the stored features and compute batch inference. 

I am getting an error though, running prediction using score_batch, I have been getting this error for the past two days and I have tried everything its still not resolving.  I have checked my input batch dataframe(spark_df), and there are no null values, no infinite values, and no values that are beyond the float 32. Something am I missing?

[Kumaran]

Hi @Ariane,

Thank you for posting your question in the Databricks community.

If you are using the RandomForestRegressor, Could you check by setting maxDepth it to the default value to mitigate this issue?

[Ariane]

Hey @Kumaran, 

I am using a Random forest classifier however I have tried to set the max depth to none since it is the default value but the error still exists.