cc
.
factorie
.
app
.
classify
.
backend
RegressionTreeTrainer
class
RegressionTreeTrainer
extends
DecisionTreeTrainer
with
TensorSumSqDiagStatsAndLabels
with
DiagonalCovarianceSplitting
with
SampleSizeStopping
with
RMSEBasedPruning
Linear Supertypes
RMSEBasedPruning
,
SampleSizeStopping
,
DiagonalCovarianceSplitting
,
TensorSumSqDiagStatsAndLabels
,
DTreeBucketStats
,
DecisionTreeTrainer
,
AnyRef
,
Any
Ordering
Alphabetic
By inheritance
Inherited
RegressionTreeTrainer
RMSEBasedPruning
SampleSizeStopping
DiagonalCovarianceSplitting
TensorSumSqDiagStatsAndLabels
DTreeBucketStats
DecisionTreeTrainer
AnyRef
Any
Hide All
Show all
Learn more about member selection
Visibility
Public
All
Instance Constructors
new
RegressionTreeTrainer
()
Type Members
type
BucketStats
=
MutableBucketStats
Definition Classes
TensorSumSqDiagStatsAndLabels
→
DTreeBucketStats
type
Instance
=
DecisionTreeTrainer.Instance
Definition Classes
DTreeBucketStats
type
Label
=
la.Tensor1
Definition Classes
DTreeBucketStats
class
MutableBucketStats
extends
AnyRef
Definition Classes
TensorSumSqDiagStatsAndLabels
type
State
=
Double
Definition Classes
DiagonalCovarianceSplitting
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
TensorSumSqDiagStatsAndLabels
→
DTreeBucketStats
def
-=
(
left:
BucketStats
,
right:
BucketStats
)
:
Unit
Definition Classes
TensorSumSqDiagStatsAndLabels
→
DTreeBucketStats
final
def
==
(
arg0:
AnyRef
)
:
Boolean
Definition Classes
AnyRef
final
def
==
(
arg0:
Any
)
:
Boolean
Definition Classes
Any
def
RMSE
(
node:
DTree
,
stats:
Seq
[
Instance
]
)
:
Double
Definition Classes
RMSEBasedPruning
def
accumulate
(
stats:
BucketStats
,
inst:
Instance
)
:
Unit
Definition Classes
TensorSumSqDiagStatsAndLabels
→
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
(
baseStdDev:
Double
,
withFeature:
MutableBucketStats
,
withoutFeature:
MutableBucketStats
)
:
Double
Definition Classes
DiagonalCovarianceSplitting
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
DiagonalCovarianceSplitting
def
getBucketStats
(
labels:
Iterable
[
Instance
]
)
:
BucketStats
Definition Classes
TensorSumSqDiagStatsAndLabels
→
DTreeBucketStats
final
def
getClass
()
:
Class
[_]
Definition Classes
AnyRef → Any
def
getEmptyBucketStats
(
inst:
Instance
)
:
BucketStats
Definition Classes
TensorSumSqDiagStatsAndLabels
→
DTreeBucketStats
def
getPrediction
(
stats:
BucketStats
)
:
Label
Definition Classes
TensorSumSqDiagStatsAndLabels
→
DTreeBucketStats
def
getStdDev
(
stats:
BucketStats
)
:
Double
Definition Classes
DiagonalCovarianceSplitting
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
label
(
feats:
la.Tensor1
,
node:
DTree
)
:
la.Tensor1
Definition Classes
RMSEBasedPruning
def
makeLeaf
(
stats:
BucketStats
)
:
DTree
Definition Classes
TensorSumSqDiagStatsAndLabels
→
DTreeBucketStats
var
maxDepth
:
Int
Definition Classes
DecisionTreeTrainer
var
minSampleSize
:
Int
Definition Classes
RegressionTreeTrainer
→
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
RMSEBasedPruning
def
samePred
(
labels:
Seq
[
Label
]
)
:
Boolean
Definition Classes
DiagonalCovarianceSplitting
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
RMSEBasedPruning
Inherited from
SampleSizeStopping
Inherited from
DiagonalCovarianceSplitting
Inherited from
TensorSumSqDiagStatsAndLabels
Inherited from
DTreeBucketStats
Inherited from
DecisionTreeTrainer
Inherited from
AnyRef
Inherited from
Any
Ungrouped