def _convertOutputToVector(self, df, tfs_output_col):
"""
Converts the output python list to MLlib Vector.
"""
return (df.withColumn(self.getOutputCol(),
JVMAPI.listToMLlibVectorUDF(
df[tfs_output_col])).drop(tfs_output_col))
def _convertOutputToVector(self, df, tfs_output_col):
"""
Converts the output python list to MLlib Vector.
"""
return df\
.withColumn(self.getOutputCol(), JVMAPI.listToMLlibVectorUDF(df[tfs_output_col]))\
.drop(tfs_output_col)