huggingface / huggingface/text-clustering

Example fails

Open
#10 0 comments 0 reactions 0 assignees View on GitHub
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.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.