From 859c43233c8394602b7af3710859cb10e1e26112 Mon Sep 17 00:00:00 2001 From: Timo Klock Date: Tue, 20 Nov 2018 16:53:44 +0100 Subject: [PATCH] Exchanged at_least_2d function with check_array function, since former has been removed from sklearn --- random_layer.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/random_layer.py b/random_layer.py index 713a795..52f459e 100644 --- a/random_layer.py +++ b/random_layer.py @@ -24,7 +24,7 @@ from scipy.spatial.distance import cdist, pdist, squareform from sklearn.metrics import pairwise_distances -from sklearn.utils import check_random_state, atleast2d_or_csr +from sklearn.utils import check_random_state, check_array from sklearn.utils.extmath import safe_sparse_dot from sklearn.base import BaseEstimator, TransformerMixin @@ -108,7 +108,7 @@ def fit(self, X, y=None): ------- self """ - X = atleast2d_or_csr(X) + X = check_array(X) self._generate_components(X) @@ -130,7 +130,7 @@ def transform(self, X, y=None): ------- X_new : numpy array of shape [n_samples, n_components] """ - X = atleast2d_or_csr(X) + X = check_array(X) if (self.components_ is None): raise ValueError('No components initialized')