file_corpus.h

Go to the documentation of this file.

#ifndef META_FILE_CORPUS_H_
#define META_FILE_CORPUS_H_

#include <string>
#include <vector>
#include <utility>
#include "corpus/corpus.h"

namespace meta
{
namespace corpus
{

class file_corpus : public corpus
{
  public:
    file_corpus(const std::string& prefix, const std::string& doc_list,
                std::string encoding);

    bool has_next() const override;

    document next() override;

    uint64_t size() const override;

  private:
    uint64_t cur_;

    std::string prefix_;

    std::vector<std::pair<std::string, class_label>> docs_;
};
}
}

#endif