def gather(quick=False):
    modules = ['linear_model'] if quick else \
              ['linear_model', 'cluster', 'semi_supervised', 'naive_bayes',
               'svm', 'decomposition', 'neighbors', 'covariance', 'manifold',
               'gaussian_process', 'mixture', 'pls', 'tree', 'ensemble']

    by_module = {}
    benchmarks = []
    for mod in modules:
        by_module[mod] = [b
                          for b in gather_benchmarks(__import__(mod).__dict__)]
        benchmarks.extend(by_module[mod])

    return benchmarks, by_module
Beispiel #2
0
from datetime import datetime, timedelta

from vbench.api import GitRepo
from vbench.benchmark import gather_benchmarks

###################
# Gather benchmarks

modules = ['linear_model', 'cluster', 'semi_supervised', 'naive_bayes', 'svm',
           'decomposition', 'neighbors', 'covariance', 'gaussian_process',
           'mixture', 'pls']

by_module = {}
benchmarks = []
for mod in modules:
    by_module[mod] = [b for b in gather_benchmarks(__import__(mod).__dict__)]
    benchmarks.extend(by_module[mod])

for bm in benchmarks:
    assert(bm.name is not None)

###############
# Configuration

USERNAME = getpass.getuser()

# XXX: do we want platform independence here?
# I'm guessing the original supported MacOS because of the dev's laptop, and
# *nix because of the deployment server. Will we ever deploy this on IIS? :)

if sys.platform == 'darwin':