-
Notifications
You must be signed in to change notification settings - Fork 13
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
ONNX support, bug fix, user friendly APIs (#10)
* bumbed version through version.py file * gitignore for built files * Readme.md linked to example repo * assets for onnx tests * onnx support * user friendly apis * tests for onnx support and new user friendly APIs * minor nit * testing against redisai edge * pandas to test requirments
- Loading branch information
Showing
13 changed files
with
276 additions
and
129 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -1,4 +1,5 @@ | ||
.project | ||
.pydevproject | ||
*.pyc | ||
.venv/ | ||
.venv/ | ||
redisai.egg-info |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -1 +1,22 @@ | ||
from .version import __version__ | ||
from .client import (Client, Tensor, BlobTensor, DType, Device, Backend) | ||
|
||
|
||
def save_model(*args, **kwargs): | ||
""" | ||
Importing inside to avoid loading the TF/PyTorch/ONNX | ||
into the scope unnecessary. This function wraps the | ||
internal save model utility to make it user friendly | ||
""" | ||
from .model import Model | ||
Model.save(*args, **kwargs) | ||
|
||
|
||
def load_model(*args, **kwargs): | ||
""" | ||
Importing inside to avoid loading the TF/PyTorch/ONNX | ||
into the scope unnecessary. This function wraps the | ||
internal load model utility to make it user friendly | ||
""" | ||
from .model import Model | ||
return Model.load(*args, **kwargs) |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,5 @@ | ||
# Store the version here so: | ||
# 1) we don't load dependencies by storing it in __init__.py | ||
# 2) we can import it in setup.py for the same reason | ||
# 3) we can import it into your module module | ||
__version__ = '0.3.0' |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -1,3 +1,6 @@ | ||
numpy | ||
torch | ||
tensorflow | ||
tensorflow | ||
onnx | ||
skl2onnx | ||
pandas |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.