-
Notifications
You must be signed in to change notification settings - Fork 1
/
config.yaml
58 lines (57 loc) · 1.26 KB
/
config.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
git_options:
git_uri: "git@github.com:u-adrian/Tutorial.git"
branch: "main"
model:
entry_point: "main.py"
function_name: "run_hyperopt"
# data_options:
# dataset_path: "/home/kit/iti/ew0464/gan_experiment/dataset/cifar-10-python.tar.gz" #dummy dataset. # optional
# copy_data: false # optional; default?
experiment:
# use_local_workspace: false # optional
cluster: "bwunicluster"
number_chain_jobs: 4
parallel_bandwidth: 4
observation_budget: 60
#conda_env: name_of_env # added
parameters:
- name: max_depth
type: int
bounds:
min: 1
max: 10
- name: n_estimators
type: int
bounds:
min: 1
max: 10
- name: bootstrap
type: int
grid:
- 0
- 1
- name: max_features
type: categorical
categorical_values:
- 'sqrt'
- 'log2'
- name: criterion
type: categorical
categorical_values:
- 'gini'
- 'entropy'
metrics:
- name: accuracy
objective: maximize
strategy: optimize
sbatch_options:
partition: "dev_gpu_4"
#gres: "gpu:4" # optional
#ntasks: 4 # optional
#mem-per-gpu: 32000 # optional
#time: "10" # optional
sigopt_options:
dev_run: true
project_id: "test_project"
experiment_name: "tutorial_project" # added/moved here
client_id: 11949