added docs

theGreatHerrLebert · Oct 25, 2023 · 51c5916 · 51c5916
1 parent 4afe27a
commit 51c5916
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/pyims/pyims/mixture.py b/pyims/pyims/mixture.py
@@ -147,6 +147,7 @@ def predict_proba(self, data):
         return normalized_probs.numpy()
 
     def predict(self, data):
+        """Get the cluster ids under the current mixture model"""
         return np.argmax(self.predict_proba(data), axis=1)
 
     def sample(self, n_samples=1):