Skip to main content

Machine learning tool for regression.

Project description

mltool is a simple Learning to Rank tool to build a regression tree based ranking model. Currently mltool supports simple CART and Random Forest. The implementation is strongly inspired by rt-rank, but it lacks the support of Gradient Boosting Regression Trees.

Overview

Features

Despite rt-rank, mltool provides:

  • A parameter to set the seed for the random number generator to make the training deterministic

  • Serializable model

  • Show feature information gain statistics at the end of the training

  • Can be used as an API

Some highlights compared to other Random Forest implemented for Python:

  • The implementation of the Random Forest makes use of numpy and it is quite optimized

  • Parallel Random Forest training

Installation

Install from PyPI using pip:

$ pip install mltool

Now you can run mltool:

$ mltool -h

Documentation

The documentation is hosted by Read the Docs at the following url: http://readthedocs.org/docs/mltool/en/latest/

Future

  • add support for Stochastic Gradient Boosting Regression Trees

  • add support for simple regression and/or classification (i.e. not just focus on ranking)

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

mltool-0.6.tar.gz (9.8 kB view hashes)

Uploaded Source

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