File tree Expand file tree Collapse file tree 2 files changed +26
-16
lines changed Expand file tree Collapse file tree 2 files changed +26
-16
lines changed Original file line number Diff line number Diff 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
4244This will run for one hour should result in an accuracy above 0.98.
Original file line number Diff line number Diff line change 11# -*- encoding: utf-8 -*-
2- import os
3-
4- try :
5- from urllib import urlretrieve
6- except :
7- from urllib .request import urlretrieve
8-
92import autosklearn
103
114import setuptools
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
3341setuptools .setup (
You can’t perform that action at this time.
0 commit comments