Skip to content

Commit dd562ce

Browse files
committed
Added RerankScorer to search space tests
1 parent 25543d0 commit dd562ce

File tree

2 files changed

+12
-0
lines changed

2 files changed

+12
-0
lines changed

tests/assets/configs/multiclass.yaml

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,12 @@ nodes:
2020
- avsolatorio/GIST-small-Embedding-v0
2121
k: [1, 3]
2222
train_head: [false, true]
23+
- module_type: rerank_scorer
24+
k: [ 5, 10 ]
25+
weights: [uniform, distance, closest]
26+
m: [ 2, 3 ]
27+
cross_encoder_name:
28+
- cross-encoder/ms-marco-MiniLM-L-6-v2
2329
- node_type: prediction
2430
metric: prediction_accuracy
2531
search_space:

tests/assets/configs/multilabel.yaml

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,12 @@ nodes:
1616
- module_type: linear
1717
- module_type: mlknn
1818
k: [5]
19+
- module_type: rerank_scorer
20+
k: [ 5, 10 ]
21+
weights: [ uniform, distance, closest ]
22+
m: [ 2, 3 ]
23+
cross_encoder_name:
24+
- cross-encoder/ms-marco-MiniLM-L-6-v2
1925
- node_type: prediction
2026
metric: prediction_accuracy
2127
search_space:

0 commit comments

Comments
 (0)