Skip to content

Commit 29128aa

Browse files
committed
MAINT minor fix to front page snippet, make sdist work again
1 parent bba917c commit 29128aa

File tree

2 files changed

+26
-16
lines changed

2 files changed

+26
-16
lines changed

doc/index.rst

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -29,14 +29,16 @@ Example
2929
>>> import autosklearn.classification
3030
>>> import sklearn.cross_validation
3131
>>> import sklearn.datasets
32+
>>> import sklearn.metrics
3233
>>> digits = sklearn.datasets.load_digits()
3334
>>> X = digits.data
3435
>>> y = digits.target
3536
>>> X_train, X_test, y_train, y_test = \
3637
sklearn.cross_validation.train_test_split(X, y, random_state=1)
3738
>>> automl = autosklearn.classification.AutoSklearnClassifier()
3839
>>> automl.fit(X_train, y_train)
39-
>>> print(automl.score(X_test,y_test))
40+
>>> y_hat = automl.predict(X_test)
41+
>>> print("Accuracy score", sklearn.metrics.accuracy_score(y_test, y_hat))
4042

4143

4244
This will run for one hour should result in an accuracy above 0.98.

setup.py

Lines changed: 23 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,4 @@
11
# -*- encoding: utf-8 -*-
2-
import os
3-
4-
try:
5-
from urllib import urlretrieve
6-
except:
7-
from urllib.request import urlretrieve
8-
92
import autosklearn
103

114
import setuptools
@@ -20,14 +13,29 @@
2013
include_dirs=[np.get_include()])
2114
])
2215

23-
requirements_file = os.path.join(os.path.dirname(__file__), 'requirements.txt')
24-
requirements = []
25-
dependency_links = []
26-
with open(requirements_file) as fh:
27-
for line in fh:
28-
line = line.strip()
29-
if line:
30-
requirements.append(line)
16+
requirements = [
17+
"unittest2",
18+
"setuptools",
19+
"mock",
20+
"nose",
21+
"six",
22+
"Cython",
23+
"numpy>=1.9.0",
24+
"scipy>=0.14.1",
25+
"scikit-learn==0.17.1",
26+
"lockfile",
27+
"joblib",
28+
"psutil",
29+
"pyyaml",
30+
"ConfigArgParse",
31+
"liac-arff",
32+
"pandas",
33+
"xgboost==0.4a30",
34+
"ConfigSpace",
35+
"pynisher>=0.4",
36+
"pyrfr",
37+
"smac==0.0.1"
38+
]
3139

3240

3341
setuptools.setup(

0 commit comments

Comments
 (0)