nnet-test-utils.h

Go to the documentation of this file.

// nnet3/nnet-test-utils.h

// Copyright   2015  Johns Hopkins University (author: Daniel Povey)
// Copyright   2016  Daniel Galvez
// See ../../COPYING for clarification regarding multiple authors
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//  http://www.apache.org/licenses/LICENSE-2.0
//
// THIS CODE IS PROVIDED *AS IS* BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
// KIND, EITHER EXPRESS OR IMPLIED, INCLUDING WITHOUT LIMITATION ANY IMPLIED
// WARRANTIES OR CONDITIONS OF TITLE, FITNESS FOR A PARTICULAR PURPOSE,
// MERCHANTABLITY OR NON-INFRINGEMENT.
// See the Apache 2 License for the specific language governing permissions and
// limitations under the License.

#ifndef KALDI_NNET3_NNET_TEST_UTILS_H_
#define KALDI_NNET3_NNET_TEST_UTILS_H_

#include "nnet3/nnet-nnet.h"
#include "nnet3/nnet-utils.h"
#include "nnet3/nnet-example.h"

namespace kaldi {
namespace nnet3 {

struct NnetGenerationOptions {
  bool allow_context;
  bool allow_nonlinearity;
  bool allow_recursion;
  bool allow_clockwork;
  bool allow_multiple_inputs;
  bool allow_multiple_outputs;
  bool allow_final_nonlinearity;
  bool allow_use_of_x_dim;
  bool allow_ivector;
  bool allow_statistics_pooling;
  // if set to a value >0, the output-dim of the network
  // will be set to this value.
  int32 output_dim;

  NnetGenerationOptions():
      allow_context(true),
      allow_nonlinearity(true),
      allow_recursion(true),
      allow_clockwork(true),
      allow_multiple_inputs(true),
      allow_multiple_outputs(false),
      allow_final_nonlinearity(true),
      allow_use_of_x_dim(true),
      allow_ivector(false),
      allow_statistics_pooling(true),
      output_dim(-1) { }
};

void GenerateConfigSequence(const NnetGenerationOptions &opts,
                            std::vector<std::string> *configs);

void GenerateConfigSequenceCompositeBlock(const NnetGenerationOptions &opts,
                                          std::vector<std::string> *configs);

void ComputeExampleComputationRequestSimple(
    const Nnet &nnet,
    ComputationRequest *request,
    std::vector<Matrix<BaseFloat> > *inputs);

Component *GenerateRandomSimpleComponent();


bool NnetParametersAreIdentical(const Nnet &nnet1,
                                const Nnet &nnet2,
                                BaseFloat threshold);


void GenerateSimpleNnetTrainingExample(
    int32 num_supervised_frames,
    int32 left_context,
    int32 right_context,
    int32 input_dim,
    int32 output_dim,
    int32 ivector_dim,
    NnetExample *example);


bool ExampleApproxEqual(const NnetExample &eg1,
                        const NnetExample &eg2,
                        BaseFloat delta);

} // namespace nnet3
} // namespace kaldi

#endif