Skip to contents

Extract KAF branch scales

Source: R/diagnostics.R
extract_kaf_scales.Rd

Extracts the learned GELU/base and Fourier branch scales from each KAF layer.

Usage

extract_kaf_scales(object)

Arguments

object

A fitted object returned by kaf_fit(), or a raw nn_kaf torch module.

Value

A data frame with one row per layer-feature pair.