Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add poisson demo

Browse Source
This commit is contained in:
hetong007 2015-05-11 15:21:54 -07:00
parent 9c0ba67088
commit 60d307c445
3 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
R-package/demo/00Index
View File

@ -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

7
R-package/demo/poisson_regression.R Normal file
View File

@ -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))

2
R-package/demo/runall.R
View File

@ -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)