org.apache.spark.ml.util

Class DatasetUtils

Object

org.apache.spark.ml.util.DatasetUtils

public class DatasetUtils
extends Object

Constructor Summary

Constructors

Constructor and Description
DatasetUtils()

Method Summary

Modifier and Type	Method and Description
static RDD<Vector>	columnToOldVector(Dataset<?> dataset, String colName)
static Column	columnToVector(Dataset<?> dataset, String colName) Cast a column in a Dataset to Vector type.

Methods inherited from class Object

equals, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

DatasetUtils

public DatasetUtils()

Method Detail

columnToVector

public static Column columnToVector(Dataset<?> dataset,
                                    String colName)

Cast a column in a Dataset to Vector type.

The supported data types of the input column are - Vector - float/double type Array.

Note: The returned column does not have Metadata.

Parameters:

dataset - input DataFrame

colName - column name.

Returns:

Vector column

columnToOldVector

public static RDD<Vector> columnToOldVector(Dataset<?> dataset,
                                            String colName)