Documentation
¶
Index ¶
Constants ¶
This section is empty.
Variables ¶
This section is empty.
Functions ¶
func SetSagemakerConfig ¶
Types ¶
type Config ¶
type Config struct {
RoleArn string `json:"roleArn" pflag:",The role the SageMaker plugin uses to communicate with the SageMaker service"`
Region string `json:"region" pflag:",The AWS region the SageMaker plugin communicates to"`
RoleAnnotationKey string `json:"roleAnnotationKey" pflag:",Map key to use to lookup role from task annotations."`
PrebuiltAlgorithms []PrebuiltAlgorithmConfig `json:"prebuiltAlgorithms" pflag:"-,A List of PrebuiltAlgorithm configs"`
}
Sagemaker plugin configs
func GetSagemakerConfig ¶
func GetSagemakerConfig() *Config
Retrieves the current config value or default.
type PrebuiltAlgorithmConfig ¶
type PrebuiltAlgorithmConfig struct {
Name string `json:"name" pflag:",The name of the ML algorithm. Should match Sagemaker"`
RegionalConfig []RegionalConfig `json:"regionalConfigs" pflag:"-,Per region specific configuration for this algorithm"`
}
type RegionalConfig ¶
type RegionalConfig struct {
Region string `json:"region" pflag:",Region for which this config is applicable"`
VersionConfigs []VersionConfig `json:"versionConfigs" pflag:",Configuration for various versions of the algorithms'"`
}
type VersionConfig ¶
Click to show internal directories.
Click to hide internal directories.