提交 a52f79b3 authored 作者: Amjad Almahairi's avatar Amjad Almahairi

fix for python 3x

上级 a677494a
......@@ -81,7 +81,7 @@ def test_n_samples_compatibility():
pickle.dump([X, samples], open("multinomial_test_graph.pkl", "w"))
"""
folder = os.path.dirname(os.path.abspath(__file__))
X, samples = pickle.load(open(os.path.join(folder, "multinomial_test_graph.pkl")))
X, samples = pickle.load(open(os.path.join(folder, "multinomial_test_graph.pkl"), 'rb'))
f = theano.function([X], samples)
res = f(numpy.random.randn(20,10))
assert numpy.all(res.sum(axis=1) == 1)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论