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.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.objectives |
Provides regression loss functions for Stochastic Gradient Descent.
|
Class and Description |
---|
SGDObjective
An interface for a loss function that can produce the loss and gradient incurred by
a single prediction.
|
Class and Description |
---|
AbstractLinearSGDModel |
AbstractLinearSGDTrainer
A trainer for a linear model which uses SGD.
|
AbstractSGDModel |
AbstractSGDTrainer
A trainer for a model which uses SGD.
|
SGDObjective
An interface for a loss function that can produce the loss and gradient incurred by
a single prediction.
|
Class and Description |
---|
SGDObjective
An interface for a loss function that can produce the loss and gradient incurred by
a single prediction.
|
Class and Description |
---|
AbstractSGDModel |
AbstractSGDModel.PredAndActive
A nominal tuple used to capture the prediction and the number of active features used by the model.
|
AbstractSGDTrainer
A trainer for a model which uses SGD.
|
SGDObjective
An interface for a loss function that can produce the loss and gradient incurred by
a single prediction.
|
Class and Description |
---|
SGDObjective
An interface for a loss function that can produce the loss and gradient incurred by
a single prediction.
|
Class and Description |
---|
SGDObjective
An interface for a loss function that can produce the loss and gradient incurred by
a single prediction.
|
Class and Description |
---|
SGDObjective
An interface for a loss function that can produce the loss and gradient incurred by
a single prediction.
|
Class and Description |
---|
SGDObjective
An interface for a loss function that can produce the loss and gradient incurred by
a single prediction.
|
Copyright © 2015–2021 Oracle and/or its affiliates. All rights reserved.