Class 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.

  • Constructor Details

    • LogisticRegressionTrainer

      public LogisticRegressionTrainer()
      Constructs a simple logistic regression, using AdaGrad with a learning rate of 1.0 as the gradient optimizer, training for 5 epochs. It's equivalent to this: new LinearSGDTrainer(new LogMulticlass(), new AdaGrad(1.0, 0.1), 5, Trainer.DEFAULT_SEED);