Skip to content

Commit

Permalink
refactor for maintance - splitting up modules, score method to evalue…
Browse files Browse the repository at this point in the history
…, gpu settings consildated
  • Loading branch information
momonga-ml committed Dec 8, 2023
1 parent 71865ae commit c527273
Show file tree
Hide file tree
Showing 12 changed files with 754 additions and 619 deletions.
19 changes: 17 additions & 2 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -38,7 +38,7 @@ df = make_dataframe()
clf = DenseClus(df)
clf.fit(df)

scores = clf.score()
scores = clf.evaluate()
print(scores[0:10])
```

Expand All @@ -47,7 +47,7 @@ print(scores[0:10])

### Prediction

DenseClus uses a `predict` method whhne `umap_combine_method` is set to `ensemble`.
DenseClus uses a `predict` method when `umap_combine_method` is set to `ensemble`.
Results are return in 2d array with the first part being the labels and the second part the probabilities.

```python
Expand Down Expand Up @@ -77,6 +77,21 @@ clf = DenseClus(
)
```

### To Use with GPU with Ensemble

To use with gpu first have [rapids installed](https://docs.rapids.ai/install#selector).
You can do this as setup by providing cuda verision.
`pip install denseclus[gpu-cu12]`

Then to run:

```python
clf = DenseClus(
umap_combine_method="ensemble",
use_gpu=True
)
```


### Advanced Usage

Expand Down
Loading

0 comments on commit c527273

Please sign in to comment.