Welcome to mirror list, hosted at ThFree Co, Russian Federation.

pipeline.hh « interpolate « lm - github.com/kpu/kenlm.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
blob: a2b8e1c87a93f070e9fb3b51437530bafd1b8e87 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
#ifndef LM_INTERPOLATE_PIPELINE_H
#define LM_INTERPOLATE_PIPELINE_H

#include "../common/model_buffer.hh"
#include "../../util/fixed_array.hh"
#include "../../util/stream/config.hh"

#include <cstddef>
#include <string>

namespace lm { namespace interpolate {

struct Config {
  std::vector<float> lambdas;
  util::stream::SortConfig sort;
  std::size_t BufferSize() const { return sort.buffer_size; }
};

void Pipeline(util::FixedArray<ModelBuffer> &models, const Config &config, int write_file);

}} // namespaces
#endif // LM_INTERPOLATE_PIPELINE_H