diff --git a/nolearn/lasagne/base.py b/nolearn/lasagne/base.py index 155fdb1..881e3b5 100644 --- a/nolearn/lasagne/base.py +++ b/nolearn/lasagne/base.py @@ -22,8 +22,7 @@ from lasagne.utils import unique import numpy as np from sklearn.base import BaseEstimator -from sklearn.cross_validation import KFold -from sklearn.cross_validation import StratifiedKFold +from sklearn.model_selection import KFold, StratifiedKFold from sklearn.metrics import accuracy_score from sklearn.metrics import r2_score from sklearn.preprocessing import LabelEncoder diff --git a/nolearn/lasagne/tests/test_base.py b/nolearn/lasagne/tests/test_base.py index 49fed89..b410af8 100644 --- a/nolearn/lasagne/tests/test_base.py +++ b/nolearn/lasagne/tests/test_base.py @@ -15,14 +15,14 @@ from lasagne.objectives import categorical_crossentropy from lasagne.objectives import aggregate from lasagne.updates import nesterov_momentum -from mock import Mock -from mock import patch +from unittest.mock import Mock +from unittest.mock import patch import numpy as np import pytest from sklearn.base import clone from sklearn.datasets import make_classification from sklearn.datasets import make_regression -from sklearn.grid_search import GridSearchCV +from sklearn.model_selection import GridSearchCV from sklearn.metrics import accuracy_score from sklearn.metrics import mean_absolute_error from sklearn.metrics import r2_score diff --git a/nolearn/lasagne/tests/test_handlers.py b/nolearn/lasagne/tests/test_handlers.py index 6e1231a..17b4ff0 100644 --- a/nolearn/lasagne/tests/test_handlers.py +++ b/nolearn/lasagne/tests/test_handlers.py @@ -8,8 +8,8 @@ from lasagne.layers import InputLayer from lasagne.nonlinearities import softmax from lasagne.updates import nesterov_momentum -from mock import patch -from mock import Mock +from unittest.mock import patch +from unittest.mock import Mock import numpy import pytest diff --git a/nolearn/lasagne/tests/test_visualize.py b/nolearn/lasagne/tests/test_visualize.py index 7114a30..f83a9d0 100644 --- a/nolearn/lasagne/tests/test_visualize.py +++ b/nolearn/lasagne/tests/test_visualize.py @@ -8,7 +8,7 @@ class TestCNNVisualizeFunctions: def X_non_square(self, X_train): X = np.hstack( (X_train[:, :20 * 28], X_train[:, :20 * 28], X_train[:, :20 * 28])) - X = X.reshape(-1, 3, 20, 28) + X = X.reshape((-1, 3, 20, 28)) return X def test_plot_loss(self, net_fitted): diff --git a/nolearn/overfeat.py b/nolearn/overfeat.py index 10e4cf1..15445cb 100644 --- a/nolearn/overfeat.py +++ b/nolearn/overfeat.py @@ -113,7 +113,7 @@ def _compute_features(self, fnames): n_feat, n_rows, n_cols = data[i * 2].split() n_feat, n_rows, n_cols = int(n_feat), int(n_rows), int(n_cols) feat = np.fromstring(data[i * 2 + 1], dtype=np.float32, sep=' ') - feat = feat.reshape(n_feat, n_rows, n_cols) + feat = feat.reshape((n_feat, n_rows, n_cols)) if self.merge == 'maxmean': feat = feat.max(2).mean(1) elif self.merge == 'meanmax': diff --git a/nolearn/tests/test_cache.py b/nolearn/tests/test_cache.py index 7e3be06..b2652f6 100644 --- a/nolearn/tests/test_cache.py +++ b/nolearn/tests/test_cache.py @@ -1,4 +1,4 @@ -from mock import patch +from unittest.mock import patch def test_cached(tmpdir): diff --git a/nolearn/tests/test_grid_search.py b/nolearn/tests/test_grid_search.py index 5293111..001fe41 100644 --- a/nolearn/tests/test_grid_search.py +++ b/nolearn/tests/test_grid_search.py @@ -1,6 +1,6 @@ from sklearn.linear_model import LogisticRegression -from sklearn.grid_search import GridSearchCV -from mock import Mock +from sklearn.model_selection import GridSearchCV +from unittest.mock import Mock import numpy as np diff --git a/nolearn/tests/test_metrics.py b/nolearn/tests/test_metrics.py index e78f953..62a6a95 100644 --- a/nolearn/tests/test_metrics.py +++ b/nolearn/tests/test_metrics.py @@ -1,6 +1,6 @@ -from mock import Mock +from unittest.mock import Mock import numpy as np -from sklearn.cross_validation import train_test_split +from sklearn.model_selection import train_test_split from sklearn.linear_model import LogisticRegression diff --git a/nolearn/util.py b/nolearn/util.py index f049989..a59bbfc 100644 --- a/nolearn/util.py +++ b/nolearn/util.py @@ -6,7 +6,7 @@ def chunks(l, n): """ Yield successive n-sized chunks from l. """ - for i in xrange(0, len(l), n): + for i in range(0, len(l), n): yield l[i:i + n] diff --git a/requirements.txt b/requirements.txt index 364c11a..4205de3 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,7 +1,8 @@ -numpy==1.10.4 -scipy==0.16.1 +numpy==1.16.3 +scipy==1.2.1 Theano==0.8 -e git+https://github.com/Lasagne/Lasagne.git@8f4f9b2#egg=Lasagne==0.2.git joblib==0.9.3 -scikit-learn==0.17 +scikit-learn==0.19.2 tabulate==0.7.5 +matplotlib \ No newline at end of file diff --git a/setup.cfg b/setup.cfg index b45301d..5f4e47b 100644 --- a/setup.cfg +++ b/setup.cfg @@ -5,5 +5,4 @@ addopts = --flakes --pep8 -v - nolearn/tests/ nolearn/lasagne/tests/ python_files = test*py