sandy4321/sklearn_xgb

an improved scikit-learn interface for xgboost

An improved scikit-learn-like interface of XGBoost

About

scikit-learn is machine learning library for Python.

XGBoost is a new and useful gradient boosting library, which provides a customized Python interface as well as a simplified scikit-learn-like interface.

This repo contains a slightly improved and customized scikit-learn-like interface of XGBoost, heavily based on the official codes, with some small modifications.

Installation

Install scikit-learn and xgboost, download this repo and place it into your projects.

License

The code in this repo follows Apache License version 2.0.

scikit-learn follows New BSD License.

XGBoost follows Apache License version 2.0.

Reference

[1] Scikit-learn: Machine Learning in Python, Pedregosa et al., JMLR 12, pp. 2825-2830, 2011.

http://scikit-learn.org/stable/

[2] XGBoost: eXtreme Gradient Boosting

https://github.com/dmlc/xgboost