Palashio / libra

Ergonomic machine learning for everyone.
http://libradocs.org/
MIT License
1.91k stars 109 forks source link

Added XGBoost classifier #314

Closed a10mic closed 4 years ago

a10mic commented 4 years ago

name: b) New feature labels: xgboost classifier

This pull request closes #272 .

- What I did I added xgboost_query for running xgboost classifier

- How I did it I added train_xgboost function in classification_models.py then made a query xgboost_query in query.py to train the classifier. Updated the analyze function and plots function to support xgboost_query and added tests in test.py

This pull request adds a new feature to libra. @Palashio, could you please take a look at it?