bb25 0.1.1__cp312-cp312-macosx_11_0_arm64.whl

bb25/__init__.py

@@ -0,0 +1,5 @@
+from .bb25 import *
+
+__doc__ = bb25.__doc__
+if hasattr(bb25, "__all__"):
+    __all__ = bb25.__all__

bb25-0.1.1.dist-info/METADATA

@@ -0,0 +1,116 @@
+Metadata-Version: 2.4
+Name: bb25
+Version: 0.1.1
+Classifier: License :: Other/Proprietary License
+Classifier: Programming Language :: Rust
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3 :: Only
+Classifier: Programming Language :: Python :: 3.8
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Operating System :: OS Independent
+Summary: Bayesian BM25 scoring and experimental validation (Rust core + Python bindings)
+Keywords: bm25,information-retrieval,search,ranking,bayesian,hybrid-search
+License: UNLICENSED
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown; charset=UTF-8; variant=GFM
+Project-URL: Homepage, https://github.com/sigridjineth/bayesian_bm25_rs
+Project-URL: Issues, https://github.com/sigridjineth/bayesian_bm25_rs/issues
+Project-URL: Repository, https://github.com/sigridjineth/bayesian_bm25_rs
+
+# bb25 (Bayesian BM25)
+
+bb25 is a fast, self-contained BM25 + Bayesian calibration implementation with a minimal Python API. It also includes a small reference corpus and experiment suite so you can validate the expected numerical properties.
+
+- PyPI package name: `bb25`
+- Python import name: `bb25`
+
+## Install
+
+```
+pip install bb25
+```
+
+## Quick start
+
+### Use the built-in corpus and queries
+
+```
+import bb25 as bb
+
+corpus = bb.build_default_corpus()
+docs = corpus.documents()
+queries = bb.build_default_queries()
+
+bm25 = bb.BM25Scorer(corpus, 1.2, 0.75)
+score = bm25.score(queries[0].terms, docs[0])
+print("score0", score)
+```
+
+### Build your own corpus
+
+```
+import bb25 as bb
+
+corpus = bb.Corpus()
+corpus.add_document("d1", "neural networks for ranking", [0.1] * 8)
+corpus.add_document("d2", "bm25 is a strong baseline", [0.2] * 8)
+corpus.build_index()  # must be called before creating scorers
+
+bm25 = bb.BM25Scorer(corpus, 1.2, 0.75)
+print(bm25.idf("bm25"))
+```
+
+### Bayesian calibration + hybrid fusion
+
+```
+import bb25 as bb
+
+corpus = bb.build_default_corpus()
+docs = corpus.documents()
+queries = bb.build_default_queries()
+
+bm25 = bb.BM25Scorer(corpus, 1.2, 0.75)
+bayes = bb.BayesianBM25Scorer(bm25, 1.0, 0.5)
+vector = bb.VectorScorer()
+hybrid = bb.HybridScorer(bayes, vector)
+
+q = queries[0]
+prob_or = hybrid.score_or(q.terms, q.embedding, docs[0])
+prob_and = hybrid.score_and(q.terms, q.embedding, docs[0])
+print("OR", prob_or, "AND", prob_and)
+```
+
+## Run the experiments
+
+```
+import bb25 as bb
+
+results = bb.run_experiments()
+print(all(r.passed for r in results))
+```
+
+## Sample script
+
+See `docs/sample_usage.py` for an end-to-end example using BM25, Bayesian calibration, and hybrid fusion.
+
+## Build from source (Rust)
+
+```
+make build
+```
+
+## PyPI publishing
+
+Build a wheel with maturin:
+
+```
+python -m pip install maturin
+maturin build --release
+```
+
+For Pyodide builds, see `docs/pyodide.md`.
+

bb25-0.1.1.dist-info/RECORD

@@ -0,0 +1,5 @@
+bb25/__init__.py,sha256=cAbH7A3F8yvvXg52Ovx7vuohGYPBlaSlFhTY4WSQ3u0,99
+bb25/bb25.cpython-312-darwin.so,sha256=GpO4x2yAe_WpOd7bY7re1_CUlk1_bnRd0BfoQ1zLKCQ,988784
+bb25-0.1.1.dist-info/METADATA,sha256=96zK1oppaNKgKcO1BBQxHptb1H2pauToV10pTSgObfc,3006
+bb25-0.1.1.dist-info/WHEEL,sha256=qr8B0oB3ZR0cwIaW0mxUADH9b4sNRMHcgJlpYNTDAPw,105
+bb25-0.1.1.dist-info/RECORD,,

bb25-0.1.1.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: maturin (1.11.5)
+Root-Is-Purelib: false
+Tag: cp312-cp312-macosx_11_0_arm64