Package | Description |
---|---|
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.
|
Modifier and Type | Method and Description |
---|---|
LabelObjective |
LinearSGDOptions.getLoss()
Returns the loss function specified in the arguments.
|
Constructor and Description |
---|
LinearSGDTrainer(LabelObjective objective,
StochasticGradientOptimiser optimiser,
int epochs,
int loggingInterval,
int minibatchSize,
long seed)
Constructs an SGD trainer for a linear model.
|
LinearSGDTrainer(LabelObjective objective,
StochasticGradientOptimiser optimiser,
int epochs,
int loggingInterval,
long seed)
Sets the minibatch size to 1.
|
LinearSGDTrainer(LabelObjective objective,
StochasticGradientOptimiser optimiser,
int epochs,
long seed)
Sets the minibatch size to 1 and the logging interval to 1000.
|
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.
|
Copyright © 2015–2021 Oracle and/or its affiliates. All rights reserved.