Update README.rst

nicodv · Dec 7, 2019 · 4f78070 · 4f78070
1 parent 5f9c858
commit 4f78070
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/README.rst b/README.rst
@@ -146,6 +146,12 @@ when using the :code:`kprototypes` algorithm.
 A: One or more of your numerical feature columns have string values in them. Make sure that all 
 columns have consistent data types.
 
+Q: How does k-protypes know which of my features are numerical and which are categorical?
+
+A: You tell it which column indices are categorical using the `categorical` argument. All others are assumed numerical.
+```kproto = KPrototypes()
+clusters = kproto.fit_predict(X, categorical=[1, 2])```
+
 References
 ----------