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
Methods inherited from class Object |
equals, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait |
ChiSqSelectorModel
public ChiSqSelectorModel(int[] selectedFeatures)
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.