org.deeplearning4j.spark.impl.layer

Class DL4jWorker

java.lang.Object

org.deeplearning4j.spark.impl.layer.DL4jWorker

All Implemented Interfaces:

Serializable, org.apache.spark.api.java.function.Function<org.nd4j.linalg.dataset.DataSet,org.nd4j.linalg.api.ndarray.INDArray>

public class DL4jWorker
extends 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

Author:

josh, Adam Gibson

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
DL4jWorker(String json, org.nd4j.linalg.api.ndarray.INDArray params)

Method Summary

Modifier and Type	Method and Description
org.nd4j.linalg.api.ndarray.INDArray	call(org.nd4j.linalg.dataset.DataSet v1)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

DL4jWorker

public DL4jWorker(String json,
                  org.nd4j.linalg.api.ndarray.INDArray params)

Method Detail

call

public org.nd4j.linalg.api.ndarray.INDArray call(org.nd4j.linalg.dataset.DataSet v1)
                                          throws Exception

Specified by:

call in interface org.apache.spark.api.java.function.Function<org.nd4j.linalg.dataset.DataSet,org.nd4j.linalg.api.ndarray.INDArray>

Throws:

Exception