Skip to content

Commit 6e1b920

Browse files
committed
feat: update assets
1 parent 99dd0f3 commit 6e1b920

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

tests/assets/configs/description.yaml

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -6,15 +6,16 @@
66
embedder_config:
77
- model_name: sentence-transformers/all-MiniLM-L6-v2
88
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]
139
- node_type: scoring
1410
target_metric: scoring_roc_auc
1511
search_space:
1612
- module_name: description
1713
temperature: [1.0, 0.5, 0.1, 0.05]
14+
embedder_config:
15+
- model_name: sentence-transformers/all-MiniLM-L6-v2
16+
cross_encoder_config:
17+
- model_name: cross-encoder/ms-marco-MiniLM-L-6-v2
18+
encoder_type: [cross, bi]
1819
- node_type: decision
1920
target_metric: decision_accuracy
2021
search_space:

0 commit comments

Comments
 (0)