R/model-class.R

In brain-r/brain.widget: 'Htmlwidgets' of 'Tensorflow'

#' Compile
#'
#' Compile model.
#' 
#' @inheritParams basic-layers
#' @param optimizer Optimizer function.
#' @param loss A loss function.
#'
#' @examples
#' \dontrun{
#' library(shiny)
#' 
#' ui <- fluidPage(brainyOutput("brainy"))
#' 
#' server <- function(input, output){
#' 
#'   output$brainy <- renderBrainy({
#'     brainy() %>%
#'     layer_add(
#'       layer_dense(units = 1)
#'     ) %>%
#'     model_compile(
#'       optimizer = "meanSquaredError",
#'       loss = "sgd"
#'     )
#'   })
#'   
#' }
#' 
#' shinyApp(ui, server)
#' }
#' 
#' @export
model_compile <- function(brain, optimizer = NULL, loss = NULL, metrics = "accuracy"){
    opts <- list(
        metrics = as.list(metrics)
    )

    if(!is.null(optimizer)) opts$optimizer <- optimizer
    if(!is.null(loss)) opts$loss <- loss

    brain$x$compile <- opts
    return(brain)
}