Ecosyste.ms: Awesome

An open API service indexing awesome lists of open source software.

Awesome Lists | Featured Topics | Projects

https://github.com/smoke-trees/forest-utils

Pypi Package for Smoketrees model zoo
https://github.com/smoke-trees/forest-utils

model-zoo pip pypi python3

Last synced: about 1 month ago
JSON representation

Pypi Package for Smoketrees model zoo

Awesome Lists containing this project

README

        

## forest-utils - A utility tool for SmokeTrees Model Zoo

[![PyPI version](https://badge.fury.io/py/forest-utils.svg)](https://badge.fury.io/py/forest-utils) [![Generic badge](https://img.shields.io/badge/modelzoo-website-brightgreen.svg)](https://smoketrees.dev/forest/)

This package will be useful for carrying out all the utilities of the SmokeTrees Model Zoo lovingly called SmokeTrees Forest.`

### Features

- [X] Pull down HDF5 (.h5) models from the zoo and load with keras
- [X] Pull down SavedModel (.pb) models from the zoo and load with keras
- [X] Pull down Checkpoints (.ckpt) models from the zoo and load with keras
- [X] Pull down the spacy model from the zoo and load them into an nlp pipeline
- [X] Pull down tokenizers and models in accordance with huggingface model architectures and reuse them
- [X] Pull down datasets so as to use them in your own projects

### Steps to install package from PyPI

```bash
pip install forest-utils
```

### Steps to install the package from source code

- Clone the repo by the following command
``` bash
git clone https://github.com/smoke-trees/forest-utils.git
```
- Change the directory to the package
``` bash
cd forest-utils
```
- Execute the following command
``` bash
pip install -e .
```

### Example Usage

- Load Model using Tensorflow

``` Python
from forest_utils import export_keras

model = export_keras.ModelFromH5().model
```

- Load Dataset

``` Python
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