diff --git a/mlrose/neural.py b/mlrose/neural.py index 16c8971a..c726626f 100644 --- a/mlrose/neural.py +++ b/mlrose/neural.py @@ -9,7 +9,7 @@ import numpy as np from sklearn.base import BaseEstimator, ClassifierMixin, RegressorMixin from sklearn.metrics import mean_squared_error, log_loss -from sklearn.externals import six +import six from .activation import identity, relu, sigmoid, softmax, tanh from .algorithms import random_hill_climb, simulated_annealing, genetic_alg from .opt_probs import ContinuousOpt diff --git a/setup.py b/setup.py index 3c010a8e..cbcada8a 100644 --- a/setup.py +++ b/setup.py @@ -36,6 +36,6 @@ def readme(): "Topic :: Software Development :: Libraries", "Topic :: Software Development :: Libraries :: Python Modules"], packages=['mlrose'], - install_requires=['numpy', 'scipy', 'scikit-learn'], + install_requires=['six', 'numpy', 'scipy', 'scikit-learn'], python_requires='>=3', zip_safe=False)