org.apache.spark.mllib.feature

Class ChiSqSelectorModel

Object

org.apache.spark.mllib.feature.ChiSqSelectorModel

All Implemented Interfaces:

java.io.Serializable, VectorTransformer

public class ChiSqSelectorModel
extends Object
implements VectorTransformer

:: Experimental :: Chi Squared selector model.

param: selectedFeatures list of indices to select (filter). Must be ordered asc

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
ChiSqSelectorModel(int[] selectedFeatures)

Method Summary

Methods

Modifier and Type	Method and Description
int[]	selectedFeatures()
Vector	transform(Vector vector) Applies transformation on a vector.

Methods inherited from class Object

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

Methods inherited from interface org.apache.spark.mllib.feature.VectorTransformer

transform, transform

Constructor Detail

ChiSqSelectorModel

public ChiSqSelectorModel(int[] selectedFeatures)

Method Detail

selectedFeatures

public int[] selectedFeatures()

transform

public Vector transform(Vector vector)

Applies transformation on a vector.

Specified by:

transform in interface VectorTransformer

Parameters:

vector - vector to be transformed.

Returns:

transformed vector.