ml_switcheroo.plugins.keras_sequential

Plugin for Keras Sequential Container.

Handles: 1. Packing *args into [layers]. 2. Renaming API to keras.Sequential.

Functions

transform_keras_sequential(→ libcst.Call)

Plugin Hook: Packs positional layer arguments into a list for Keras Sequential.

Module Contents

ml_switcheroo.plugins.keras_sequential.transform_keras_sequential(node: libcst.Call, ctx: ml_switcheroo.core.hooks.HookContext) → libcst.Call

Plugin Hook: Packs positional layer arguments into a list for Keras Sequential. Also renames the function to keras.Sequential.