/
makeLearnerParamFuns.R
83 lines (73 loc) · 3.88 KB
/
makeLearnerParamFuns.R
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
#' @rdname LearnerParam
#' @export
makeNumericLearnerParam = function(id, lower = -Inf, upper = Inf, allow.inf = FALSE, default,
when = "train", requires = NULL, tunable = TRUE, special.vals = list()) {
makeParam(id = id, type = "numeric", learner.param = TRUE, lower = lower,
upper = upper, allow.inf = allow.inf,
default = default, requires = requires, tunable = tunable, special.vals = special.vals, when = when)
}
#' @rdname LearnerParam
#' @export
makeNumericVectorLearnerParam = function(id, len = as.integer(NA), lower = -Inf,
upper = Inf, allow.inf = FALSE, default, when = "train", requires = NULL, tunable = TRUE, special.vals = list()) {
makeParam(id = id, type = "numericvector", learner.param = TRUE, len = len, lower = lower,
upper = upper, allow.inf = allow.inf,
default = default, requires = requires, tunable = tunable, special.vals = special.vals, when = when)
}
#' @rdname LearnerParam
#' @export
makeIntegerLearnerParam = function(id, lower = -Inf, upper = Inf,
default, when = "train", requires = NULL, tunable = TRUE, special.vals = list()) {
makeParam(id = id, type = "integer", learner.param = TRUE, lower = lower,
upper = upper,
default = default, requires = requires, tunable = tunable, special.vals = special.vals, when = when)
}
#' @rdname LearnerParam
#' @export
makeIntegerVectorLearnerParam = function(id, len = as.integer(NA), lower = -Inf,
upper = Inf, default, when = "train", requires = NULL, tunable = TRUE, special.vals = list()) {
makeParam(id = id, type = "integervector", learner.param = TRUE, len = len, lower = lower,
upper = upper,
default = default, requires = requires, tunable = tunable, special.vals = special.vals, when = when)
}
#' @rdname LearnerParam
#' @export
makeDiscreteLearnerParam = function(id, values, default,
when = "train", requires = NULL, tunable = TRUE, special.vals = list()) {
makeParam(id = id, type = "discrete", learner.param = TRUE, values = values,
default = default, requires = requires, tunable = tunable, special.vals = special.vals, when = when)
}
#' @rdname LearnerParam
#' @export
makeDiscreteVectorLearnerParam = function(id, len = as.integer(NA), values, default,
when = "train", requires = NULL, tunable = TRUE, special.vals = list()) {
makeParam(id = id, type = "discretevector", learner.param = TRUE, len = len, values = values,
default = default, requires = requires, tunable = tunable, special.vals = special.vals, when = when)
}
#' @rdname LearnerParam
#' @export
makeLogicalLearnerParam = function(id, default, when = "train", requires = NULL, tunable = TRUE, special.vals = list()) {
values = list("TRUE" = TRUE, "FALSE" = FALSE)
makeParam(id = id, type = "logical", learner.param = TRUE, values = values,
default = default, requires = requires, tunable = tunable, special.vals = special.vals, when = when)
}
#' @rdname LearnerParam
#' @export
makeLogicalVectorLearnerParam = function(id, len = as.integer(NA), default, when = "train",
requires = NULL, tunable = TRUE, special.vals = list()) {
values = list("TRUE" = TRUE, "FALSE" = FALSE)
makeParam(id = id, type = "logicalvector", learner.param = TRUE, len = len, values = values,
default = default, requires = requires, tunable = tunable, special.vals = special.vals, when = when)
}
#' @rdname LearnerParam
#' @export
makeUntypedLearnerParam = function(id, default, when = "train", requires = NULL, tunable = TRUE, special.vals = list()) {
makeParam(id = id, type = "untyped", learner.param = TRUE,
default = default, requires = requires, tunable = tunable, special.vals = special.vals, when = when)
}
#' @rdname LearnerParam
#' @export
makeFunctionLearnerParam = function(id, default, when = "train", requires = NULL, tunable = TRUE, special.vals = list()) {
makeParam(id = id, type = "function", learner.param = TRUE,
default = default, requires = requires, tunable = tunable, special.vals = special.vals, when = when)
}