Package pyspark :: Package mllib :: Module tree :: Class DecisionTreeModel

Class DecisionTreeModel

object --+
         |
        DecisionTreeModel


A decision tree model for classification or regression.

EXPERIMENTAL: This is an experimental API.
              It will probably be modified for Spark v1.2.

Instance Methods

__init__(self, sc, java_model)
:param sc: Spark context :param java_model: Handle to Java model object

source code

__del__(self)

source code

predict(self, x)
Predict the label of one or more examples.

source code

numNodes(self)

source code

depth(self)

source code

__str__(self)
str(x)

source code

Inherited from object: __delattr__, __format__, __getattribute__, __hash__, __new__, __reduce__, __reduce_ex__, __repr__, __setattr__, __sizeof__, __subclasshook__

Properties
Inherited from `object`: `__class__`

Method Details

init(self, sc, java_model)
(Constructor)

source code

:param sc: Spark context :param java_model: Handle to Java model object

Overrides: object.__init__

predict(self, x)

source code


Predict the label of one or more examples.
:param x:  Data point (feature vector),
           or an RDD of data points (feature vectors).

str(self)
(Informal representation operator)

source code

str(x)

Overrides: object.__str__: (inherited documentation)

Class DecisionTreeModel

__init__(self, sc, java_model) (Constructor)

predict(self, x)

__str__(self) (Informal representation operator)

init(self, sc, java_model)
(Constructor)

str(self)
(Informal representation operator)