Agent is a wrapper around the ONNXModel class, interfaces with the Environment class to support ML/RL model inference via ONNXModel. More...
#include <agent.h>
Collaboration diagram for MLBridge::Agent:
Public Member Functions | |
| Agent (std::string model_path) | |
| unsigned | computeAction (Observation &obs) |
| Runs the ONNX model on the input Observation and returns the output. | |
Private Attributes | |
| ONNXModel * | model |
Detailed Description
Agent is a wrapper around the ONNXModel class, interfaces with the Environment class to support ML/RL model inference via ONNXModel.
Constructor & Destructor Documentation
◆ Agent()
Member Function Documentation
◆ computeAction()
| unsigned MLBridge::Agent::computeAction | ( | Observation & | obs | ) |
Member Data Documentation
◆ model
The documentation for this class was generated from the following files:
