Skip to content
This repository was archived by the owner on Mar 6, 2021. It is now read-only.
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 2 additions & 2 deletions plot_elm_comparison.py
Original file line number Diff line number Diff line change
Expand Up @@ -54,7 +54,7 @@
ELM(20,rbf(0.1)) :20 rbf units gamma=0.1

"""
print __doc__
print(__doc__)


# Code source: Gael Varoqueux
Expand All @@ -69,7 +69,7 @@
from matplotlib.colors import ListedColormap
from sklearn.datasets import make_moons, make_circles, make_classification
from sklearn.preprocessing import StandardScaler
from sklearn.cross_validation import train_test_split
from sklearn.model_selection import train_test_split
from sklearn.linear_model import LogisticRegression

from elm import GenELMClassifier
Expand Down
6 changes: 3 additions & 3 deletions random_layer.py
Original file line number Diff line number Diff line change
Expand Up @@ -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

Expand Down Expand Up @@ -108,7 +108,7 @@ def fit(self, X, y=None):
-------
self
"""
X = atleast2d_or_csr(X)
X = check_array(X)

self._generate_components(X)

Expand All @@ -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')
Expand Down