training_data.h

Go to the documentation of this file.

#ifndef META_PARSER_TRAINING_DATA_H_
#define META_PARSER_TRAINING_DATA_H_

#include "parser/sr_parser.h"

namespace meta
{
namespace parser
{

class sr_parser::training_data
{
  public:
    training_data(std::vector<parse_tree>& trees,
                  std::default_random_engine::result_type seed);

    transition_map preprocess();

    void shuffle();

    size_t size() const;

    const parse_tree& tree(size_t idx) const;

    const std::vector<trans_id>& transitions(size_t idx) const;

  private:
    std::vector<parse_tree>& trees_;

    std::vector<std::vector<trans_id>> all_transitions_;

    std::vector<size_t> indices_;

    std::default_random_engine rng_;
};
}
}
#endif