File tree Expand file tree Collapse file tree 2 files changed +12
-0
lines changed
Expand file tree Collapse file tree 2 files changed +12
-0
lines changed Original file line number Diff line number Diff 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 :
Original file line number Diff line number Diff 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 :
You can’t perform that action at this time.
0 commit comments