Java example source code file (BaseOutputLayer.java)

This example Java source code file (BaseOutputLayer.java) is included in the alvinalexander.com "Java Source Code Warehouse" project. The intent of this project is to help you "Learn Java by Example" ^TM.

Learn more about this Java project at its project page.

Java - Java tags/keywords

baseoutputlayer, builder, data, equalsandhashcode, feedforwardlayer, lossfunction, noargsconstructor, string, tostring

The BaseOutputLayer.java Java example source code

package org.deeplearning4j.nn.conf.layers;

import lombok.Data;
import lombok.EqualsAndHashCode;
import lombok.NoArgsConstructor;
import lombok.ToString;

import org.nd4j.linalg.lossfunctions.LossFunctions.LossFunction;

@Data @NoArgsConstructor
@ToString(callSuper = true)
@EqualsAndHashCode(callSuper = true)
public abstract class BaseOutputLayer extends FeedForwardLayer {
	protected LossFunction lossFunction;
    protected String customLossFunction;

    protected BaseOutputLayer(Builder builder) {
    	super(builder);
        this.lossFunction = builder.lossFunction;
        this.customLossFunction = builder.customLossFunction;
    }
    
    public static abstract class Builder<T extends Builder extends FeedForwardLayer.Builder {
        protected LossFunction lossFunction = LossFunction.RMSE_XENT;
        protected String customLossFunction;

        public Builder() {}

        public Builder(LossFunction lossFunction) {
            this.lossFunction = lossFunction;
        }

        public T lossFunction(LossFunction lossFunction) {
            this.lossFunction = lossFunction;
            return (T)this;
        }

        public T customLossFunction(String customLossFunction) {
            this.customLossFunction = customLossFunction;
            return (T)this;
        }
    }
}