Package for Smoketrees model zoo
Project description
forest-utils - A utility tool 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-1.0.1.tar.gz
(7.6 kB
view hashes)
Built Distribution
Close
Hashes for forest_utils-1.0.1-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 65854bc890711e9e25c3da1a2f22b8fb70e108e44750242080a2e4c2db63cdf5 |
|
MD5 | 7d0a29efd5bf7d3360874676bbea0b5c |
|
BLAKE2b-256 | f6191e1d94e081308e790afa318266ae29d1855f8b2a1027f6caf864775c7ee2 |