org.apache.spark.mllib.feature

Class ChiSqSelectorModel

java.lang.Object

org.apache.spark.mllib.feature.ChiSqSelectorModel

All Implemented Interfaces:

java.io.Serializable, VectorTransformer, Saveable

public class ChiSqSelectorModel
extends java.lang.Object
implements VectorTransformer, Saveable

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
protected java.lang.String	formatVersion() Current version of model save/load format.
protected boolean	isSorted(int[] array)
static ChiSqSelectorModel	load(SparkContext sc, java.lang.String path)
void	save(SparkContext sc, java.lang.String path) Save this model to the given path.
int[]	selectedFeatures()
Vector	transform(Vector vector) Applies transformation on a vector.

Methods inherited from class java.lang.Object

clone, equals, finalize, 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

load

public static ChiSqSelectorModel load(SparkContext sc,
                      java.lang.String path)

selectedFeatures

public int[] selectedFeatures()

isSorted

protected boolean isSorted(int[] array)

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.

save

public void save(SparkContext sc,
        java.lang.String path)

Description copied from interface: Saveable

Save this model to the given path.

This saves: - human-readable (JSON) model metadata to path/metadata/ - Parquet formatted data to path/data/

The model may be loaded using Loader.load.

Specified by:

save in interface Saveable

Parameters:

sc - Spark context used to save model data.

path - Path specifying the directory in which to save this model. If the directory already exists, this method throws an exception.

formatVersion

protected java.lang.String formatVersion()

Description copied from interface: Saveable

Current version of model save/load format.

Specified by:

formatVersion in interface Saveable