Skip to content

Commit 24cb1f0

Browse files
committed
PY3 fix RBM test
1 parent de6e1a3 commit 24cb1f0

File tree

2 files changed

+4
-7
lines changed

2 files changed

+4
-7
lines changed

.gitattributes

+1-1
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@
1818
/sklearn/utils/arrayfuncs.c -diff
1919
/sklearn/utils/graph_shortest_path.c -diff
2020
/sklearn/utils/lgamma.c -diff
21-
sklearn/utils/_logistic_sigmoid.c -diff
21+
/sklearn/utils/_logistic_sigmoid.c -diff
2222
/sklearn/utils/murmurhash.c -diff
2323
/sklearn/utils/seq_dataset.c -diff
2424
/sklearn/utils/sparsefuncs.c -diff

sklearn/neural_network/tests/test_rbm.py

+3-6
Original file line numberDiff line numberDiff line change
@@ -101,15 +101,12 @@ def test_gibbs_smoke():
101101

102102

103103
def test_pseudo_likelihood_no_clipping():
104-
"""
105-
checks that the pseudo likelihood is computed without clipping,
106-
which happened until commit 52d1b778ca7164ac04ea9f8ba39077054954b77a
107-
the new implementation (as of commit
108-
52d1b778ca7164ac04ea9f8ba39077054954b77a ) follows:
104+
"""Check that the pseudo likelihood is computed without clipping.
105+
109106
http://fa.bianp.net/blog/2013/numerical-optimizers-for-logistic-regression/
110107
"""
111108
rng = np.random.RandomState(42)
112-
X = np.array([[0. for i in xrange(1000)], [1. for i in xrange(1000)]])
109+
X = np.vstack([np.zeros(1000), np.ones(1000)])
113110
rbm1 = BernoulliRBM(n_components=10, batch_size=2,
114111
n_iter=10, random_state=rng)
115112
rbm1.fit(X)

0 commit comments

Comments
 (0)