Package for Smoketrees model zoo
Project description
Forest Utils package for smoketrees model zoo
This package will be useful for carrying out all the utilities of the SmokeTrees Model Zoo lovingly called SmokeTrees Forest.`
Features
- Pull down HDF5 (.h5) models from the zoo and load with keras
- Pull down SavedModel (.pb) models from the zoo and load with keras
- Pull down Checkpoints (.ckpt) models from the zoo and load with keras
- Pull down the spacy model from the zoo and load them into an nlp pipeline
- Pull down tokenizers and models in accordance with huggingface model architectures and reuse them
- Pull down datasets so as to use them in your own projects
Steps to install package from PyPI
pip install forest-utils
Steps to install the package from source code
- Clone the repo by the following command
git clone https://github.com/smoke-trees/forest-utils.git
- Change the directory to the package
cd forest-utils
- Execute the following command
pip install -e .
Example Usage
- Load Model using Tensorflow
from forest_utils import export_keras
model = export_keras.ModelFromH5().model
- Load Dataset
from forest_utils import datasets
tweets = datasets.Dataset().get_emo_tweets()
After pulling down the model use it for predictions and other evalutaion functionalities.
See SmokeTrees ModelZoo for more usage examples
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
forest-utils-0.0.9.tar.gz
(7.5 kB
view hashes)
Built Distribution
Close
Hashes for forest_utils-0.0.9-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 51a09e42e6e3862065c0fea266a464e16498be87f165675077cd3d5e7c17dd69 |
|
MD5 | f316977e7761ce4caea970beaac1d1ca |
|
BLAKE2b-256 | 5a33214e768d879a644d3aaef540351b880cb6f4befe154f9f67188b2dbf7ff4 |