Skip to content

Commit 8267e80

Browse files
Update R/build_and_compile_model.R
Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>
1 parent f2e8ec9 commit 8267e80

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

R/build_and_compile_model.R

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -302,7 +302,8 @@ build_and_compile_functional_model <- function(
302302
# This is primarily for output layers that might need num_classes
303303
if ("num_classes" %in% block_fml_names) {
304304
# Check if this block is an output block and if it's a classification task
305-
if (is.list(y_processed$y_proc) && !is.null(names(y_processed$y_proc))) { # Multi-output case
305+
if (is.list(y_processed$y_proc) && !is.null(names(y_processed$y_proc))) {
306+
# Multi-output case
306307
# Find the corresponding output in y_processed based on block_name
307308
y_names <- names(y_processed$y_proc)
308309
# If there is only one output, and this block is named 'output',

0 commit comments

Comments
 (0)