-
Notifications
You must be signed in to change notification settings - Fork 2
/
Copy pathmodel_definition.yaml
111 lines (105 loc) · 2.06 KB
/
model_definition.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
input_features:
-
name: Gender
type: category
lowercase: true
-
name: Country
type: category
lowercase: true
-
name: self_employed
type: category
lowercase: true
-
name: family_history
type: category
lowercase: true
-
name: no_employees
type: category
lowercase: true
-
name: remote_work
type: category
lowercase: true
-
name: tech_company
type: category
lowercase: true
-
name: benefits
type: category
lowercase: true
-
name: care_options
type: category
lowercase: true
-
name: wellness_program
type: category
lowercase: true
-
name: seek_help
type: category
lowercase: true
-
name: anonymity
type: category
lowercase: true
-
name: leave
type: category
lowercase: true
-
name: mental_health_consequence
type: category
lowercase: true
-
name: phys_health_consequence
type: category
lowercase: true
-
name: coworkers
type: category
lowercase: true
-
name: supervisor
type: category
lowercase: true
-
name: mental_health_interview
type: category
lowercase: true
-
name: phys_health_interview
type: category
lowercase: true
-
name: mental_vs_physical
type: category
lowercase: true
-
name: obs_consequence
type: category
lowercase: true
-
name: age_range
type: category
combiner:
type: concat
num_fc_layers: 1
fc_size: 64
output_features:
-
name: treatment
type: category
lowercase: true
training:
batch_size: 10
epochs: 7000
dropout_rate: 0.0
early_stop: 50
preprocessing:
split_probabilities: [0.75, 0.2, 0.05]
stratify: treatment