Packages

o

org.apache.spark.ml.stat

ChiSquareTest

object ChiSquareTest

Chi-square hypothesis testing for categorical data.

See Wikipedia for more information on the Chi-squared test.

Annotations
@Since( "2.2.0" )
Source
ChiSquareTest.scala
Linear Supertypes
AnyRef, Any
Ordering
  1. Alphabetic
  2. By Inheritance
Inherited
  1. ChiSquareTest
  2. AnyRef
  3. Any
  1. Hide All
  2. Show All
Visibility
  1. Public
  2. All

Value Members

  1. def test(dataset: DataFrame, featuresCol: String, labelCol: String, flatten: Boolean): DataFrame

    dataset

    DataFrame of categorical labels and categorical features. Real-valued features will be treated as categorical for each distinct value.

    featuresCol

    Name of features column in dataset, of type Vector (VectorUDT)

    labelCol

    Name of label column in dataset, of any numerical type

    flatten

    If false, the returned DataFrame contains only a single Row, otherwise, one row per feature.

    Annotations
    @Since( "3.1.0" )
  2. def test(dataset: DataFrame, featuresCol: String, labelCol: String): DataFrame

    Conduct Pearson's independence test for every feature against the label.

    Conduct Pearson's independence test for every feature against the label. For each feature, the (feature, label) pairs are converted into a contingency matrix for which the Chi-squared statistic is computed. All label and feature values must be categorical.

    The null hypothesis is that the occurrence of the outcomes is statistically independent.

    dataset

    DataFrame of categorical labels and categorical features. Real-valued features will be treated as categorical for each distinct value.

    featuresCol

    Name of features column in dataset, of type Vector (VectorUDT)

    labelCol

    Name of label column in dataset, of any numerical type

    returns

    DataFrame containing the test result for every feature against the label. This DataFrame will contain a single Row with the following fields:

    • pValues: Vector
    • degreesOfFreedom: Array[Int]
    • statistics: Vector Each of these fields has one value per feature.
    Annotations
    @Since( "2.2.0" )