Add clustering example to the docs

anton-bushuiev · Jun 24, 2024 · 7548b02 · 7548b02
1 parent 38b1063
commit 7548b02
Show file tree

Hide file tree

Showing 3 changed files with 348 additions and 1 deletion.
diff --git a/docs/clustering_ppis.ipynb b/docs/clustering_ppis.ipynb
diff --git a/docs/index.rst b/docs/index.rst
@@ -21,6 +21,7 @@ Contents
    analysing_ppis
    comparing_ppis
    deduplicating_ppis
+   clustering_ppis
    retrieving_ppis
    splitting_ppis
    modules
diff --git a/tests/test_comparison.py b/tests/test_comparison.py
@@ -59,4 +59,4 @@ def test_idist_cluster_embeddings():
 
     # Test that all near duplicates (same letters in IDs) are in same clusters
     df = pd.DataFrame({'letters': [x[0] for x in idist.embeddings.keys()], 'labels': labels})
-    assert df.groupby('letters')['labels'].nunique().eq(1).all()
+    assert df.groupby('letters')['labels'].nunique().eq(1).all()