org.apache.spark.mllib.tree.model

DecisionTreeModel

class DecisionTreeModel extends Serializable

:: Experimental :: Decision tree model for classification or regression. This model stores the decision tree structure and parameters.

Annotations
@Experimental()
Linear Supertypes
Serializable, Serializable, AnyRef, Any
Ordering
  1. Alphabetic
  2. By inheritance
Inherited
  1. DecisionTreeModel
  2. Serializable
  3. Serializable
  4. AnyRef
  5. Any
  1. Hide All
  2. Show all
Learn more about member selection
Visibility
  1. Public
  2. All

Instance Constructors

  1. new DecisionTreeModel(topNode: Node, algo: Algo)

    topNode

    root node

    algo

    algorithm type -- classification or regression

Value Members

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

    Definition Classes
    AnyRef → Any

  2. final def ##(): Int

    Definition Classes
    AnyRef → Any

  3. final def ==(arg0: Any): Boolean

    Definition Classes
    AnyRef → Any

  4. val algo: Algo

    algorithm type -- classification or regression

  5. final def asInstanceOf[T0]: T0

    Definition Classes
    Any

  6. def clone(): AnyRef

    Attributes
    protected[java.lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )

  7. def depth: Int

    Get depth of tree.

    Get depth of tree. E.g.: Depth 0 means 1 leaf node. Depth 1 means 1 internal node and 2 leaf nodes.

  8. final def eq(arg0: AnyRef): Boolean

    Definition Classes
    AnyRef

  9. def equals(arg0: Any): Boolean

    Definition Classes
    AnyRef → Any

  10. def finalize(): Unit

    Attributes
    protected[java.lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( classOf[java.lang.Throwable] )

  11. final def getClass(): Class[_]

    Definition Classes
    AnyRef → Any

  12. def hashCode(): Int

    Definition Classes
    AnyRef → Any

  13. final def isInstanceOf[T0]: Boolean

    Definition Classes
    Any

  14. final def ne(arg0: AnyRef): Boolean

    Definition Classes
    AnyRef

  15. final def notify(): Unit

    Definition Classes
    AnyRef

  16. final def notifyAll(): Unit

    Definition Classes
    AnyRef

  17. def numNodes: Int

    Get number of nodes in tree, including leaf nodes.

  18. def predict(features: JavaRDD[Vector]): JavaRDD[Double]

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

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

    features

    JavaRDD representing data points to be predicted

    returns

    JavaRDD of predictions for each of the given data points

  19. def predict(features: 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.

    features

    RDD representing data points to be predicted

    returns

    RDD of predictions for each of the given data points

  20. def predict(features: Vector): Double

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

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

    features

    array representing a single data point

    returns

    Double prediction from the trained model

  21. final def synchronized[T0](arg0: ⇒ T0): T0

    Definition Classes
    AnyRef

  22. def toDebugString: String

    Print the full model to a string.

  23. def toString(): String

    Print a summary of the model.

    Print a summary of the model.

    Definition Classes
    DecisionTreeModel → AnyRef → Any

  24. val topNode: Node

    root node

  25. final def wait(): Unit

    Definition Classes
    AnyRef
    Annotations
    @throws( ... )

  26. final def wait(arg0: Long, arg1: Int): Unit

    Definition Classes
    AnyRef
    Annotations
    @throws( ... )

  27. final def wait(arg0: Long): Unit

    Definition Classes
    AnyRef
    Annotations
    @throws( ... )

Inherited from Serializable

Inherited from Serializable

Inherited from AnyRef

Inherited from Any

Ungrouped