Based on the example code on https://www.tidymodels.org/learn/models/parsnip-nnet/
library(tidymodels)
#> ─ Attaching packages ──────────────────────────────────────────── tidymodels 0.1.0 ─
#> ✓ broom 0.5.6 ✓ recipes 0.1.12
#> ✓ dials 0.0.6 ✓ rsample 0.0.6
#> ✓ dplyr 0.8.5 ✓ tibble 3.0.1
#> ✓ ggplot2 3.3.0.9000 ✓ tune 0.1.0
#> ✓ infer 0.5.1 ✓ workflows 0.1.1
#> ✓ parsnip 0.1.0 ✓ yardstick 0.0.6
#> ✓ purrr 0.3.4
#> ─ Conflicts ────────────────────────────────────────────── tidymodels_conflicts() ─
#> x purrr::discard() masks scales::discard()
#> x dplyr::filter() masks stats::filter()
#> x dplyr::lag() masks stats::lag()
#> x ggplot2::margin() masks dials::margin()
#> x recipes::step() masks stats::step()
data(bivariate)
biv_rec <-
recipe(Class ~ ., data = bivariate_train) %>%
step_BoxCox(all_predictors())%>%
step_normalize(all_predictors()) %>%
prep(training = bivariate_train, retain = TRUE)
val_normalized <- bake(biv_rec, new_data = bivariate_val, all_predictors())
set.seed(57974)
create_xgb_dmatrix <- function(d) {
o <- parsnip:::convert_form_to_xy_fit(Class ~ ., d, composition = "matrix")
xgboost::xgb.DMatrix(data = o$x, label = as.integer(o$y == "One"), missing = NA)
}
train_normalized <- bake(biv_rec, new_data = bivariate_train)
test_normalized <- bake(biv_rec, new_data = bivariate_test)
watchlist <- list(
train = create_xgb_dmatrix(train_normalized),
test = create_xgb_dmatrix(test_normalized)
)
fit <-
boost_tree(
tree_depth = 7,
trees = 10000,
learn_rate = 0.03
) %>%
set_mode("classification") %>%
set_engine("xgboost",
watchlist = watchlist,
early_stopping_rounds = 200,
verbose = 2
) %>%
fit(Class ~ ., data = juice(biv_rec))
#> [1] train-error:0.805748 test-error:0.763380
#> Multiple eval metrics are present. Will use test_error for early stopping.
#> Will train until test_error hasn't improved in 200 rounds.
#>
#> [2] train-error:0.805748 test-error:0.763380
#> [3] train-error:0.805748 test-error:0.763380
#> [4] train-error:0.805748 test-error:0.763380
#> [5] train-error:0.804757 test-error:0.766197
#> [6] train-error:0.812686 test-error:0.770423
#> [7] train-error:0.815659 test-error:0.773239
#> [8] train-error:0.813677 test-error:0.771831
#> [9] train-error:0.813677 test-error:0.770423
#> [10] train-error:0.813677 test-error:0.770423
#> [11] train-error:0.815659 test-error:0.771831
#> [12] train-error:0.814668 test-error:0.770423
#> [13] train-error:0.816650 test-error:0.773239
#> [14] train-error:0.815659 test-error:0.770423
#> [15] train-error:0.818632 test-error:0.770423
#> [16] train-error:0.818632 test-error:0.770423
#> [17] train-error:0.818632 test-error:0.770423
#> [18] train-error:0.818632 test-error:0.770423
#> [19] train-error:0.819623 test-error:0.769014
#> [20] train-error:0.819623 test-error:0.767606
#> [21] train-error:0.823588 test-error:0.769014
#> [22] train-error:0.822597 test-error:0.769014
#> [23] train-error:0.823588 test-error:0.767606
#> [24] train-error:0.821606 test-error:0.767606
#> [25] train-error:0.823588 test-error:0.769014
#> [26] train-error:0.823588 test-error:0.769014
#> [27] train-error:0.823588 test-error:0.767606
#> [28] train-error:0.826561 test-error:0.763380
#> [29] train-error:0.823588 test-error:0.766197
#> [30] train-error:0.826561 test-error:0.761972
#> [31] train-error:0.828543 test-error:0.761972
#> [32] train-error:0.829534 test-error:0.761972
#> [33] train-error:0.829534 test-error:0.764789
#> [34] train-error:0.829534 test-error:0.764789
#> [35] train-error:0.830525 test-error:0.766197
#> [36] train-error:0.831516 test-error:0.766197
#> [37] train-error:0.832507 test-error:0.766197
#> [38] train-error:0.833499 test-error:0.766197
#> [39] train-error:0.834490 test-error:0.767606
#> [40] train-error:0.833499 test-error:0.767606
#> [41] train-error:0.832507 test-error:0.767606
#> [42] train-error:0.833499 test-error:0.763380
#> [43] train-error:0.832507 test-error:0.763380
#> [44] train-error:0.831516 test-error:0.767606
#> [45] train-error:0.832507 test-error:0.767606
#> [46] train-error:0.835481 test-error:0.767606
#> [47] train-error:0.836472 test-error:0.766197
#> [48] train-error:0.835481 test-error:0.767606
#> [49] train-error:0.835481 test-error:0.766197
#> [50] train-error:0.835481 test-error:0.764789
#> [51] train-error:0.839445 test-error:0.771831
#> [52] train-error:0.843409 test-error:0.769014
#> [53] train-error:0.842418 test-error:0.770423
#> [54] train-error:0.846383 test-error:0.766197
#> [55] train-error:0.843409 test-error:0.766197
#> [56] train-error:0.843409 test-error:0.766197
#> [57] train-error:0.847374 test-error:0.767606
#> [58] train-error:0.846383 test-error:0.767606
#> [59] train-error:0.846383 test-error:0.764789
#> [60] train-error:0.846383 test-error:0.767606
#> [61] train-error:0.846383 test-error:0.767606
#> [62] train-error:0.846383 test-error:0.767606
#> [63] train-error:0.849356 test-error:0.767606
#> [64] train-error:0.852329 test-error:0.767606
#> [65] train-error:0.852329 test-error:0.767606
#> [66] train-error:0.852329 test-error:0.769014
#> [67] train-error:0.853320 test-error:0.769014
#> [68] train-error:0.851338 test-error:0.769014
#> [69] train-error:0.853320 test-error:0.769014
#> [70] train-error:0.851338 test-error:0.767606
#> [71] train-error:0.854311 test-error:0.764789
#> [72] train-error:0.857284 test-error:0.764789
#> [73] train-error:0.858276 test-error:0.763380
#> [74] train-error:0.858276 test-error:0.764789
#> [75] train-error:0.859267 test-error:0.764789
#> [76] train-error:0.858276 test-error:0.764789
#> [77] train-error:0.858276 test-error:0.766197
#> [78] train-error:0.858276 test-error:0.766197
#> [79] train-error:0.858276 test-error:0.769014
#> [80] train-error:0.858276 test-error:0.770423
#> [81] train-error:0.858276 test-error:0.769014
#> [82] train-error:0.859267 test-error:0.769014
#> [83] train-error:0.860258 test-error:0.767606
#> [84] train-error:0.860258 test-error:0.767606
#> [85] train-error:0.859267 test-error:0.767606
#> [86] train-error:0.859267 test-error:0.769014
#> [87] train-error:0.859267 test-error:0.769014
#> [88] train-error:0.859267 test-error:0.769014
#> [89] train-error:0.858276 test-error:0.769014
#> [90] train-error:0.858276 test-error:0.767606
#> [91] train-error:0.858276 test-error:0.767606
#> [92] train-error:0.859267 test-error:0.767606
#> [93] train-error:0.861249 test-error:0.766197
#> [94] train-error:0.860258 test-error:0.767606
#> [95] train-error:0.861249 test-error:0.767606
#> [96] train-error:0.860258 test-error:0.764789
#> [97] train-error:0.862240 test-error:0.764789
#> [98] train-error:0.862240 test-error:0.764789
#> [99] train-error:0.862240 test-error:0.764789
#> [100] train-error:0.862240 test-error:0.764789
#> [101] train-error:0.863231 test-error:0.766197
#> [102] train-error:0.865213 test-error:0.766197
#> [103] train-error:0.864222 test-error:0.764789
#> [104] train-error:0.865213 test-error:0.766197
#> [105] train-error:0.865213 test-error:0.766197
#> [106] train-error:0.865213 test-error:0.766197
#> [107] train-error:0.865213 test-error:0.766197
#> [108] train-error:0.865213 test-error:0.766197
#> [109] train-error:0.865213 test-error:0.766197
#> [110] train-error:0.865213 test-error:0.767606
#> [111] train-error:0.865213 test-error:0.767606
#> [112] train-error:0.866204 test-error:0.767606
#> [113] train-error:0.866204 test-error:0.767606
#> [114] train-error:0.866204 test-error:0.767606
#> [115] train-error:0.866204 test-error:0.767606
#> [116] train-error:0.867195 test-error:0.766197
#> [117] train-error:0.867195 test-error:0.766197
#> [118] train-error:0.868186 test-error:0.766197
#> [119] train-error:0.868186 test-error:0.766197
#> [120] train-error:0.868186 test-error:0.763380
#> [121] train-error:0.868186 test-error:0.764789
#> [122] train-error:0.868186 test-error:0.764789
#> [123] train-error:0.868186 test-error:0.764789
#> [124] train-error:0.868186 test-error:0.761972
#> [125] train-error:0.868186 test-error:0.761972
#> [126] train-error:0.868186 test-error:0.761972
#> [127] train-error:0.870169 test-error:0.761972
#> [128] train-error:0.869177 test-error:0.763380
#> [129] train-error:0.869177 test-error:0.763380
#> [130] train-error:0.869177 test-error:0.763380
#> [131] train-error:0.869177 test-error:0.761972
#> [132] train-error:0.869177 test-error:0.761972
#> [133] train-error:0.869177 test-error:0.761972
#> [134] train-error:0.870169 test-error:0.761972
#> [135] train-error:0.870169 test-error:0.763380
#> [136] train-error:0.870169 test-error:0.763380
#> [137] train-error:0.870169 test-error:0.763380
#> [138] train-error:0.870169 test-error:0.763380
#> [139] train-error:0.870169 test-error:0.763380
#> [140] train-error:0.871160 test-error:0.763380
#> [141] train-error:0.871160 test-error:0.763380
#> [142] train-error:0.870169 test-error:0.763380
#> [143] train-error:0.870169 test-error:0.763380
#> [144] train-error:0.870169 test-error:0.763380
#> [145] train-error:0.871160 test-error:0.763380
#> [146] train-error:0.871160 test-error:0.764789
#> [147] train-error:0.871160 test-error:0.764789
#> [148] train-error:0.871160 test-error:0.764789
#> [149] train-error:0.872151 test-error:0.764789
#> [150] train-error:0.872151 test-error:0.763380
#> [151] train-error:0.872151 test-error:0.763380
#> [152] train-error:0.873142 test-error:0.761972
#> [153] train-error:0.874133 test-error:0.759155
#> [154] train-error:0.876115 test-error:0.754930
#> [155] train-error:0.876115 test-error:0.753521
#> [156] train-error:0.878097 test-error:0.753521
#> [157] train-error:0.876115 test-error:0.756338
#> [158] train-error:0.877106 test-error:0.753521
#> [159] train-error:0.877106 test-error:0.753521
#> [160] train-error:0.878097 test-error:0.753521
#> [161] train-error:0.878097 test-error:0.754930
#> [162] train-error:0.878097 test-error:0.754930
#> [163] train-error:0.878097 test-error:0.753521
#> [164] train-error:0.878097 test-error:0.753521
#> [165] train-error:0.878097 test-error:0.753521
#> [166] train-error:0.878097 test-error:0.753521
#> [167] train-error:0.878097 test-error:0.753521
#> [168] train-error:0.879088 test-error:0.753521
#> [169] train-error:0.881070 test-error:0.752113
#> [170] train-error:0.882061 test-error:0.752113
#> [171] train-error:0.882061 test-error:0.750704
#> [172] train-error:0.882061 test-error:0.749296
#> [173] train-error:0.882061 test-error:0.749296
#> [174] train-error:0.882061 test-error:0.749296
#> [175] train-error:0.883053 test-error:0.747887
#> [176] train-error:0.885035 test-error:0.749296
#> [177] train-error:0.886026 test-error:0.749296
#> [178] train-error:0.887017 test-error:0.750704
#> [179] train-error:0.888008 test-error:0.747887
#> [180] train-error:0.889990 test-error:0.750704
#> [181] train-error:0.889990 test-error:0.749296
#> [182] train-error:0.888999 test-error:0.749296
#> [183] train-error:0.888999 test-error:0.749296
#> [184] train-error:0.888999 test-error:0.749296
#> [185] train-error:0.889990 test-error:0.749296
#> [186] train-error:0.889990 test-error:0.747887
#> [187] train-error:0.891972 test-error:0.746479
#> [188] train-error:0.891972 test-error:0.746479
#> [189] train-error:0.891972 test-error:0.746479
#> [190] train-error:0.891972 test-error:0.746479
#> [191] train-error:0.891972 test-error:0.746479
#> [192] train-error:0.891972 test-error:0.746479
#> [193] train-error:0.891972 test-error:0.746479
#> [194] train-error:0.891972 test-error:0.746479
#> [195] train-error:0.891972 test-error:0.746479
#> [196] train-error:0.891972 test-error:0.746479
#> [197] train-error:0.891972 test-error:0.746479
#> [198] train-error:0.891972 test-error:0.746479
#> [199] train-error:0.892963 test-error:0.746479
#> [200] train-error:0.892963 test-error:0.746479
#> [201] train-error:0.892963 test-error:0.746479
#> [202] train-error:0.891972 test-error:0.746479
#> [203] train-error:0.891972 test-error:0.746479
#> [204] train-error:0.891972 test-error:0.746479
#> [205] train-error:0.891972 test-error:0.747887
#> [206] train-error:0.891972 test-error:0.747887
#> [207] train-error:0.892963 test-error:0.747887
#> [208] train-error:0.893954 test-error:0.747887
#> [209] train-error:0.893954 test-error:0.747887
#> [210] train-error:0.893954 test-error:0.746479
#> [211] train-error:0.894946 test-error:0.746479
#> [212] train-error:0.895937 test-error:0.746479
#> [213] train-error:0.895937 test-error:0.746479
#> [214] train-error:0.895937 test-error:0.746479
#> [215] train-error:0.895937 test-error:0.746479
#> [216] train-error:0.895937 test-error:0.745070
#> [217] train-error:0.896928 test-error:0.745070
#> [218] train-error:0.897919 test-error:0.745070
#> [219] train-error:0.897919 test-error:0.745070
#> [220] train-error:0.897919 test-error:0.745070
#> [221] train-error:0.896928 test-error:0.746479
#> [222] train-error:0.896928 test-error:0.746479
#> [223] train-error:0.896928 test-error:0.746479
#> [224] train-error:0.897919 test-error:0.745070
#> [225] train-error:0.898910 test-error:0.745070
#> [226] train-error:0.897919 test-error:0.745070
#> [227] train-error:0.898910 test-error:0.746479
#> [228] train-error:0.897919 test-error:0.746479
#> [229] train-error:0.897919 test-error:0.746479
#> [230] train-error:0.898910 test-error:0.746479
#> [231] train-error:0.898910 test-error:0.746479
#> [232] train-error:0.898910 test-error:0.746479
#> [233] train-error:0.898910 test-error:0.746479
#> [234] train-error:0.897919 test-error:0.746479
#> [235] train-error:0.897919 test-error:0.746479
#> [236] train-error:0.898910 test-error:0.746479
#> [237] train-error:0.899901 test-error:0.746479
#> [238] train-error:0.897919 test-error:0.749296
#> [239] train-error:0.897919 test-error:0.749296
#> [240] train-error:0.897919 test-error:0.749296
#> [241] train-error:0.897919 test-error:0.749296
#> [242] train-error:0.897919 test-error:0.749296
#> [243] train-error:0.898910 test-error:0.749296
#> [244] train-error:0.897919 test-error:0.750704
#> [245] train-error:0.897919 test-error:0.749296
#> [246] train-error:0.897919 test-error:0.749296
#> [247] train-error:0.898910 test-error:0.749296
#> [248] train-error:0.898910 test-error:0.749296
#> [249] train-error:0.899901 test-error:0.749296
#> [250] train-error:0.902874 test-error:0.747887
#> [251] train-error:0.902874 test-error:0.747887
#> [252] train-error:0.901883 test-error:0.747887
#> [253] train-error:0.901883 test-error:0.747887
#> [254] train-error:0.901883 test-error:0.747887
#> [255] train-error:0.900892 test-error:0.747887
#> [256] train-error:0.900892 test-error:0.747887
#> [257] train-error:0.901883 test-error:0.749296
#> [258] train-error:0.901883 test-error:0.747887
#> [259] train-error:0.900892 test-error:0.747887
#> [260] train-error:0.900892 test-error:0.747887
#> [261] train-error:0.901883 test-error:0.747887
#> [262] train-error:0.901883 test-error:0.747887
#> [263] train-error:0.901883 test-error:0.749296
#> [264] train-error:0.901883 test-error:0.749296
#> [265] train-error:0.901883 test-error:0.749296
#> [266] train-error:0.902874 test-error:0.749296
#> [267] train-error:0.902874 test-error:0.749296
#> [268] train-error:0.904856 test-error:0.749296
#> [269] train-error:0.904856 test-error:0.749296
#> [270] train-error:0.905847 test-error:0.747887
#> [271] train-error:0.906838 test-error:0.747887
#> [272] train-error:0.906838 test-error:0.747887
#> [273] train-error:0.906838 test-error:0.747887
#> [274] train-error:0.906838 test-error:0.747887
#> [275] train-error:0.905847 test-error:0.747887
#> [276] train-error:0.905847 test-error:0.747887
#> [277] train-error:0.905847 test-error:0.747887
#> [278] train-error:0.906838 test-error:0.746479
#> [279] train-error:0.906838 test-error:0.746479
#> [280] train-error:0.907830 test-error:0.746479
#> [281] train-error:0.908821 test-error:0.747887
#> [282] train-error:0.908821 test-error:0.747887
#> [283] train-error:0.908821 test-error:0.747887
#> [284] train-error:0.908821 test-error:0.749296
#> [285] train-error:0.908821 test-error:0.749296
#> [286] train-error:0.908821 test-error:0.749296
#> [287] train-error:0.908821 test-error:0.749296
#> [288] train-error:0.908821 test-error:0.749296
#> [289] train-error:0.908821 test-error:0.747887
#> [290] train-error:0.908821 test-error:0.749296
#> [291] train-error:0.908821 test-error:0.749296
#> [292] train-error:0.908821 test-error:0.747887
#> [293] train-error:0.908821 test-error:0.747887
#> [294] train-error:0.908821 test-error:0.747887
#> [295] train-error:0.908821 test-error:0.747887
#> [296] train-error:0.907830 test-error:0.747887
#> [297] train-error:0.910803 test-error:0.747887
#> [298] train-error:0.910803 test-error:0.747887
#> [299] train-error:0.910803 test-error:0.747887
#> [300] train-error:0.910803 test-error:0.747887
#> [301] train-error:0.910803 test-error:0.747887
#> [302] train-error:0.910803 test-error:0.747887
#> [303] train-error:0.910803 test-error:0.747887
#> [304] train-error:0.910803 test-error:0.747887
#> [305] train-error:0.911794 test-error:0.747887
#> [306] train-error:0.911794 test-error:0.747887
#> [307] train-error:0.911794 test-error:0.747887
#> [308] train-error:0.911794 test-error:0.747887
#> [309] train-error:0.911794 test-error:0.747887
#> [310] train-error:0.911794 test-error:0.747887
#> [311] train-error:0.911794 test-error:0.747887
#> [312] train-error:0.911794 test-error:0.746479
#> [313] train-error:0.911794 test-error:0.746479
#> [314] train-error:0.911794 test-error:0.746479
#> [315] train-error:0.911794 test-error:0.746479
#> [316] train-error:0.911794 test-error:0.746479
#> [317] train-error:0.911794 test-error:0.746479
#> [318] train-error:0.911794 test-error:0.745070
#> [319] train-error:0.911794 test-error:0.745070
#> [320] train-error:0.911794 test-error:0.745070
#> [321] train-error:0.911794 test-error:0.746479
#> [322] train-error:0.911794 test-error:0.746479
#> [323] train-error:0.911794 test-error:0.746479
#> [324] train-error:0.911794 test-error:0.746479
#> [325] train-error:0.911794 test-error:0.745070
#> [326] train-error:0.912785 test-error:0.746479
#> [327] train-error:0.912785 test-error:0.745070
#> [328] train-error:0.912785 test-error:0.746479
#> [329] train-error:0.913776 test-error:0.745070
#> [330] train-error:0.913776 test-error:0.746479
#> [331] train-error:0.913776 test-error:0.746479
#> [332] train-error:0.913776 test-error:0.746479
#> [333] train-error:0.913776 test-error:0.746479
#> [334] train-error:0.913776 test-error:0.746479
#> [335] train-error:0.913776 test-error:0.746479
#> [336] train-error:0.913776 test-error:0.746479
#> [337] train-error:0.913776 test-error:0.746479
#> [338] train-error:0.913776 test-error:0.746479
#> [339] train-error:0.914767 test-error:0.746479
#> [340] train-error:0.914767 test-error:0.746479
#> [341] train-error:0.914767 test-error:0.746479
#> [342] train-error:0.916749 test-error:0.746479
#> [343] train-error:0.916749 test-error:0.746479
#> [344] train-error:0.916749 test-error:0.746479
#> [345] train-error:0.915758 test-error:0.746479
#> [346] train-error:0.915758 test-error:0.746479
#> [347] train-error:0.915758 test-error:0.746479
#> [348] train-error:0.915758 test-error:0.746479
#> [349] train-error:0.915758 test-error:0.746479
#> [350] train-error:0.915758 test-error:0.747887
#> [351] train-error:0.915758 test-error:0.747887
#> [352] train-error:0.916749 test-error:0.746479
#> [353] train-error:0.916749 test-error:0.747887
#> [354] train-error:0.916749 test-error:0.747887
#> [355] train-error:0.917740 test-error:0.747887
#> [356] train-error:0.918731 test-error:0.749296
#> [357] train-error:0.918731 test-error:0.749296
#> [358] train-error:0.918731 test-error:0.749296
#> [359] train-error:0.918731 test-error:0.747887
#> [360] train-error:0.918731 test-error:0.747887
#> [361] train-error:0.918731 test-error:0.747887
#> [362] train-error:0.918731 test-error:0.747887
#> [363] train-error:0.917740 test-error:0.749296
#> [364] train-error:0.917740 test-error:0.750704
#> [365] train-error:0.917740 test-error:0.750704
#> [366] train-error:0.917740 test-error:0.750704
#> [367] train-error:0.917740 test-error:0.750704
#> [368] train-error:0.917740 test-error:0.752113
#> [369] train-error:0.917740 test-error:0.752113
#> [370] train-error:0.917740 test-error:0.752113
#> [371] train-error:0.917740 test-error:0.752113
#> [372] train-error:0.918731 test-error:0.750704
#> [373] train-error:0.918731 test-error:0.750704
#> [374] train-error:0.918731 test-error:0.752113
#> [375] train-error:0.918731 test-error:0.752113
#> [376] train-error:0.918731 test-error:0.752113
#> [377] train-error:0.918731 test-error:0.749296
#> [378] train-error:0.918731 test-error:0.749296
#> [379] train-error:0.920714 test-error:0.749296
#> [380] train-error:0.920714 test-error:0.749296
#> [381] train-error:0.920714 test-error:0.749296
#> [382] train-error:0.920714 test-error:0.749296
#> [383] train-error:0.920714 test-error:0.749296
#> [384] train-error:0.920714 test-error:0.749296
#> [385] train-error:0.920714 test-error:0.749296
#> [386] train-error:0.920714 test-error:0.749296
#> [387] train-error:0.920714 test-error:0.749296
#> [388] train-error:0.920714 test-error:0.749296
#> [389] train-error:0.920714 test-error:0.749296
#> [390] train-error:0.920714 test-error:0.749296
#> [391] train-error:0.920714 test-error:0.749296
#> [392] train-error:0.920714 test-error:0.749296
#> [393] train-error:0.920714 test-error:0.749296
#> [394] train-error:0.920714 test-error:0.749296
#> [395] train-error:0.920714 test-error:0.749296
#> [396] train-error:0.920714 test-error:0.749296
#> [397] train-error:0.920714 test-error:0.747887
#> [398] train-error:0.920714 test-error:0.747887
#> [399] train-error:0.921705 test-error:0.747887
#> [400] train-error:0.922696 test-error:0.747887
#> [401] train-error:0.922696 test-error:0.749296
#> [402] train-error:0.922696 test-error:0.749296
#> [403] train-error:0.922696 test-error:0.749296
#> [404] train-error:0.922696 test-error:0.749296
#> [405] train-error:0.922696 test-error:0.749296
#> [406] train-error:0.922696 test-error:0.749296
#> [407] train-error:0.922696 test-error:0.749296
#> [408] train-error:0.922696 test-error:0.749296
#> [409] train-error:0.922696 test-error:0.749296
#> [410] train-error:0.922696 test-error:0.749296
#> [411] train-error:0.922696 test-error:0.749296
#> [412] train-error:0.922696 test-error:0.749296
#> [413] train-error:0.922696 test-error:0.749296
#> [414] train-error:0.922696 test-error:0.749296
#> [415] train-error:0.922696 test-error:0.749296
#> [416] train-error:0.922696 test-error:0.749296
#> Stopping. Best iteration:
#> [216] train-error:0.895937 test-error:0.745070
val_results <-
bivariate_val %>%
bind_cols(
predict(fit, new_data = val_normalized),
predict(fit, new_data = val_normalized, type = "prob")
)
val_results %>% roc_auc(truth = Class, .pred_One)
#> # A tibble: 1 x 3
#> .metric .estimator .estimate
#> <chr> <chr> <dbl>
#> 1 roc_auc binary 0.788
val_results %>% accuracy(truth = Class, .pred_class)
#> # A tibble: 1 x 3
#> .metric .estimator .estimate
#> <chr> <chr> <dbl>
#> 1 accuracy binary 0.713
Created on 2020-05-06 by the reprex package (v0.3.0)