29
29
# ' optimizer object.
30
30
# ' @seealso [register_keras_loss()], [register_keras_metric()]
31
31
# ' @export
32
+ # ' @return No return value, called for side effects.
32
33
# ' @examples
33
34
# ' if (requireNamespace("keras3", quietly = TRUE)) {
34
35
# ' # Register a custom version of Adam with a different default beta_1
@@ -58,6 +59,7 @@ register_keras_optimizer <- function(name, optimizer_fn) {
58
59
# ' @param loss_fn The loss function.
59
60
# ' @seealso [register_keras_optimizer()], [register_keras_metric()]
60
61
# ' @export
62
+ # ' @return No return value, called for side effects.
61
63
register_keras_loss <- function (name , loss_fn ) {
62
64
.kerasnip_custom_objects $ losses [[name ]] <- loss_fn
63
65
invisible ()
@@ -78,6 +80,7 @@ register_keras_loss <- function(name, loss_fn) {
78
80
# ' @param metric_fn The metric function.
79
81
# ' @seealso [register_keras_optimizer()], [register_keras_loss()]
80
82
# ' @export
83
+ # ' @return No return value, called for side effects.
81
84
register_keras_metric <- function (name , metric_fn ) {
82
85
.kerasnip_custom_objects $ metrics [[name ]] <- metric_fn
83
86
invisible ()
@@ -144,6 +147,7 @@ get_keras_object <- function(
144
147
# ' known optimizers (keras defaults + custom registered).
145
148
# ' @keywords internal
146
149
# ' @export
150
+ # ' @return A `dials` parameter object for Keras optimizers.
147
151
optimizer_function <- function (values = NULL ) {
148
152
if (is.null(values )) {
149
153
values <- unique(c(
@@ -164,6 +168,7 @@ optimizer_function <- function(values = NULL) {
164
168
# ' known losses (keras defaults + custom registered).
165
169
# ' @keywords internal
166
170
# ' @export
171
+ # ' @return A `dials` parameter object for Keras loss.
167
172
loss_function_keras <- function (values = NULL ) {
168
173
if (is.null(values )) {
169
174
values <- unique(c(keras_losses , names(.kerasnip_custom_objects $ losses )))
@@ -402,8 +407,8 @@ process_y_sequential <- function(
402
407
# '
403
408
# ' @return The `parsnip` model environment.
404
409
# ' @examples
405
- # ' \dontrun {
406
- # ' model_env <- kerasnip::: get_model_env()
410
+ # ' \donttest {
411
+ # ' model_env <- kerasnip::get_model_env()
407
412
# ' }
408
413
# ' @keywords internal
409
414
# ' @export
0 commit comments