We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 99dd0f3 commit 6e1b920Copy full SHA for 6e1b920
tests/assets/configs/description.yaml
@@ -6,15 +6,16 @@
6
embedder_config:
7
- model_name: sentence-transformers/all-MiniLM-L6-v2
8
device: cpu
9
- cross_encoder_config:
10
- - model_name: cross-encoder/ms-marco-MiniLM-L-6-v2
11
- device: cpu
12
- encoder_type: [cross, bi]
13
- node_type: scoring
14
target_metric: scoring_roc_auc
15
search_space:
16
- module_name: description
17
temperature: [1.0, 0.5, 0.1, 0.05]
+ embedder_config:
+ - model_name: sentence-transformers/all-MiniLM-L6-v2
+ cross_encoder_config:
+ - model_name: cross-encoder/ms-marco-MiniLM-L-6-v2
18
+ encoder_type: [cross, bi]
19
- node_type: decision
20
target_metric: decision_accuracy
21
0 commit comments