Agent class to support ML/RL model inference via ONNX. More...
#include "MLModelRunner/ONNXModelRunner/onnx.h"#include "MLModelRunner/ONNXModelRunner/utils.h"#include <stdio.h>#include <string>
Include dependency graph for agent.h:
This graph shows which files directly or indirectly include this file:
Go to the source code of this file.
Classes | |
| class | MLBridge::Agent |
| Agent is a wrapper around the ONNXModel class, interfaces with the Environment class to support ML/RL model inference via ONNXModel. More... | |
Namespaces | |
| namespace | MLBridge |
Detailed Description
Agent class to support ML/RL model inference via ONNX.
Usage:
- Construct an agent object with the path to the ONNX model
- Call computeAction() to get the action from the model
Definition in file agent.h.
