Skip to content

Commit

Permalink
Merge pull request #2 from lancedb/readme
Browse files Browse the repository at this point in the history
Add basic Readme
  • Loading branch information
AyushExel authored May 15, 2024
2 parents a62d1ed + 7291fab commit 13e6ccd
Showing 1 changed file with 44 additions and 0 deletions.
44 changes: 44 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,44 @@
# Ragged

Simple utilities for piece-wise evaluation of LLM based chat and retrieval system.

### Setup
Build from source
```
pip install -e .
```

## GUI quickstart
### VectorDB retrieval eval
```
ragged --quickstart vectordb
```

## API Usage
### VectorDB retrieval eval
```python
from ragged.dataset import LlamaIndexDataset
from ragged.metrics.retriever import HitRate
from ragged.search_utils import QueryType
from lancedb.rerankers import CrossEncoderReranker

# 1. Select dataset
# Automatically download the dataset from llama-hub or pass existing path="/path/to/dataset"
dataset = LlamaIndexDataset("Uber10KDataset2021")

# 2. Select eval metrics
hit_rate = HitRate(
dataset,
embedding_registry_id="sentence-transformers",
embed_model_kwarg={"name":"BAAI/bge-small-en-v1.5"},
reranker=CohereReranker(),
)

# 3. Evaluate on desired query types

#print(hit_rate.evaluate(top_k=5, query_type=QueryType.VECTOR)) # Evaluate vector search
print(hit_rate.evaluate(top_k=5, query_type="all")) # Evaliate all possible query types
```

## Create custom Dataset, Metrics, Reranking connectors
# TODO

0 comments on commit 13e6ccd

Please sign in to comment.