def create_smdebug_hook()

in 4-Deployment/RealTime/config/xgboost_customer_churn.py [0:0]

• 10 lines of code
• 1 McCabe index (conditional complexity)

def create_smdebug_hook(out_dir, train_data=None, validation_data=None, frequency=1, collections=None,):

    save_config = SaveConfig(save_interval=frequency)
    hook = Hook(
        out_dir=out_dir,
        train_data=train_data,
        validation_data=validation_data,
        save_config=save_config,
        include_collections=collections,
    )

    return hook