Skip to main content

A Scikit-learn style wrapper for H2O estimators.

Project description

Wetsuit

image image

A Scikit-Learn wrapper for H2O Estimators.

Why Wetsuit

While H2O Estimators have the .fit() and .predict() methods of the Scikit-Learn API, they don't always function as expected, especially with Pipeline objects. This package contains two estimators and a single transformer to remedy.

For example. the H2OEstimator.fit() method expects two H2OFrame objects, vice pandas DataFrame or numpy NDArray objects. Wetsuit provides two options for handling this behavior:

  • WetsuitRegressor and WetsuitClassifier classes that wrap H2OEstimator objects and handle type conversion automatically, within the .fit() and .predict() methods.
  • H2oFrameTransformer class that converts both DataFrame and NDArray objects to H2OFrame objects via .fit_transform(), and an .inverse_transform() method to convert back.

Install

Create a virtual environment with Python >= 3.9 and install from git:

pip install git+https://github.com/chris-santiago/wetsuit.git

Use

Documentation

Documentation hosted on Github Pages: https://chris-santiago.github.io/wetsuit/

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

wetsuit-0.1.0.tar.gz (7.4 kB view hashes)

Uploaded Source

Built Distribution

wetsuit-0.1.0-py3-none-any.whl (7.6 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