Package | Description |
---|---|
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.
|
Modifier and Type | Method and Description |
---|---|
MultiLabelObjective |
LinearSGDOptions.getLoss()
Returns the loss function specified in the arguments.
|
Constructor and Description |
---|
LinearSGDTrainer(MultiLabelObjective objective,
StochasticGradientOptimiser optimiser,
int epochs,
int loggingInterval,
int minibatchSize,
long seed)
Constructs an SGD trainer for a linear model.
|
LinearSGDTrainer(MultiLabelObjective objective,
StochasticGradientOptimiser optimiser,
int epochs,
int loggingInterval,
long seed)
Constructs an SGD trainer for a linear model.
|
LinearSGDTrainer(MultiLabelObjective objective,
StochasticGradientOptimiser optimiser,
int epochs,
long seed)
Constructs an SGD trainer for a linear model.
|
Modifier and Type | Class and Description |
---|---|
class |
BinaryCrossEntropy
A multilabel version of binary cross entropy loss which expects logits.
|
class |
Hinge
Hinge loss, scores the correct value margin and any incorrect predictions -margin.
|
Copyright © 2015–2021 Oracle and/or its affiliates. All rights reserved.