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