Issue facing while developing regression model using elephas estimator. Any solution?

Question

I am trying to develop a Neural network regression model using spark pipeline and elephas estimator. After developing model getting the following error. Most of the work using e