X-Git-Url: https://git.auder.net/rpsls.js?a=blobdiff_plain;ds=sidebyside;f=pkg%2FR%2Fm_RegressionTree.R;fp=pkg%2FR%2Fm_RegressionTree.R;h=0000000000000000000000000000000000000000;hb=57dbbab63a48b42f85e0a711640e1dd10b58f18a;hp=d51e4084b8f06d0d6bbd964de9f7f9736cdc5163;hpb=357b8be00388d07c04c10d4bf7f503fd947185d2;p=aggexp.git diff --git a/pkg/R/m_RegressionTree.R b/pkg/R/m_RegressionTree.R deleted file mode 100644 index d51e408..0000000 --- a/pkg/R/m_RegressionTree.R +++ /dev/null @@ -1,36 +0,0 @@ -#' @include b_Algorithm.R - -#' @title Regression Tree -#' -#' @description Regression Tree using the \code{tree} package. -#' Inherits \code{\link{Algorithm}} -#' -#' @field nleaf Number of leaf nodes after pruning. Default: Inf (no pruning) -#' -RegressionTree = setRefClass( - Class = "RegressionTree", - - fields = c( - nleaf = "numeric" - ), - - contains = "Algorithm", - - methods = list( - initialize = function(...) - { - callSuper(...) - if (length(nleaf) == 0 || nleaf < 1) - nleaf <<- Inf - }, - predict_noNA = function(XY, x) - { - require(tree, quietly=TRUE) - rt = tree(Measure ~ ., data=XY) - treeSize = sum( rt$frame[["var"]] == "" ) - if (treeSize > nleaf) - rt = prune.tree(rt, best = nleaf) - return (stats::predict(rt, as.data.frame(x))) - } - ) -)