BEIR dataset configuration.
| 24 | |
| 25 | @dataclass |
| 26 | class BEIRConfig: |
| 27 | """BEIR dataset configuration.""" |
| 28 | dataset_name: str = os.getenv("DATASET_NAME", "scifact") |
| 29 | embedding_model: str = os.getenv("EMBEDDING_MODEL", "all-MiniLM-L6-v2") |
| 30 | batch_size: int = int(os.getenv("BATCH_SIZE", "32")) |
| 31 | insert_batch_size: int = int(os.getenv("INSERT_BATCH_SIZE", "1000")) |
| 32 | data_dir: str = "./data" |
| 33 | |
| 34 | |
| 35 | @dataclass |