cc
.
factorie
.
app
.
classify
.
backend
CARTDecisionTreeTrainer
class
CARTDecisionTreeTrainer
extends
DecisionTreeTrainer
with
TensorSumStatsAndLabels
with
GiniSplitting
with
SampleSizeStopping
with
NoPruning
Linear Supertypes
NoPruning
,
SampleSizeStopping
,
GiniSplitting
,
TensorSumStatsAndLabels
,
DTreeBucketStats
,
DecisionTreeTrainer
,
AnyRef
,
Any
Ordering
Alphabetic
By inheritance
Inherited
CARTDecisionTreeTrainer
NoPruning
SampleSizeStopping
GiniSplitting
TensorSumStatsAndLabels
DTreeBucketStats
DecisionTreeTrainer
AnyRef
Any
Hide All
Show all
Learn more about member selection
Visibility
Public
All
Instance Constructors
new
CARTDecisionTreeTrainer
()
Type Members
type
BucketStats
=
MutableBucketStats
Definition Classes
TensorSumStatsAndLabels
→
DTreeBucketStats
type
Instance
=
DecisionTreeTrainer.Instance
Definition Classes
DTreeBucketStats
type
Label
=
la.Tensor1
Definition Classes
DTreeBucketStats
class
MutableBucketStats
extends
AnyRef
Definition Classes
TensorSumStatsAndLabels
type
State
=
Double
Definition Classes
GiniSplitting
Value Members
final
def
!=
(
arg0:
AnyRef
)
:
Boolean
Definition Classes
AnyRef
final
def
!=
(
arg0:
Any
)
:
Boolean
Definition Classes
Any
final
def
##
()
:
Int
Definition Classes
AnyRef → Any
def
+=
(
left:
BucketStats
,
right:
BucketStats
)
:
Unit
Definition Classes
TensorSumStatsAndLabels
→
DTreeBucketStats
def
-=
(
left:
BucketStats
,
right:
BucketStats
)
:
Unit
Definition Classes
TensorSumStatsAndLabels
→
DTreeBucketStats
final
def
==
(
arg0:
AnyRef
)
:
Boolean
Definition Classes
AnyRef
final
def
==
(
arg0:
Any
)
:
Boolean
Definition Classes
Any
def
accumulate
(
stats:
BucketStats
,
inst:
Instance
)
:
Unit
Definition Classes
TensorSumStatsAndLabels
→
DTreeBucketStats
final
def
asInstanceOf
[
T0
]
:
T0
Definition Classes
Any
def
clone
()
:
AnyRef
Attributes
protected[
java.lang
]
Definition Classes
AnyRef
Annotations
@throws
(
...
)
final
def
eq
(
arg0:
AnyRef
)
:
Boolean
Definition Classes
AnyRef
def
equals
(
arg0:
Any
)
:
Boolean
Definition Classes
AnyRef → Any
def
evaluateSplittingCriteria
(
baseGini:
Double
,
withFeature:
MutableBucketStats
,
withoutFeature:
MutableBucketStats
)
:
Double
Definition Classes
GiniSplitting
def
evaluateSplittingCriteria
(
instances:
Seq
[
Instance
]
,
possibleFeatureThresholds:
HashMap
[
Int
,
Array
[
Double
]]
)
:
HashMap
[
Int
,
Array
[
Double
]]
Definition Classes
DecisionTreeTrainer
def
finalize
()
:
Unit
Attributes
protected[
java.lang
]
Definition Classes
AnyRef
Annotations
@throws
(
classOf[java.lang.Throwable]
)
def
getBucketPrediction
(
labels:
Seq
[
Instance
]
)
:
Label
Definition Classes
DecisionTreeTrainer
def
getBucketState
(
instances:
Iterable
[
Instance
]
)
:
Double
Definition Classes
GiniSplitting
def
getBucketStats
(
labels:
Iterable
[
Instance
]
)
:
BucketStats
Definition Classes
TensorSumStatsAndLabels
→
DTreeBucketStats
final
def
getClass
()
:
Class
[_]
Definition Classes
AnyRef → Any
def
getEmptyBucketStats
(
inst:
Instance
)
:
BucketStats
Definition Classes
TensorSumStatsAndLabels
→
DTreeBucketStats
def
getGini
(
stats:
BucketStats
)
:
Double
Definition Classes
GiniSplitting
def
getPrediction
(
stats:
BucketStats
)
:
Label
Definition Classes
TensorSumStatsAndLabels
→
DTreeBucketStats
def
hasFeature
(
featureIdx:
Int
,
feats:
la.Tensor1
,
threshold:
Double
)
:
Boolean
Definition Classes
DecisionTreeTrainer
Annotations
@inline
()
def
hashCode
()
:
Int
Definition Classes
AnyRef → Any
final
def
isInstanceOf
[
T0
]
:
Boolean
Definition Classes
Any
def
makeLeaf
(
stats:
BucketStats
)
:
DTree
Definition Classes
TensorSumStatsAndLabels
→
DTreeBucketStats
var
maxDepth
:
Int
Definition Classes
DecisionTreeTrainer
val
minSampleSize
:
Int
Definition Classes
CARTDecisionTreeTrainer
→
SampleSizeStopping
final
def
ne
(
arg0:
AnyRef
)
:
Boolean
Definition Classes
AnyRef
final
def
notify
()
:
Unit
Definition Classes
AnyRef
final
def
notifyAll
()
:
Unit
Definition Classes
AnyRef
def
prune
(
tree:
DTree
,
pruningSet:
Seq
[
Instance
]
)
:
DTree
Definition Classes
NoPruning
def
samePred
(
labels:
Seq
[
Label
]
)
:
Boolean
Definition Classes
GiniSplitting
def
shouldStop
(
stats:
Seq
[
Instance
]
,
depth:
Int
)
:
Boolean
Definition Classes
SampleSizeStopping
final
def
synchronized
[
T0
]
(
arg0: ⇒
T0
)
:
T0
Definition Classes
AnyRef
def
toString
()
:
String
Definition Classes
AnyRef → Any
def
train
(
trainInstances:
Seq
[
Instance
]
,
pruneInstances:
Seq
[
Instance
] =
Nil
,
numFeaturesToUse:
Int
=
1
)
(
implicit
rng:
Random
)
:
DTree
Definition Classes
DecisionTreeTrainer
final
def
wait
()
:
Unit
Definition Classes
AnyRef
Annotations
@throws
(
...
)
final
def
wait
(
arg0:
Long
,
arg1:
Int
)
:
Unit
Definition Classes
AnyRef
Annotations
@throws
(
...
)
final
def
wait
(
arg0:
Long
)
:
Unit
Definition Classes
AnyRef
Annotations
@throws
(
...
)
Inherited from
NoPruning
Inherited from
SampleSizeStopping
Inherited from
GiniSplitting
Inherited from
TensorSumStatsAndLabels
Inherited from
DTreeBucketStats
Inherited from
DecisionTreeTrainer
Inherited from
AnyRef
Inherited from
Any
Ungrouped