Package | Description |
---|---|
org.tribuo.common.sgd |
Provides the base classes for models trained with stochastic gradient descent.
|
org.tribuo.math |
Contains the implementation of Tribuo's math library, it's gradient descent optimisers, kernels and a set of
math related utils.
|
Modifier and Type | Class and Description |
---|---|
class |
AbstractSGDTrainer<T extends Output<T>,U,V extends Model<T>,X extends FeedForwardParameters>
A trainer for a model which uses SGD.
|
Modifier and Type | Field and Description |
---|---|
protected FeedForwardParameters |
AbstractSGDModel.modelParameters
The weights for this model.
|
Modifier and Type | Method and Description |
---|---|
FeedForwardParameters |
AbstractSGDModel.getModelParameters()
Returns a copy of the model parameters.
|
Constructor and Description |
---|
AbstractSGDModel(String name,
ModelProvenance provenance,
ImmutableFeatureMap featureIDMap,
ImmutableOutputInfo<T> outputIDInfo,
FeedForwardParameters weights,
boolean generatesProbabilities,
boolean addBias)
Constructs a linear model trained via SGD.
|
Modifier and Type | Class and Description |
---|---|
class |
LinearParameters
A
Parameters for producing linear models. |
Modifier and Type | Method and Description |
---|---|
FeedForwardParameters |
FeedForwardParameters.copy()
Returns a copy of the parameters.
|
Copyright © 2015–2021 Oracle and/or its affiliates. All rights reserved.