huggingface / huggingface/text-clustering
Example fails
Open
- Dominant language
- Python
- Stars
- 608
- Forks
- 48
- PR merge metrics
- No merged PRs in 30d
Description
Hi your base example seems to fail to produce `cluster_labels`, and subsequently causes `cc.show()` to fail.
Your example,
NOTE: Using GPU, not summarizing
```python
from src.text_clustering import ClusterClassifier
from datasets import load_dataset
SAMPLE = 100_000
texts = load_dataset("HuggingFaceTB/cosmopedia-100k", split="train").select(range(SAMPLE))["text"]
cc = ClusterClassifier(embed_device="cuda", summary_create=False)
# run the pipeline:
embs, labels, summaries = cc.fit(texts)
# show the results
cc.show()
```
Contributor guide
No contributing guide indexed for this repository
Assessment
This issue has not been assessed yet.