org.tribuo.classification.sgd.linear

Class LogisticRegressionTrainer

java.lang.Object

org.tribuo.classification.sgd.linear.LinearSGDTrainer

org.tribuo.classification.sgd.linear.LogisticRegressionTrainer

All Implemented Interfaces:

com.oracle.labs.mlrg.olcut.config.Configurable, com.oracle.labs.mlrg.olcut.provenance.Provenancable<TrainerProvenance>, Trainer<Label>, WeightedExamples

public class LogisticRegressionTrainer
extends LinearSGDTrainer

A logistic regression trainer that uses a reasonable objective, optimiser, number of epochs and minibatch size. If you wish to modify any of these aspects, you can create your own LinearSGDTrainer.

This is strictly a convenience class for folks who are looking for a simple logistic regression.

Field Summary

Fields inherited from interface org.tribuo.Trainer

DEFAULT_SEED

Constructor Summary

Constructors

Constructor and Description
LogisticRegressionTrainer()

Method Summary

Methods inherited from class org.tribuo.classification.sgd.linear.LinearSGDTrainer

getInvocationCount, getProvenance, postConfig, setShuffle, toString, train

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, wait, wait, wait

Methods inherited from interface org.tribuo.Trainer

train

Constructor Detail

LogisticRegressionTrainer

public LogisticRegressionTrainer()