mirror of
https://github.com/deepset-ai/haystack.git
synced 2025-07-29 20:00:46 +00:00
15 lines
320 B
Python
15 lines
320 B
Python
|
import pytest
|
||
|
|
|
||
|
|
from haystack.nodes import FARMReader
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.fixture
|
||
|
|
def reader_without_normalized_scores():
|
||
|
|
return FARMReader(
|
||
|
|
model_name_or_path="deepset/bert-medium-squad2-distilled",
|
||
|
|
use_gpu=False,
|
||
|
|
top_k_per_sample=5,
|
||
|
|
num_processes=0,
|
||
|
|
use_confidence_scores=False,
|
||
|
|
)
|