Examples with Function org.nd4j.linalg.function.Function used on opensource projects

Example 1 with Function

use of org.nd4j.linalg.function.Function in project nd4j by deeplearning4j.

the class TwoPointApproximationTest method testLinspaceDerivative.

@Test
public void testLinspaceDerivative() throws Exception {
    String basePath = "/two_points_approx_deriv_numpy/";
    INDArray linspace = Nd4j.createNpyFromInputStream(new ClassPathResource(basePath + "x.npy").getInputStream());
    INDArray yLinspace = Nd4j.createNpyFromInputStream(new ClassPathResource(basePath + "y.npy").getInputStream());
    Function<INDArray, INDArray> f = new Function<INDArray, INDArray>() {

        @Override
        public INDArray apply(INDArray indArray) {
            return indArray.add(1);
        }
    };
    INDArray test = TwoPointApproximation.approximateDerivative(f, linspace, null, yLinspace, Nd4j.create(new double[] { Float.MIN_VALUE, Float.MAX_VALUE }));
    INDArray npLoad = Nd4j.createNpyFromInputStream(new ClassPathResource(basePath + "approx_deriv_small.npy").getInputStream());
    assertEquals(npLoad, test);
    System.out.println(test);
}