Diff of /partyMod/tests/mob.R [000000] .. [fbf06f]

Switch to side-by-side view

--- a
+++ b/partyMod/tests/mob.R
@@ -0,0 +1,26 @@
+library("party")
+
+data("BostonHousing", package = "mlbench")
+BostonHousing$lstat <- log(BostonHousing$lstat)
+BostonHousing$rm <- BostonHousing$rm^2
+BostonHousing$chas <- factor(BostonHousing$chas, levels = 0:1, labels = c("no", "yes"))
+BostonHousing$rad <- factor(BostonHousing$rad, ordered = TRUE)
+fmBH <- mob(medv ~ lstat + rm | zn + indus + chas + nox + age + dis + rad + tax + crim + b + ptratio,
+  control = mob_control(minsplit = 40, verbose = TRUE),
+  data = BostonHousing, model = linearModel)
+fmBH
+summary(fmBH)
+
+### check for one-node tree
+fmBH <- try(mob(medv ~ lstat + rm | zn, control = mob_control(minsplit = 4000, verbose = TRUE),
+  data = BostonHousing, model = linearModel))
+stopifnot(class(fmBH) != "try-error")
+
+
+data("PimaIndiansDiabetes", package = "mlbench")
+fmPID <- mob(diabetes ~ glucose | pregnant + pressure + triceps + insulin + mass + pedigree + age,
+  control = mob_control(verbose = TRUE),
+  data = PimaIndiansDiabetes, model = glinearModel, family = binomial())
+fmPID
+summary(fmPID)
+