public MultiLayerNetwork fit ( JavaRDD < DataSet > trainingData ) { if ( Nd4j . getExecutioner ( ) instanceof GridExecutioner ) ( ( GridExecutioner ) Nd4j . getExecutioner ( ) ) . flushQueue ( ) ; trainingMaster . executeTraining ( this , trainingData ) ; network . incrementEpochCount ( ) ; return network ; }
Fit the DataSet RDD