-
Notifications
You must be signed in to change notification settings - Fork 6
/
component.yaml
133 lines (133 loc) · 2.88 KB
/
component.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
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
description: ""
implementation:
container:
command:
- python3
- main.py
- --n_blocks
- inputValue: n_blocks
- --max_text_len
- inputValue: max_text_len
- --vocab_size
- inputValue: vocab_size
- --task
- inputValue: task
- --model
- inputValue: model
- --embedding_dim
- inputValue: embedding_dim
- --hidden_dim_1
- inputValue: hidden_dim_1
- --hidden_dim_2
- inputValue: hidden_dim_2
- --hidden_dim
- inputValue: hidden_dim
- --dropout
- inputValue: dropout
- --device
- inputValue: device
- --dp
- inputValue: dp
- --user_level
- inputValue: user_level
- --n_workers
- inputValue: n_workers
- --epsilon_fraction
- inputValue: epsilon_fraction
- --delta_fraction
- inputValue: delta_fraction
- --dataset_monofile
- inputPath: dataset_monofile
- --block_counts
- inputPath: block_counts
- --metrics_path
- outputPath: metrics_path
- --model_path
- inputPath: model_path
image: privatekube/dp_evaluate:06-28-07-51-14
inputs:
- description: n_blocks
name: n_blocks
type: Int
- description: max_text_len
name: max_text_len
type: Int
- description: vocab_size
name: vocab_size
type: Int
- description: task
name: task
type: String
- description: model
name: model
type: String
- description: embedding_dim
name: embedding_dim
type: Int
- description: hidden_dim_1
name: hidden_dim_1
type: Int
- description: hidden_dim_2
name: hidden_dim_2
type: Int
- description: hidden_dim
name: hidden_dim
type: Int
- description: dropout
name: dropout
type: Float
- description: device
name: device
type: String
- description: learning_rate
name: learning_rate
type: Float
- description: dp
name: dp
type: Int
- description: user_level
name: user_level
type: Int
- description: epsilon
name: epsilon
type: Float
- description: delta
name: delta
type: Float
- description: n_epochs
name: n_epochs
type: Int
- description: batch_size
name: batch_size
type: Int
- description: noise
name: noise
type: Float
- description: timeframe_days
name: timeframe_days
type: Int
- description: learning_rate_scheduler
name: learning_rate_scheduler
type: Int
- description: max_grad_norm
name: max_grad_norm
type: Float
- description: n_workers
name: n_workers
type: Int
- name: epsilon_fraction
type: Float
- name: delta_fraction
type: Float
- name: block_counts
type: Data
- name: dataset_monofile
type: Data
- description: model_path
name: model_path
type: Data
name: DP evaluate
outputs:
- description: metrics_path
name: metrics_path
type: Data