Created by: Darinochka
Тесты для InferencePipeline
Текущие проблемы:
- тест
tests/pipeline/test_inference.pyпроходит только если добавить в конфигиtests/assets/configs/для модуля скорингаmodel_name:
nodes:
- node_type: retrieval
metric: retrieval_hit_rate
search_space:
- module_type: vector_db
k: [10]
model_name:
- sentence-transformers/all-MiniLM-L6-v2
- node_type: scoring
metric: scoring_roc_auc
search_space:
- module_type: description
temperature: [1.0, 0.5, 0.1, 0.05]
model_name:
- sentence-transformers/all-MiniLM-L6-v2
- node_type: prediction
metric: prediction_accuracy
search_space:
- module_type: argmax
иначе выдает ошибку:
TypeError: KNNScorer.__init__() missing 1 required positional argument: 'model_name'
Это нормальное поведение? Стоит ли менять конфиг? Как я поняла, model_name это энкодер из векторной базы и он должен подставляться автоматически исходя из лучшего энкодера