Skip to content

smoke-trees/forest-utils

Repository files navigation

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

About

Pypi Package for Smoketrees model zoo

Topics

Resources

Stars

Watchers

Forks

Packages

No packages published

Contributors 6

Languages