diff --git a/R-package/demo/00Index b/R-package/demo/00Index index f0b41ec2a..43df3ae61 100644 --- a/R-package/demo/00Index +++ b/R-package/demo/00Index @@ -7,3 +7,4 @@ cross_validation Cross validation create_sparse_matrix Create Sparse Matrix predict_leaf_indices Predicting the corresponding leaves early_Stopping Early Stop in training +poisson_regression Poisson Regression on count data diff --git a/R-package/demo/poisson_regression.R b/R-package/demo/poisson_regression.R new file mode 100644 index 000000000..f9dc4ac62 --- /dev/null +++ b/R-package/demo/poisson_regression.R @@ -0,0 +1,7 @@ +data(mtcars) +head(mtcars) +bst = xgboost(data=as.matrix(mtcars[,-11]),label=mtcars[,11], + objective='count:poisson',nrounds=5) +pred = predict(bst,as.matrix(mtcars[,-11])) +sqrt(mean((pred-mtcars[,11])^2)) + diff --git a/R-package/demo/runall.R b/R-package/demo/runall.R index 2d0384156..2ea4c446e 100644 --- a/R-package/demo/runall.R +++ b/R-package/demo/runall.R @@ -7,3 +7,5 @@ demo(generalized_linear_model) demo(cross_validation) demo(create_sparse_matrix) demo(predict_leaf_indices) +demo(early_Stopping) +demo(poisson_regression)