Ensure random_state is not passed to deterministic classifiers

jni · Nov 24, 2015 · 32a80b5 · 32a80b5
1 parent bc61e5d
commit 32a80b5
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/gala/classify.py b/gala/classify.py
@@ -172,6 +172,8 @@ def get_classifier(name='random forest', *args, **kwargs):
         return DefaultRandomForest(*args, **kwargs)
     elif is_naive_bayes:
         from sklearn.naive_bayes import GaussianNB
+        if 'random_state' in kwargs:
+            del kwargs['random_state']
         return GaussianNB(*args, **kwargs)
     else:
         raise NotImplementedError('Classifier "%s" is either not installed '