Skip to main content
Join the official 2019 Python Developers SurveyStart the survey!

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.

Files for mltool, version 0.6
Filename, size File type Python version Upload date Hashes
Filename, size mltool-0.6.tar.gz (9.8 kB) File type Source Python version None Upload date Hashes View hashes

Supported by

Elastic Elastic Search Pingdom Pingdom Monitoring Google Google BigQuery Sentry Sentry Error logging AWS AWS Cloud computing DataDog DataDog Monitoring Fastly Fastly CDN SignalFx SignalFx Supporter DigiCert DigiCert EV certificate StatusPage StatusPage Status page