Skip to content

Commit b8c9b63

Browse files
author
Saini
committed
README added for XGBoost
1 parent 741c118 commit b8c9b63

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

im-xgboost/README.md

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,5 @@
1+
### Single machine Iron Man XGBoost multiclass classification
2+
This notebook `single-xgboost-multiclassification.ipynb` demos end2end training and scoring multi-classification model with single machine XGBoost. The dataset used is MNIST digit recognition dataset in LIBSVM format, which has a training set of 60,000 examples, and a test set of 10,000 examples. The parameters are the same as the open source XGBoost. This notebook will walk the users through data preparation, IM training, IM hosting, and prediction.
3+
4+
### Distributed Iron Man XGBoost multi-classification
5+
This notebook `distributed-xgboost-multiclassification.ipynb` demos end2end training and scoring multi-classification model with distributed XGBoost backed by YARN. The dataset used is MNIST digit recognition dataset in LIBSVM format. which has a training set of 60,000 examples, and a test set of 10,000 examples. The parameters are the same as the open source XGBoost. This notebook will walk the user through data preparation, IM training, IM hosting, and prediction.

0 commit comments

Comments
 (0)