org.apache.spark.mllib.classification

LogisticRegressionModel

class LogisticRegressionModel extends GeneralizedLinearModel with ClassificationModel with Serializable with Saveable

Classification model trained using Multinomial/Binary Logistic Regression.

Linear Supertypes
Saveable, ClassificationModel, GeneralizedLinearModel, Serializable, Serializable, AnyRef, Any
Ordering
  1. Alphabetic
  2. By inheritance
Inherited
  1. LogisticRegressionModel
  2. Saveable
  3. ClassificationModel
  4. GeneralizedLinearModel
  5. Serializable
  6. Serializable
  7. AnyRef
  8. Any
  1. Hide All
  2. Show all
Learn more about member selection
Visibility
  1. Public
  2. All

Instance Constructors

  1. new LogisticRegressionModel(weights: Vector, intercept: Double)

    Constructs a LogisticRegressionModel with weights and intercept for binary classification.

  2. new LogisticRegressionModel(weights: Vector, intercept: Double, numFeatures: Int, numClasses: Int)

    weights

    Weights computed for every feature.

    intercept

    Intercept computed for this model. (Only used in Binary Logistic Regression. In Multinomial Logistic Regression, the intercepts will not be a single value, so the intercepts will be part of the weights.)

    numFeatures

    the dimension of the features.

    numClasses

    the number of possible outcomes for k classes classification problem in Multinomial Logistic Regression. By default, it is binary logistic regression so numClasses will be set to 2.

Value Members

  1. final def !=(arg0: AnyRef): Boolean

    Definition Classes
    AnyRef
  2. final def !=(arg0: Any): Boolean

    Definition Classes
    Any
  3. final def ##(): Int

    Definition Classes
    AnyRef → Any
  4. final def ==(arg0: AnyRef): Boolean

    Definition Classes
    AnyRef
  5. final def ==(arg0: Any): Boolean

    Definition Classes
    Any
  6. final def asInstanceOf[T0]: T0

    Definition Classes
    Any
  7. def clearThreshold(): LogisticRegressionModel.this.type

    :: Experimental :: Clears the threshold so that predict will output raw prediction scores.

    :: Experimental :: Clears the threshold so that predict will output raw prediction scores.

    Annotations
    @Experimental()
  8. def clone(): AnyRef

    Attributes
    protected[java.lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  9. final def eq(arg0: AnyRef): Boolean

    Definition Classes
    AnyRef
  10. def equals(arg0: Any): Boolean

    Definition Classes
    AnyRef → Any
  11. def finalize(): Unit

    Attributes
    protected[java.lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( classOf[java.lang.Throwable] )
  12. def formatVersion: String

    Current version of model save/load format.

    Current version of model save/load format.

    Attributes
    protected
    Definition Classes
    LogisticRegressionModelSaveable
  13. final def getClass(): Class[_]

    Definition Classes
    AnyRef → Any
  14. def getThreshold: Option[Double]

    :: Experimental :: Returns the threshold (if any) used for converting raw prediction scores into 0/1 predictions.

    :: Experimental :: Returns the threshold (if any) used for converting raw prediction scores into 0/1 predictions.

    Annotations
    @Experimental()
  15. def hashCode(): Int

    Definition Classes
    AnyRef → Any
  16. val intercept: Double

    Intercept computed for this model.

    Intercept computed for this model. (Only used in Binary Logistic Regression. In Multinomial Logistic Regression, the intercepts will not be a single value, so the intercepts will be part of the weights.)

    Definition Classes
    LogisticRegressionModelGeneralizedLinearModel
  17. final def isInstanceOf[T0]: Boolean

    Definition Classes
    Any
  18. final def ne(arg0: AnyRef): Boolean

    Definition Classes
    AnyRef
  19. final def notify(): Unit

    Definition Classes
    AnyRef
  20. final def notifyAll(): Unit

    Definition Classes
    AnyRef
  21. val numClasses: Int

    the number of possible outcomes for k classes classification problem in Multinomial Logistic Regression.

    the number of possible outcomes for k classes classification problem in Multinomial Logistic Regression. By default, it is binary logistic regression so numClasses will be set to 2.

  22. val numFeatures: Int

    the dimension of the features.

  23. def predict(testData: JavaRDD[Vector]): JavaRDD[Double]

    Predict values for examples stored in a JavaRDD.

    Predict values for examples stored in a JavaRDD.

    testData

    JavaRDD representing data points to be predicted

    returns

    a JavaRDD[java.lang.Double] where each entry contains the corresponding prediction

    Definition Classes
    ClassificationModel
  24. def predict(testData: Vector): Double

    Predict values for a single data point using the model trained.

    Predict values for a single data point using the model trained.

    testData

    array representing a single data point

    returns

    Double prediction from the trained model

    Definition Classes
    GeneralizedLinearModel
  25. def predict(testData: RDD[Vector]): RDD[Double]

    Predict values for the given data set using the model trained.

    Predict values for the given data set using the model trained.

    testData

    RDD representing data points to be predicted

    returns

    RDD[Double] where each entry contains the corresponding prediction

    Definition Classes
    GeneralizedLinearModel
  26. def predictPoint(dataMatrix: Vector, weightMatrix: Vector, intercept: Double): Double

    Predict the result given a data point and the weights learned.

    Predict the result given a data point and the weights learned.

    dataMatrix

    Row vector containing the features for this data point

    weightMatrix

    Column vector containing the weights of the model

    intercept

    Intercept of the model.

    Attributes
    protected
    Definition Classes
    LogisticRegressionModelGeneralizedLinearModel
  27. def save(sc: SparkContext, path: String): Unit

    Save this model to the given path.

    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.

    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.

    Definition Classes
    LogisticRegressionModelSaveable
  28. def setThreshold(threshold: Double): LogisticRegressionModel.this.type

    :: Experimental :: Sets the threshold that separates positive predictions from negative predictions in Binary Logistic Regression.

    :: Experimental :: Sets the threshold that separates positive predictions from negative predictions in Binary Logistic Regression. An example with prediction score greater than or equal to this threshold is identified as an positive, and negative otherwise. The default value is 0.5.

    Annotations
    @Experimental()
  29. final def synchronized[T0](arg0: ⇒ T0): T0

    Definition Classes
    AnyRef
  30. def toString(): String

    Definition Classes
    GeneralizedLinearModel → AnyRef → Any
  31. final def wait(): Unit

    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  32. final def wait(arg0: Long, arg1: Int): Unit

    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  33. final def wait(arg0: Long): Unit

    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  34. val weights: Vector

    Weights computed for every feature.

    Weights computed for every feature.

    Definition Classes
    LogisticRegressionModelGeneralizedLinearModel

Inherited from Saveable

Inherited from ClassificationModel

Inherited from GeneralizedLinearModel

Inherited from Serializable

Inherited from Serializable

Inherited from AnyRef

Inherited from Any

Ungrouped