Class LossMixtureDensity.Builder
- java.lang.Object
-
- org.nd4j.linalg.lossfunctions.impl.LossMixtureDensity.Builder
-
- Enclosing class:
- LossMixtureDensity
public static class LossMixtureDensity.Builder extends Object
-
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description LossMixtureDensity
build()
Creates a new instance of the mixture density cost function.LossMixtureDensity.Builder
gaussians(int aGaussians)
Specifies the number of gaussian functions to attempt fitting against the data.LossMixtureDensity.Builder
labelWidth(int aLabelWidth)
Specifies the width of the labels vector which also corresponds to the width of the 'mean' vector for each of the gaussian functions.
-
-
-
Method Detail
-
gaussians
public LossMixtureDensity.Builder gaussians(int aGaussians)
Specifies the number of gaussian functions to attempt fitting against the data.- Parameters:
aGaussians
- Number of gaussian functions to fit.- Returns:
- DynamicCustomOpsBuilder.
-
labelWidth
public LossMixtureDensity.Builder labelWidth(int aLabelWidth)
Specifies the width of the labels vector which also corresponds to the width of the 'mean' vector for each of the gaussian functions.- Parameters:
aLabelWidth
- Width of the labels vector.- Returns:
- DynamicCustomOpsBuilder.
-
build
public LossMixtureDensity build()
Creates a new instance of the mixture density cost function.- Returns:
- A new mixture density cost function built with the specified parameters.
-
-