Skip to content

Commit

Permalink
Add hyperparameter tuning information
Browse files Browse the repository at this point in the history
  • Loading branch information
dobraczka committed Sep 12, 2024
1 parent 2b1cf8c commit 6d66b52
Show file tree
Hide file tree
Showing 2 changed files with 106 additions and 0 deletions.
46 changes: 46 additions & 0 deletions run_scripts/hyperparam_sweeps/relational_deepblocker_ae.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,46 @@
command:
- ${env}
- /home/h7/daob673c/.local/bin/micromamba
- run
- -n
- klinker-conda
- -r
- "y"
- python
- ${program}
- --random-seed
- "42"
- --wandb
- open-ea-dataset
- relational-deepblocker
- --encoder
- autoencoder
- --inner-encoder
- sentencetransformertokenized
- --inner-encoder-batch-size
- "512"
- --batch-size
- "512"
- --force
- "True"
- --n-neighbors
- "250"
- --rel-n-neighbors
- "250"
- ${args}
method: bayes
metric:
goal: maximize
name: h3r
parameters:
hidden-dimension:
values:
- 48
- 96
- 192
- 384
learning-rate:
distribution: uniform
max: 0.01
min: 0.0001
program: experiment.py
60 changes: 60 additions & 0 deletions run_scripts/hyperparam_sweeps/relational_deepblocker_ctt.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,60 @@
command:
- ${env}
- /home/h7/daob673c/.local/bin/micromamba
- run
- -n
- klinker-conda
- -r
- "y"
- python
- ${program}
- --random-seed
- "42"
- --wandb
- open-ea-dataset
- relational-deepblocker
- --encoder
- crosstupletraining
- --inner-encoder
- sentencetransformertokenized
- --inner-encoder-batch-size
- "512"
- --batch-size
- "512"
- --force
- "True"
- --n-neighbors
- "250"
- --rel-n-neighbors
- "250"
- ${args}
method: bayes
metric:
goal: maximize
name: h3r
parameters:
hidden-dimension:
values:
- 48
- 96
- 192
- 384
learning-rate:
distribution: uniform
max: 0.01
min: 0.0001
max-perturbation:
distribution: uniform
max: 0.6
min: 0.1
pos-to-neg-ratio:
distribution: uniform
max: 2
min: 0.1
synth-tuples-per-tuple:
values:
- 2
- 3
- 4
- 5
program: experiment.py

0 comments on commit 6d66b52

Please sign in to comment.