-
Notifications
You must be signed in to change notification settings - Fork 0
/
ForecastPOC.yaml
95 lines (83 loc) · 2.8 KB
/
ForecastPOC.yaml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
---
AWSTemplateFormatVersion: '2010-09-09'
Description: IAM Policies, and SageMaker Notebook to work with Amazon Forecast, it will also clone the POC codebase into the Notebook before you get started.
Parameters:
NotebookName:
Type: String
Default: ForecastPOCNotebook
Description: Enter the name of the SageMaker Notebook instance. Default is ForecastPOCNotebook.
VolumeSize:
Type: Number
Default: 10
MinValue: 5
MaxValue: 16384
ConstraintDescription: Must be an integer between 5 (GB) and 16384 (16 TB).
Description: Enter the size of the EBS volume in GB.
UseVPC:
Type: String
AllowedValues:
- "true"
- "false"
SecurityGroup:
# Use String rather than AWS::EC2::SecurityGroup::Id to allow for empty values
Type: String
Description: The security group to provide the notebook access within the VPC
PrivateSubnet:
# Use String rather than AWS::EC2::Subnet::Id to allow for empty values
Type: String
Description: The ID of the subnet in a VPC to which you would like to have a connectivity from your ML compute instance
Metadata:
AWS::CloudFormation::Interface:
ParameterLabels:
UseVPC:
default: Use existing VPC?
SecurityGroup:
default: Security Group Id
PrivateSubnet:
default: Private Subnet Id
NotebookName:
default: Notebook Name
VolumeSize:
default: EBS (Hard disk) Volume Size
ParameterGroups:
- Label:
default: Datastorage
- Label:
default: VPC Settings (Required if UseVPC is true)
Parameters:
- UseVPC
- SecurityGroup
- PrivateSubnet
Conditions:
UseOwnVPC: !Equals [!Ref UseVPC, "true"]
Resources:
# SageMaker Execution Role
SageMakerIamRole:
Type: "AWS::IAM::Role"
Properties:
AssumeRolePolicyDocument:
Version: "2012-10-17"
Statement:
-
Effect: Allow
Principal:
Service: sagemaker.amazonaws.com
Action: sts:AssumeRole
Path: "/"
ManagedPolicyArns:
- "arn:aws:iam::aws:policy/AmazonSageMakerFullAccess"
- "arn:aws:iam::aws:policy/AmazonS3FullAccess"
- "arn:aws:iam::aws:policy/AmazonForecastFullAccess"
- "arn:aws:iam::aws:policy/IAMFullAccess"
# SageMaker notebook
NotebookInstance:
Type: "AWS::SageMaker::NotebookInstance"
Properties:
InstanceType: "ml.t2.medium"
NotebookInstanceName: !Ref NotebookName
RoleArn: !GetAtt SageMakerIamRole.Arn
VolumeSizeInGB: !Ref VolumeSize
DefaultCodeRepository: https://github.com/chrisking/ForecastPOC.git
SecurityGroupIds:
- !If [ UseOwnVPC, !Ref SecurityGroup, !Ref "AWS::NoValue"]
SubnetId: !If [ UseOwnVPC, !Ref PrivateSubnet, !Ref "AWS::NoValue"]