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
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
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':