Examples with TrainingTester com.simiacryptus.mindseye.test.unit.TrainingTester used on opensource projects

Example 1 with TrainingTester

use of com.simiacryptus.mindseye.test.unit.TrainingTester in project MindsEye by SimiaCryptus.

the class PipelineTest method test.

/**
 * Test double.
 *
 * @param log       the log
 * @param layer     the layer
 * @param header    the header
 * @param inputDims the input dims
 * @return the double
 */
@Nullable
public TrainingTester.ComponentResult test(@Nonnull final NotebookOutput log, @Nonnull final Layer layer, final String header, @Nonnull final int[]... inputDims) {
    @Nonnull final Layer component = layer.copy();
    final Tensor[] randomize = randomize(inputDims);
    new SerializationTest().test(log, component, randomize);
    return new TrainingTester() {

        @Override
        protected void printHeader(NotebookOutput log) {
            log.h1(header);
        }
    }.test(log, component, randomize);
}

Example 2 with TrainingTester

use of com.simiacryptus.mindseye.test.unit.TrainingTester in project MindsEye by SimiaCryptus.

the class NLayerTest method test.

/**
 * Test double.
 *
 * @param log       the log
 * @param layer     the layer
 * @param inputDims the input dims
 * @return the double
 */
@Nullable
public TrainingTester.ComponentResult test(@Nonnull final NotebookOutput log, @Nonnull final Layer layer, @Nonnull final int[]... inputDims) {
    @Nonnull final Layer component = layer.copy();
    final Tensor[] randomize = randomize(inputDims);
    new SerializationTest().test(log, component, randomize);
    return new TrainingTester().test(log, component, randomize);
}