-
Notifications
You must be signed in to change notification settings - Fork 2
/
params.yaml
74 lines (69 loc) · 1.07 KB
/
params.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
seeds:
- 0
- 1
- 2
- 3
- 4
- 5
- 6
- 7
- 8
- 9
dataset_kwargs:
small:
n_samples: 6000
n_features: 3
n_classes: 4
n_labels: 2
length: 500
allow_unlabeled: true
sparse: false
test_size: 1000
avg_trace: 0.925 # Should be overwritten/ignored by sampling from a gamma distribution (see below)
gamma: # This will be popped out of the params dict
shape: 2.0
scale: 0.01
large:
n_samples: 37500
n_features: 20
n_classes: 50
n_labels: 5
length: 500
allow_unlabeled: true
sparse: false
test_size: 7500
avg_trace: 0.995 # Should be overwritten/ignored by sampling from a gamma distribution (see below)
gamma: # This will be popped out of the params dict
shape: 2.0
scale: 0.01
train:
cv_n_folds: 5
seed: 0
clf_dict: models/clf_dict.pkl
cross_validator: models/cross_validator.pkl
eval:
ks:
- 1
- 2
- 3
- 4
- 5
- 10
- 20
- 50
- 100
temperatures:
- 0.01
- 0.1
- 1
- 10
- 100
alphas:
- 0.2
- 0.4
- 0.5
- 0.6
- 0.8
- 0.85
- 0.9
- 0.95