Skip to content
Snippets Groups Projects
Commit 24cb1f01 authored by Lars Buitinck's avatar Lars Buitinck
Browse files

PY3 fix RBM test

parent de6e1a38
Branches
Tags
No related merge requests found
...@@ -18,7 +18,7 @@ ...@@ -18,7 +18,7 @@
/sklearn/utils/arrayfuncs.c -diff /sklearn/utils/arrayfuncs.c -diff
/sklearn/utils/graph_shortest_path.c -diff /sklearn/utils/graph_shortest_path.c -diff
/sklearn/utils/lgamma.c -diff /sklearn/utils/lgamma.c -diff
sklearn/utils/_logistic_sigmoid.c -diff /sklearn/utils/_logistic_sigmoid.c -diff
/sklearn/utils/murmurhash.c -diff /sklearn/utils/murmurhash.c -diff
/sklearn/utils/seq_dataset.c -diff /sklearn/utils/seq_dataset.c -diff
/sklearn/utils/sparsefuncs.c -diff /sklearn/utils/sparsefuncs.c -diff
......
...@@ -101,15 +101,12 @@ def test_gibbs_smoke(): ...@@ -101,15 +101,12 @@ def test_gibbs_smoke():
def test_pseudo_likelihood_no_clipping(): def test_pseudo_likelihood_no_clipping():
""" """Check that the pseudo likelihood is computed without clipping.
checks that the pseudo likelihood is computed without clipping,
which happened until commit 52d1b778ca7164ac04ea9f8ba39077054954b77a
the new implementation (as of commit
52d1b778ca7164ac04ea9f8ba39077054954b77a ) follows:
http://fa.bianp.net/blog/2013/numerical-optimizers-for-logistic-regression/ http://fa.bianp.net/blog/2013/numerical-optimizers-for-logistic-regression/
""" """
rng = np.random.RandomState(42) rng = np.random.RandomState(42)
X = np.array([[0. for i in xrange(1000)], [1. for i in xrange(1000)]]) X = np.vstack([np.zeros(1000), np.ones(1000)])
rbm1 = BernoulliRBM(n_components=10, batch_size=2, rbm1 = BernoulliRBM(n_components=10, batch_size=2,
n_iter=10, random_state=rng) n_iter=10, random_state=rng)
rbm1.fit(X) rbm1.fit(X)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment