agent.h

Go to the documentation of this file.

//=== MLModelRunner/ONNXModelRunner/agent.h - Agent Model Helper - C++ -===//
//
// Part of the MLCompilerBridge Project, under the Apache License v2.0 with LLVM
// Exceptions. See the LICENSE file for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===---------------------------------------------------------------------===//
//===----------------------------------------------------------------------===//
 
#ifndef ONNX_MODELRUNNER_AGENT_H
#define ONNX_MODELRUNNER_AGENT_H
 
#include "MLModelRunner/ONNXModelRunner/onnx.h"
#include "MLModelRunner/ONNXModelRunner/utils.h"
 
#include <stdio.h>
#include <string>
 
namespace MLBridge {
class Agent {
  ONNXModel *model;
 
public:
  Agent(std::string model_path);
  unsigned computeAction(Observation &obs);
};
} // namespace MLBridge
 
#endif // ONNX_MODELRUNNER_AGENT_H