Package | Description |
---|---|
org.tribuo.classification.sgd.linear |
Provides an implementation of a classification linear model using Stochastic Gradient Descent.
|
org.tribuo.common.sgd |
Provides the base classes for models trained with stochastic gradient descent.
|
Modifier and Type | Class and Description |
---|---|
class |
LinearSGDTrainer
A trainer for a linear classifier using SGD.
|
class |
LogisticRegressionTrainer
A logistic regression trainer that uses a reasonable objective, optimiser,
number of epochs and minibatch size.
|
Modifier and Type | Class and Description |
---|---|
class |
AbstractLinearSGDTrainer<T extends Output<T>,U>
A trainer for a linear model which uses SGD.
|
Copyright © 2015–2021 Oracle and/or its affiliates. All rights reserved.