Package | Description |
---|---|
org.tribuo.classification.sgd |
Provides infrastructure for Stochastic Gradient Descent for classification problems.
|
org.tribuo.classification.sgd.linear |
Provides an implementation of a classification linear model using Stochastic Gradient Descent.
|
org.tribuo.classification.sgd.objectives |
Provides classification loss functions for Stochastic Gradient Descent.
|
org.tribuo.common.sgd |
Provides the base classes for models trained with stochastic gradient descent.
|
org.tribuo.multilabel.sgd |
Provides infrastructure for Stochastic Gradient Descent for multi-label classification problems.
|
org.tribuo.multilabel.sgd.linear |
Provides an implementation of a multi-label classification linear model using Stochastic Gradient Descent.
|
org.tribuo.multilabel.sgd.objectives |
Provides multi-label classification loss functions for Stochastic Gradient Descent.
|
org.tribuo.regression.sgd |
Provides infrastructure for Stochastic Gradient Descent based regression models.
|
org.tribuo.regression.sgd.linear |
Provides an implementation of linear regression using Stochastic Gradient Descent.
|
org.tribuo.regression.sgd.objectives |
Provides regression loss functions for Stochastic Gradient Descent.
|
Modifier and Type | Interface and Description |
---|---|
interface |
LabelObjective
An interface for single label prediction objectives.
|
Modifier and Type | Method and Description |
---|---|
protected SGDObjective<Integer> |
LinearSGDTrainer.getObjective() |
Modifier and Type | Class and Description |
---|---|
class |
Hinge
Hinge loss, scores the correct value margin and any incorrect predictions -margin.
|
class |
LogMulticlass
A multiclass version of the log loss.
|
Modifier and Type | Method and Description |
---|---|
protected abstract SGDObjective<U> |
AbstractSGDTrainer.getObjective()
Returns the objective used by this trainer.
|
Modifier and Type | Interface and Description |
---|---|
interface |
MultiLabelObjective
An interface for multi-label prediction objectives.
|
Modifier and Type | Method and Description |
---|---|
protected SGDObjective<SGDVector> |
LinearSGDTrainer.getObjective() |
Modifier and Type | Class and Description |
---|---|
class |
BinaryCrossEntropy
A multilabel version of binary cross entropy loss which expects logits.
|
Modifier and Type | Interface and Description |
---|---|
interface |
RegressionObjective
An interface for regression objectives.
|
Modifier and Type | Method and Description |
---|---|
protected SGDObjective<DenseVector> |
LinearSGDTrainer.getObjective() |
Modifier and Type | Class and Description |
---|---|
class |
AbsoluteLoss
Absolute loss (i.e., l1).
|
class |
Huber
Huber loss, i.e., a mixture of l2 and l1 losses.
|
class |
SquaredLoss
Squared loss, i.e., l2.
|
Copyright © 2015–2021 Oracle and/or its affiliates. All rights reserved.