FIX random_state setting

arjoly · arjoly · commit 535d7fca650c · 2014-08-22T16:50:28.000+02:00
diff --git a/benchmarks/bench_make_multilabel_classification.py b/benchmarks/bench_make_multilabel_classification.py
@@ -0,0 +1,23 @@
+from __future__ import print_function, division
+
+import argparse
+import numpy as np
+
+from sklearn.datasets import make_multilabel_classification
+
+parser = argparse.ArgumentParser(description='Benchmark')
+parser.add_argument('-s','--seed', type=int, default=None, nargs="?",
+                    help="Random number generator seed")
+parser.add_argument('-n', '--n_samples', type=int, default=10 ** 3, nargs="?")
+parser.add_argument('-p', '--n_features', type=int, default=10 ** 3, nargs="?")
+parser.add_argument('-d', '--n_classes', type=int, default=10 ** 3, nargs="?")
+parser.add_argument('-k', '--n_labels', type=int, default=10 ** 2, nargs="?")
+
+args = vars(parser.parse_args())
+print(args)
+
+np.random.seed(args["seed"])
+
+make_multilabel_classification(n_samples=args["n_samples"],
+    n_features=args["n_features"], n_classes=args["n_classes"],
+    n_labels=args["n_labels"])
