Instructions to use ai4data/devdata-search-gist-small-cmnrl with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- sentence-transformers
How to use ai4data/devdata-search-gist-small-cmnrl with sentence-transformers:
from sentence_transformers import SentenceTransformer model = SentenceTransformer("ai4data/devdata-search-gist-small-cmnrl") sentences = [ "The weather is lovely today.", "It's so sunny outside!", "He drove to the stadium." ] embeddings = model.encode(sentences) similarities = model.similarity(embeddings, embeddings) print(similarities.shape) # [3, 3] - Notebooks
- Google Colab
- Kaggle
devdata-search-gist-small-cmnrl
A bi-encoder embedding model for search over structured statistical
metadata, part of the DevData Search family. It is a fine-tune of
avsolatorio/GIST-small-Embedding-v0 produced with schema-invariant fine-tuning on
DevDataBench: full-schema
serialization with per-example field-order permutation and field dropout, so the
encoder binds meaning to field labels rather than to serialization order. This is
an embedding model that powers retrieval; it is not a hosted search service.
See the paper Field Order Should Not Matter: Permutation-Invariant Fine-Tuning for Structured Metadata Retrieval.
Training
- Base model:
avsolatorio/GIST-small-Embedding-v0 - Loss:
cmnrl - Field permutation:
True; field dropout:0.15 - Max sequence length:
512 - No query/document prefixes
Usage
from sentence_transformers import SentenceTransformer
model = SentenceTransformer("ai4data/devdata-search-gist-small-cmnrl")
queries = ["mobile-broadband subscriptions per 100 people, reported annually"]
docs = ["name: Active mobile-broadband subscriptions | ..."]
q = model.encode(queries)
d = model.encode(docs)
Cosine similarity of q and d ranks documents for each query.
License
Apache-2.0. Derived from avsolatorio/GIST-small-Embedding-v0; trained on public World Bank Data360 metadata.
- Downloads last month
- 19
Model tree for ai4data/devdata-search-gist-small-cmnrl
Base model
avsolatorio/GIST-small-Embedding-v0