Usage Sample ''''''''''''
.. code:: python
from model_wrapper import SplitClassifyModelWrapper
classes = ['class1', 'class2', 'class3'...]
X = [[...], [...],]
y = [0, 0, 1, 2, 1...]
model = ...
wrapper = SplitClassifyModelWrapper(model, classes=classes)
wrapper.train(X, y, val_size=0.2)
X_test = [[...], [...],]
y_test = [0, 1, 1, 2, 1...]
result = wrapper.evaluate(X_test, y_test)
# 0.953125
result = wrapper.predict(X_test)
# [0, 1]
result = wrapper.predict_classes(X_test)
# ['class1', 'class2']
result = wrapper.predict_proba(X_test)
# ([0, 1], array([0.99439645, 0.99190724], dtype=float32))
result = wrapper.predict_classes_proba(X_test)
# (['class1', 'class2'], array([0.99439645, 0.99190724], dtype=float32))
1 maintainer
Release files for model-wrapper 1.1.3
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Source distribution (sdist)
| File | Size | Uploaded | |
|---|---|---|---|
| model-wrapper-1.1.3.tar.gz | 24.4 kB | Details |
Release files / model-wrapper-1.1.3.tar.gz
| Download URL | model-wrapper-1.1.3.tar.gz |
|---|---|
| Size | 24.4 kB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
689f396b2fa8d90a820b355850a5628fb62d6604cd991a91811ebc7faf1958d2
|
|
BLAKE2b-256 checksum How to use checksums |
63f147cde6a9f5606363b6b8ad08c3459148818ab8ab7c02e5c195fc46d5812c
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/5.0.0 CPython/3.9.18
|
