- All Implemented Interfaces:
- java.io.Serializable, org.apache.spark.api.java.function.Function<org.nd4j.linalg.dataset.DataSet,org.nd4j.linalg.api.ndarray.INDArray>
public class DL4jWorker
extends java.lang.Object
implements org.apache.spark.api.java.function.Function<org.nd4j.linalg.dataset.DataSet,org.nd4j.linalg.api.ndarray.INDArray>
This is considered the "Worker"
This is the code that will run the .fitDataSet() method on the network
the issue here is that this is getting called 1x per record
and before we could call it in a more controlled mini-batch setting
- See Also:
- Serialized Form