de.datexis.sector

Class SectorAnnotator.Builder

java.lang.Object

de.datexis.sector.SectorAnnotator.Builder

Enclosing class:

SectorAnnotator

public static class SectorAnnotator.Builder
extends Object

Builder pattern for creating new SECTOR Annotators.

Field Summary

Fields

Modifier and Type	Field and Description
protected org.nd4j.linalg.activations.Activation	activation
protected Encoder[]	encoders
protected org.nd4j.linalg.lossfunctions.ILossFunction	lossFunc
protected boolean	requireSubsampling

Constructor Summary

Constructors

Constructor and Description
Builder()

Method Summary

Modifier and Type	Method and Description
SectorAnnotator	build()
SectorAnnotator.Builder	enableTrainingUI(boolean enable)
SectorAnnotator.Builder	pretrain(Dataset train) pretrain encoders
SectorAnnotator.Builder	withDataset(String datasetName, WordHelpers.Language lang)
SectorAnnotator.Builder	withId(String id)
SectorAnnotator.Builder	withInputEncoders(String desc, Encoder bagEncoder, Encoder embEncoder, Encoder flagEncoder)
SectorAnnotator.Builder	withLossFunction(org.nd4j.linalg.lossfunctions.ILossFunction lossFunc, org.nd4j.linalg.activations.Activation activation, boolean requireSubsampling)
SectorAnnotator.Builder	withLossFunction(org.nd4j.linalg.lossfunctions.LossFunctions.LossFunction lossFunc, org.nd4j.linalg.activations.Activation activation, boolean requireSubsampling)
SectorAnnotator.Builder	withModelParams(int ffwLayerSize, int lstmLayerSize, int embeddingLayerSize)
SectorAnnotator.Builder	withTargetEncoder(Encoder targetEncoder)
SectorAnnotator.Builder	withTrainingParams(double learningRate, double dropOut, int examplesPerEpoch, int batchSize, int numEpochs)
SectorAnnotator.Builder	withTrainingParams(double learningRate, double dropOut, int examplesPerEpoch, int maxTimeSeriesLength, int batchSize, int numEpochs)

Methods inherited from class java.lang.Object

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

Field Detail

encoders

protected Encoder[] encoders

lossFunc

protected org.nd4j.linalg.lossfunctions.ILossFunction lossFunc

activation

protected org.nd4j.linalg.activations.Activation activation

requireSubsampling

protected boolean requireSubsampling

Constructor Detail

Builder

public Builder()

Method Detail

withId

public SectorAnnotator.Builder withId(String id)

withDataset

public SectorAnnotator.Builder withDataset(String datasetName,
                                           WordHelpers.Language lang)

withLossFunction

public SectorAnnotator.Builder withLossFunction(org.nd4j.linalg.lossfunctions.LossFunctions.LossFunction lossFunc,
                                                org.nd4j.linalg.activations.Activation activation,
                                                boolean requireSubsampling)

withLossFunction

public SectorAnnotator.Builder withLossFunction(org.nd4j.linalg.lossfunctions.ILossFunction lossFunc,
                                                org.nd4j.linalg.activations.Activation activation,
                                                boolean requireSubsampling)

withModelParams

public SectorAnnotator.Builder withModelParams(int ffwLayerSize,
                                               int lstmLayerSize,
                                               int embeddingLayerSize)

withTrainingParams

public SectorAnnotator.Builder withTrainingParams(double learningRate,
                                                  double dropOut,
                                                  int examplesPerEpoch,
                                                  int batchSize,
                                                  int numEpochs)

withTrainingParams

public SectorAnnotator.Builder withTrainingParams(double learningRate,
                                                  double dropOut,
                                                  int examplesPerEpoch,
                                                  int maxTimeSeriesLength,
                                                  int batchSize,
                                                  int numEpochs)

withInputEncoders

public SectorAnnotator.Builder withInputEncoders(String desc,
                                                 Encoder bagEncoder,
                                                 Encoder embEncoder,
                                                 Encoder flagEncoder)

withTargetEncoder

public SectorAnnotator.Builder withTargetEncoder(Encoder targetEncoder)

enableTrainingUI

public SectorAnnotator.Builder enableTrainingUI(boolean enable)

pretrain

public SectorAnnotator.Builder pretrain(Dataset train)

pretrain encoders

build

public SectorAnnotator build()