org
.
incal
.
spark_ml
.
models
classification
package
classification
Visibility
Public
All
Type Members
trait
Classifier
extends
AnyRef
case class
DecisionTree
(
_id:
Option
[
BSONObjectID
] =
None
,
core:
TreeCore
=
TreeCore()
,
impurity:
Option
[
DecisionTreeImpurity.Value
] =
None
,
name:
Option
[
String
] =
None
,
createdById:
Option
[
BSONObjectID
] =
None
,
timeCreated:
Date
=
new Date()
)
extends
Classifier
with
Product
with
Serializable
case class
GradientBoostTree
(
_id:
Option
[
BSONObjectID
] =
None
,
core:
TreeCore
=
TreeCore()
,
maxIteration:
ValueOrSeq
[
Int
] =
Left(None)
,
stepSize:
ValueOrSeq
[
Double
] =
Left(None)
,
subsamplingRate:
ValueOrSeq
[
Double
] =
Left(None)
,
lossType:
Option
[
GBTClassificationLossType.Value
] =
None
,
name:
Option
[
String
] =
None
,
createdById:
Option
[
BSONObjectID
] =
None
,
timeCreated:
Date
=
new Date()
)
extends
Classifier
with
Product
with
Serializable
case class
LinearSupportVectorMachine
(
_id:
Option
[
BSONObjectID
] =
None
,
aggregationDepth:
ValueOrSeq
[
Int
] =
Left(None)
,
fitIntercept:
Option
[
Boolean
] =
None
,
maxIteration:
ValueOrSeq
[
Int
] =
Left(None)
,
regularization:
ValueOrSeq
[
Double
] =
Left(None)
,
standardization:
Option
[
Boolean
] =
None
,
threshold:
ValueOrSeq
[
Double
] =
Left(None)
,
tolerance:
ValueOrSeq
[
Double
] =
Left(None)
,
name:
Option
[
String
] =
None
,
createdById:
Option
[
BSONObjectID
] =
None
,
timeCreated:
Date
=
new Date()
)
extends
Classifier
with
Product
with
Serializable
case class
LogisticRegression
(
_id:
Option
[
BSONObjectID
] =
None
,
regularization:
ValueOrSeq
[
Double
] =
Left(None)
,
elasticMixingRatio:
ValueOrSeq
[
Double
] =
Left(None)
,
maxIteration:
ValueOrSeq
[
Int
] =
Left(None)
,
tolerance:
ValueOrSeq
[
Double
] =
Left(None)
,
fitIntercept:
Option
[
Boolean
] =
None
,
family:
Option
[
LogisticModelFamily.Value
] =
None
,
standardization:
Option
[
Boolean
] =
None
,
aggregationDepth:
ValueOrSeq
[
Int
] =
Left(None)
,
threshold:
ValueOrSeq
[
Double
] =
Left(None)
,
thresholds:
Option
[
Seq
[
Double
]] =
None
,
name:
Option
[
String
] =
None
,
createdById:
Option
[
BSONObjectID
] =
None
,
timeCreated:
Date
=
new Date()
)
extends
Classifier
with
Product
with
Serializable
case class
MultiLayerPerceptron
(
_id:
Option
[
BSONObjectID
] =
None
,
hiddenLayers:
Seq
[
Int
]
,
maxIteration:
ValueOrSeq
[
Int
] =
Left(None)
,
tolerance:
ValueOrSeq
[
Double
] =
Left(None)
,
blockSize:
ValueOrSeq
[
Int
] =
Left(None)
,
solver:
Option
[
MLPSolver.Value
] =
None
,
seed:
Option
[
Long
] =
None
,
stepSize:
ValueOrSeq
[
Double
] =
Left(None)
,
name:
Option
[
String
] =
None
,
createdById:
Option
[
BSONObjectID
] =
None
,
timeCreated:
Date
=
new Date()
)
extends
Classifier
with
Product
with
Serializable
case class
NaiveBayes
(
_id:
Option
[
BSONObjectID
] =
None
,
smoothing:
ValueOrSeq
[
Double
] =
Left(None)
,
modelType:
Option
[
BayesModelType.Value
] =
None
,
name:
Option
[
String
] =
None
,
createdById:
Option
[
BSONObjectID
] =
None
,
timeCreated:
Date
=
new Date()
)
extends
Classifier
with
Product
with
Serializable
case class
RandomForest
(
_id:
Option
[
BSONObjectID
] =
None
,
core:
TreeCore
=
TreeCore()
,
numTrees:
ValueOrSeq
[
Int
] =
Left(None)
,
subsamplingRate:
ValueOrSeq
[
Double
] =
Left(None)
,
impurity:
Option
[
DecisionTreeImpurity.Value
] =
None
,
featureSubsetStrategy:
Option
[
RandomForestFeatureSubsetStrategy.Value
] =
None
,
name:
Option
[
String
] =
None
,
createdById:
Option
[
BSONObjectID
] =
None
,
timeCreated:
Date
=
new Date()
)
extends
Classifier
with
Product
with
Serializable
Value Members
object
BayesModelType
extends
Enumeration
object
ClassificationEvalMetric
extends
Enumeration
object
DecisionTreeImpurity
extends
Enumeration
object
GBTClassificationLossType
extends
Enumeration
object
LogisticModelFamily
extends
Enumeration
object
MLPSolver
extends
Enumeration
object
RandomForestFeatureSubsetStrategy
extends
Enumeration
Ungrouped