Skip to content
neural-tree
load_beir
Initializing search
raphaelsty/neural-tree
Fine-tune
Existing tree
Evaluate
API reference
neural-tree
raphaelsty/neural-tree
Fine-tune
Fine-tune
Colbert
SentenceTransformer
TfIdf
Existing tree
Existing tree
Build an index from an existing tree
Evaluate
Evaluate
Evaluate
API reference
API reference
Overview
clustering
clustering
KMeans
average
get_mapping_nodes_documents
optimize_leafs
datasets
datasets
load_beir_test
load_beir_train
load_beir
load_beir
Table of contents
Parameters
leafs
leafs
Leaf
nodes
nodes
Node
retrievers
retrievers
ColBERT
SentenceTransformer
TfIdf
scoring
scoring
BaseScore
ColBERT
SentenceTransformer
TfIdf
trees
trees
ColBERT
SentenceTransformer
TfIdf
Tree
utils
utils
batchify
evaluate
iter
leafs_precision
sanity_check
set_env
Table of contents
Parameters
load_beir
¶
Load BEIR dataset.
Parameters
¶
dataset_name
(
str
)
split
(
str
)