/hilearn

A small library of machine learning methods and data visualization plots

Primary LanguagePythonApache License 2.0Apache-2.0

PyPI Docs Build Status

HiLearn

A small library of machine learning models and utility & plotting functions:

  1. a set of utility functions, e.g., wrap function for cross-validation on regression and classification models
  2. a set of small models, e.g., mixture of linear regression model
  3. a set of plotting functions, e.g., corr_plot, ROC_curve, PR_curve

How to install?

Easy install with pip by pip install hilearn for released version or the latest version on github (less stable though)

pip install -U git+https://github.com/huangyh09/hilearn

If you don't have the root permission, add --user.

Documentation

See the documentation for how to use, e.g., cross-validation and plotting functions.