remove some style errors in bench code.

neurodata · Mar 10, 2015 · 806cb22 · 806cb22
1 parent 613c3af
commit 806cb22
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/benchmarks/bench_multilabel_metrics.py b/benchmarks/bench_multilabel_metrics.py
@@ -139,7 +139,7 @@ def _plot(results, metrics, formats, title, x_ticks, x_label,
  ap = argparse.ArgumentParser()
  ap.add_argument('metrics', nargs='*', default=sorted(METRICS),
  help='Specifies metrics to benchmark, defaults to all. '
- 'Choices are: '.format(sorted(METRICS)))
+ 'Choices are: {}'.format(sorted(METRICS)))
  ap.add_argument('--formats', nargs='+', choices=sorted(FORMATS),
  help='Specifies multilabel formats to benchmark '
  '(defaults to all).')

diff --git a/benchmarks/bench_sparsify.py b/benchmarks/bench_sparsify.py
@@ -58,7 +58,7 @@ def sparsity_ratio(X):
 X = np.random.randn(n_samples, n_features)
 inds = np.arange(n_samples)
 np.random.shuffle(inds)
-X[inds[n_features/1.2:]] = 0 # sparsify input
+X[inds[int(n_features / 1.2):]] = 0 # sparsify input
 print("input data sparsity: %f" % sparsity_ratio(X))
 coef = 3 * np.random.randn(n_features)
 inds = np.arange(n_features)
@@ -81,17 +81,18 @@ def sparsity_ratio(X):
 clf.fit(X_train, y_train)
 print("model sparsity: %f" % sparsity_ratio(clf.coef_))
 
-@profile
+
 def benchmark_dense_predict():
  for _ in range(300):
  clf.predict(X_test)
 
-@profile
+
 def benchmark_sparse_predict():
  X_test_sparse = csr_matrix(X_test)
  for _ in range(300):
  clf.predict(X_test_sparse)
 
+
 def score(y_test, y_pred, case):
  r2 = r2_score(y_test, y_pred)
  print("r^2 on test data (%s) : %f" % (case, r2))