Class EmbedInstanceReader

src/model/embed_instance_reader.h:32–64

using ::deepx_core::InstanceReaderImpl;

class EmbedInstanceReader : public InstanceReaderImpl {
 protected:
  const GraphClient* graph_client_ = nullptr;
  const DeepClient* deep_client_ = nullptr;

  LineParser line_parser_;

 public:
  virtual bool InitGraphClient(const GraphClient* graph_client);
  virtual bool InitDeepClient(const DeepClient* deep_client);
  virtual void PostInit(const std::string& /*node_config*/) {}

  bool Open(const std::string& file) override {
    return line_parser_.Open(file);
  }

 protected:
  template <typename ValueType>
  bool NextInstanceBatch(Instance* inst, int batch,
                         std::vector<ValueType>* values) {
    if (!line_parser_.NextBatch<ValueType>(batch, values)) {
      line_parser_.Close();
      inst->clear_batch();
      return false;
    }
    return true;
  }

  // InstanceReaderImpl
  void InitX(Instance* /* inst */) override {}
  void InitXBatch(Instance* /*inst*/) override {}
  bool ParseLine() override { return true; }
};

std::unique_ptr<EmbedInstanceReader> NewEmbedInstanceReader(
    const std::string& name);