Should represent the instantiation code for your Estimator object.
The features column can be accessed through the __dku_features column.
Your clustering should be provided in a column named cluster and should be an integer between
0 and nClusters - 1.