forked from EduardoGarrido90/Spearmint
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathsetup.py
42 lines (36 loc) · 1.69 KB
/
setup.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
import os
from setuptools import setup
# Utility function to read the README file.
# Used for the long_description. It's nice, because now 1) we have a top level
# README file and 2) it's easier to type in the README file than to put a raw
# string in below ...
def read(fname):
return open(os.path.join(os.path.dirname(__file__), fname)).read()
from numpy.distutils.core import Extension
# This is needed for the SUR acquisition function for multi-objective optimization
ext1 = Extension(name = 'pbivnorm', sources = ['./pbivnorm/pbivnorm.pyf', './pbivnorm/pbivnorm.f'])
from numpy.distutils.core import setup
setup(
name = "spearmint",
version = "0.1",
author = "Jasper Snoek, Ryan Adams, Kevin Swersky, Michael Gelbart, Hugo Larochelle, Daniel Hernandez-Lobato and Jose Miguel Hernandez-Lobato",
author_email = "[email protected], [email protected], [email protected], [email protected], [email protected], [email protected], [email protected]",
description = ("A package for Bayesian optimization."),
keywords = "Bayesian Optimization, Magic, Minty Freshness",
packages=['spearmint',
'spearmint.acquisition_functions',
'spearmint.choosers',
'spearmint.grids',
'spearmint.kernels',
'spearmint.models',
'spearmint.resources',
'spearmint.sampling',
'spearmint.schedulers',
'spearmint.tasks',
'spearmint.transformations',
'spearmint.utils',
'spearmint.visualizations',
'spearmint.utils.database',],
long_description=read('README.md'),
ext_modules = [ ext1 ]
)