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 mltool using pip:
$ pip install mltool
At this point 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
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.