Skip to content

Conversation

brightsparc
Copy link
Contributor

@brightsparc brightsparc commented Feb 6, 2020

Issue #28

Description of changes:

Added support for reading debugger hook and rules from the estimator and writing to the API eg:

from sagemaker.debugger import Rule, rule_configs, DebuggerHookConfig, CollectionConfig

hook_config = DebuggerHookConfig(
    s3_output_path='s3://{0}/{1}/output/debug'.format(bucket, prefix),
    hook_parameters={
        "save_interval": "1"
    },
    collection_configs=[
        CollectionConfig("hyperparameters"),
        CollectionConfig("metrics"),
        CollectionConfig("predictions"),
        CollectionConfig("labels"),
        CollectionConfig("feature_importance")
    ]
)

xgb = sagemaker.estimator.Estimator(
    # Initialize your hook.
    debugger_hook_config=hook_config,
   
    # Initialize your rules for the hook
    rules=[Rule.sagemaker(rule_configs.confusion(),
                             rule_parameters={
                                 "category_no": "15",
                                 "min_diag": "0.7",
                                 "max_off_diag": "0.3",
                                 "start_step": "17",
                                 "end_step": "19"}
                         )]
)

By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.

@StepFunctions-Bot
Copy link
Contributor

AWS CodeBuild CI Report

  • Result: FAILED
  • Build Logs (available for 30 days)

Powered by github-codebuild-logs, available on the AWS Serverless Application Repository

@StepFunctions-Bot
Copy link
Contributor

AWS CodeBuild CI Report

  • CodeBuild project: StepFunctionsPythonSDK-integtests
  • Commit ID: 53e38d9
  • Result: SUCCEEDED
  • Build Logs (available for 30 days)

Powered by github-codebuild-logs, available on the AWS Serverless Application Repository

@StepFunctions-Bot
Copy link
Contributor

AWS CodeBuild CI Report

  • Result: SUCCEEDED
  • Build Logs (available for 30 days)

Powered by github-codebuild-logs, available on the AWS Serverless Application Repository

@StepFunctions-Bot
Copy link
Contributor

AWS CodeBuild CI Report

  • CodeBuild project: StepFunctionsPythonSDK-integtests
  • Commit ID: 109b14b
  • Result: SUCCEEDED
  • Build Logs (available for 30 days)

Powered by github-codebuild-logs, available on the AWS Serverless Application Repository

@shunjd shunjd merged commit 9225a36 into aws:master Feb 10, 2020
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants