updating docs and pkgdown

Author: davidrsch

R/utils.R

@@ -77,6 +77,7 @@ get_keras_object <- function(
 #' Dials Parameter for Keras Optimizers
 #' @param values A character vector of possible optimizers. Defaults to all
 #'   known optimizers (keras defaults + custom registered).
+#' @keywords internal
 #' @export
 optimizer_function <- function(values = NULL) {
   if (is.null(values)) {
@@ -96,6 +97,7 @@ optimizer_function <- function(values = NULL) {
 #' Dials Parameter for Keras Loss Functions
 #' @param values A character vector of possible loss functions. Defaults to all
 #'   known losses (keras defaults + custom registered).
+#' @keywords internal
 #' @export
 loss_function_keras <- function(values = NULL) {
   if (is.null(values)) {

_pkgdown.yml

@@ -5,9 +5,20 @@ template:
 reference:
   - title: "Core Functions"
     desc: >
-      The main function for generating custom model specifications.
+      Functions for generating and managing custom model specifications.
     contents:
       - create_keras_spec
+      - remove_keras_spec
+
+  - title: "Registration Helpers"
+    desc: >
+      Functions for registering custom Keras components like loss functions,
+      optimizers, and metrics.
+    contents:
+      - register_keras_loss
+      - register_keras_metric
+      - register_keras_optimizer
+      - keras_objects
 
 articles:
   - title: "Getting Started"