METHOD_0 ( ) { org . IDENT_0 . IDENT_1 . conf . IDENT_2 conf = new org . IDENT_0 . IDENT_1 . conf . IDENT_3 . Builder ( ) . METHOD_1 ( IDENT_4 . IDENT_5 ) . list ( ) . METHOD_2 ( new org . IDENT_0 . spark . impl . IDENT_6 . IDENT_7 . Builder ( ) . METHOD_3 ( 4 ) . METHOD_4 ( 3 ) . build ( ) ) . METHOD_2 ( new org . IDENT_0 . spark . impl . IDENT_6 . METHOD_5 ( IDENT_8 . IDENT_9 ) ) . METHOD_2 ( new org . IDENT_0 . spark . impl . IDENT_6 . IDENT_10 . Builder ( IDENT_11 . IDENT_12 . IDENT_13 ) . METHOD_3 ( 3 ) . METHOD_4 ( 3 ) . METHOD_6 ( IDENT_14 . IDENT_15 ) . build ( ) ) . build ( ) ; org . IDENT_0 . IDENT_1 . IDENT_6 . IDENT_16 net = new org . IDENT_0 . IDENT_1 . IDENT_6 . METHOD_7 ( conf ) ; net . init ( ) ; org . IDENT_0 . spark . impl . IDENT_6 . List < org . nd4j . linalg . dataset . IDENT_17 > IDENT_18 = org . IDENT_0 . spark . impl . IDENT_6 . Arrays . asList ( new org . nd4j . linalg . dataset . METHOD_8 ( org . nd4j . linalg . factory . Nd4j . METHOD_9 ( new int [ ] { 1 , 4 , 5 } ) , org . nd4j . linalg . factory . Nd4j . create ( new double [ ] { 1 , 1 , 1 , 0 , 0 } ) ) , new org . nd4j . linalg . dataset . METHOD_8 ( org . nd4j . linalg . factory . Nd4j . METHOD_9 ( new int [ ] { 1 , 4 , 5 } ) , org . nd4j . linalg . factory . Nd4j . create ( new double [ ] { 1 , 1 , 1 , 1 , 0 } ) ) , new org . nd4j . linalg . dataset . METHOD_8 ( org . nd4j . linalg . factory . Nd4j . METHOD_9 ( new int [ ] { 1 , 4 , 5 } ) , org . nd4j . linalg . factory . Nd4j . create ( new double [ ] { 1 , 1 , 1 , 1 , 1 } ) ) ) ; org . IDENT_0 . spark . impl . IDENT_6 . Map < java . lang . Integer , org . nd4j . linalg . api . ndarray . INDArray > expected = new org . IDENT_0 . spark . impl . IDENT_6 . HashMap ( ) ; org . IDENT_0 . spark . impl . IDENT_6 . List < IDENT_19 . IDENT_20 < java . lang . Integer , IDENT_19 . IDENT_20 < org . nd4j . linalg . api . ndarray . INDArray , org . nd4j . linalg . api . ndarray . INDArray > > > IDENT_21 = new org . IDENT_0 . spark . impl . IDENT_6 . ArrayList ( ) ; int count = 0 ; int IDENT_22 = 0 ; org . IDENT_0 . spark . impl . IDENT_6 . IDENT_23 r = new org . IDENT_0 . spark . impl . IDENT_6 . METHOD_10 ( INT_0 ) ; int i = 0 ; for ( org . nd4j . linalg . dataset . METHOD_8 d : IDENT_18 ) { org . nd4j . linalg . api . ndarray . INDArray f = d . METHOD_11 ( ) ; org . nd4j . linalg . api . ndarray . INDArray IDENT_24 = d . METHOD_12 ( ) ; IDENT_21 . add ( new IDENT_19 . METHOD_13 ( i , new IDENT_19 . METHOD_13 ( f , IDENT_24 ) ) ) ; org . nd4j . linalg . api . ndarray . INDArray out = net . output ( f , false , IDENT_24 , null ) ; expected . put ( ( i ++ ) , out ) ; } org . apache . spark . api . java . IDENT_25 < java . lang . Integer , IDENT_19 . METHOD_13 < org . nd4j . linalg . api . ndarray . INDArray , org . nd4j . linalg . api . ndarray . INDArray > > IDENT_26 = IDENT_27 . METHOD_14 ( IDENT_21 ) ; org . IDENT_0 . spark . impl . IDENT_6 . IDENT_28 IDENT_29 = new org . IDENT_0 . spark . impl . IDENT_6 . METHOD_15 ( IDENT_27 , net , null ) ; org . IDENT_0 . spark . impl . IDENT_6 . Map < java . lang . Integer , org . nd4j . linalg . api . ndarray . INDArray > map = IDENT_29 . METHOD_16 ( IDENT_26 , 16 ) . METHOD_17 ( ) ; for ( i = 0 ; i < ( expected . size ( ) ) ; i ++ ) { org . nd4j . linalg . api . ndarray . INDArray exp = expected . get ( i ) ; org . nd4j . linalg . api . ndarray . INDArray IDENT_30 = map . get ( i ) ; "<AssertPlaceHolder>" ; } } get ( int ) { return IDENT_31 . get ( i ) ; }
org . junit . Assert . assertEquals ( exp , IDENT_30 )