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
Metadata
1 maintainer
Release files for model-wrapper 1.1.4
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.4.tar.gz | 24.9 kB | Details |
Release files / model_wrapper-1.1.4.tar.gz
| Download URL | model_wrapper-1.1.4.tar.gz |
|---|---|
| Size | 24.9 kB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
8d654bae3fca0cec75242500af158dd335bd2d83a50c25e8a6eaf820d1c1b12e
|
|
BLAKE2b-256 checksum How to use checksums |
17cfc7e0e2fb2d0d183ac72a6fe4ba68f8ca984b87ea3f0babd6e4a18237a6cb
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/5.0.0 CPython/3.9.18
|
