Beispiel #1
0
 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)