A wrapper for ONNX models that adheres to the instancelib specification
Project description
instancelib-onnx
ONNX extension for instancelib.
Installation
You can install this package as follows:
pip install instancelib-onnx
Or by cloning this repo and issuing:
python setup.py
You will need at least Python 3.8 to use this library.
Usage
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(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.1.tar.gz
(19.1 kB
view hashes)
Built Distribution
Close
Hashes for instancelib_onnx-0.1.1-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 595f314cf4eb9b370c0ab401d3aa75cb8917c04b173e77c1ccf5f58a3ce30a5c |
|
MD5 | da8935b5968e37203689395d08079ae5 |
|
BLAKE2b-256 | 53815edaef9b0bb9e389ba1fc9b47030a53b2e809943aa4a5c1628f556aafce7 |