search

yandex / rep

Machine Learning toolbox for Humans
http://yandex.github.io/rep/
Other
687 stars 144 forks source link

Update XGBoost wrapper #14

Open arogozhnikov opened 9 years ago

arogozhnikov commented 9 years ago

XGBoost wrapper was significantly improved.

New wrapper is almost sklearn-compatible, but there are still failures of copy and clone, important field classes_ is not set.

So there are two ways:

  1. We leave things as they are, since our wrapper works fine.
  2. or I fix these bugs in original wrapper and we delete special wrapper for XGBoost. Less code to support, but no guarantee about proper support of that wrapper.