Custom Spark ML Model
The following should represent the instantiation code for your Estimator object.
The features column can be accessed through the __dku_features column, and the target through the {{mlTaskDesign.targetVariable}} column.
The target is provided as a double which is either 0.0 or 1.0 The target is provided as a double with discrete values between 0.0 and nCategories-1 .
Your predictions should be provided in a column named prediction, with the same format as the target column.