A wrapper for ONNX models that adheres to instancelib
Project description
instancelib-onnx
ONNX extension for instancelib
import instancelib as il
import instancelib as ilonnx
# Specify the model location and the label translation
model = ilonnx.build_data_model("example_models/data-model.onnx",
{0: "Bedrijfsnieuws", 1: "Games", 2: "Smartphones"})
Then you can use the normal instancelib functionality to interact with the model.
# Load a dataset with instancelib
env = il.read_excel_dataset("datasets/testdataset.xlsx", ["fulltext"], ["label"])
# Assess the performance like any other instancelib model
performance = il.classifier_performance(read_data_model, env.dataset, env.labels)
performance.confusion_matrix
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
instancelib-onnx-0.1.0.tar.gz
(18.9 kB
view hashes)
Built Distribution
Close
Hashes for instancelib_onnx-0.1.0-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 2c4a9fb1fed724ed72d4b83ac26eaea387419178cfcc66703f6ba59a97f892fb |
|
MD5 | 0ae24284c1840e6007bf26ce27fa0452 |
|
BLAKE2b-256 | bd77731e7f47796bfb9bafabcf02c8c7de9b5d1f0a2df0c9c1195404517e0c59 |