Skip to main content

Package for Smoketrees model zoo

Project description

forest-utils - A utility tool for SmokeTrees Model Zoo

PyPI version Generic badge

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


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.2.tar.gz (7.9 kB view hashes)

Uploaded Source

Built Distribution

forest_utils-1.0.2-py3-none-any.whl (9.5 kB view hashes)

Uploaded Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page