org.nd4j.autodiff.loss.LossInfo Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of nd4j-api Show documentation

There is a newer version: 1.0.0-M2.1

package org.nd4j.autodiff.loss;

import lombok.Builder;
import lombok.Getter;
import org.nd4j.autodiff.samediff.SDVariable;

/**
 * Information and variables for a loss function. Used with {@link LossFunctions}
 *
 * @author Alex Black
 */
@Builder(builderClassName = "Builder")
@Getter
public class LossInfo {
    private String lossName;
    private LossFunctions.Reduction reduction;
    private SDVariable loss;
    private SDVariable label;
    private SDVariable predictions;
    private SDVariable weights;

}