This repository has been archived by the owner on Sep 27, 2022. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 2
/
Copy pathconfig.yml
92 lines (80 loc) · 1.79 KB
/
config.yml
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
estimators:
- name: daal4py
estimator: daal4py.sklearn.neighbors.NearestNeighbors
chunk: False
- name: "ArgKmin (parallel_on_Y)"
estimator: nn.ArgKminParallelY
chunk: True
- name: "ArgKmin (parallel_on_X)"
estimator: nn.ArgKminParallelX
chunk: True
- name: "ArgKmin (auto)"
estimator: nn.ArgKminParallelAuto
chunk: True
n_trials: 5
return_distance: True
datasets:
# Motives: a few query point < 256 (current chunk size)
- n_samples_train: 1e5
n_samples_test: 1
n_features: 50
- n_samples_train: 1e5
n_samples_test: 128
n_features: 50
- n_samples_train: 1e5
n_samples_test: 256
n_features: 50
- n_samples_train: 1e5
n_samples_test: 1
n_features: 100
- n_samples_train: 1e5
n_samples_test: 128
n_features: 100
- n_samples_train: 1e5
n_samples_test: 256
n_features: 100
- n_samples_train: 1e5
n_samples_test: 1
n_features: 500
- n_samples_train: 1e5
n_samples_test: 128
n_features: 500
- n_samples_train: 1e5
n_samples_test: 256
n_features: 500
# Motives: Intermediate datasets
# X_train taller
- n_samples_train: 1e5
n_samples_test: 1e3
n_features: 100
- n_samples_train: 1e5
n_samples_test: 1e3
n_features: 500
# X_test taller
- n_samples_train: 1e3
n_samples_test: 1e5
n_features: 100
- n_samples_train: 1e3
n_samples_test: 1e5
n_features: 500
# Motives: Large datasets
# X_train taller
- n_samples_train: 1e6
n_samples_test: 1e4
n_features: 100
- n_samples_train: 1e6
n_samples_test: 1e4
n_features: 500
# X_test taller
- n_samples_train: 1e4
n_samples_test: 1e6
n_features: 100
- n_samples_train: 1e4
n_samples_test: 1e6
n_features: 500
chunk_size:
- 256
n_neighbors:
- 1
- 10
- 100