Details
-
Task
-
Status: Closed
-
Major
-
Resolution: Fixed
-
2.0.0
-
None
Description
Automatically generate classifications map from model's config.json.
Currently, the implementations utilizing ONNX Runtime require a Map that stores the model-assigned value along with the human readable name for each value. This map must be created manually:
Map<Integer, String> classifications = new HashMap<>();
classifications.put(0, "negative");
classifications.put(1, "positive");
How to create this map is determined by looking at the model's config.json file. This task is to have OpenNLP read the config.json file and make the map automatically instead of requiring the user to make it manually.