public class VaeReconstructionErrorWithKeyFunction<K> extends BasePairFlatMapFunctionAdaptee<java.util.Iterator<scala.Tuple2<K,org.nd4j.linalg.api.ndarray.INDArray>>,K,java.lang.Double>
ReconstructionDistribution
adapter
Constructor and Description |
---|
VaeReconstructionErrorWithKeyFunction(org.apache.spark.broadcast.Broadcast<org.nd4j.linalg.api.ndarray.INDArray> params,
org.apache.spark.broadcast.Broadcast<java.lang.String> jsonConfig,
int batchSize) |
call