@ Override public int [ ] predict ( INDArray input ) { INDArray output = activate ( input , false , LayerWorkspaceMgr . noWorkspacesImmutable ( ) ) ; int [ ] ret = new int [ input . rows ( ) ] ; for ( int i = 0 ; i < ret . length ; i ++ ) ret [ i ] = Nd4j . getBlasWrapper ( ) . iamax ( output . getRow ( i ) ) ; return ret ; }
Returns the predictions for each example in the dataset