Skip to content

Commit

Permalink
trying something else (#1)
Browse files Browse the repository at this point in the history
  • Loading branch information
eric-tramel committed Sep 9, 2015
1 parent 032afd5 commit 0dcbb67
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion examples/mnistexample_dropout.jl
Original file line number Diff line number Diff line change
Expand Up @@ -28,7 +28,7 @@ function run_mnist()
X, y = testdata() # test data is smaller, no need to downsample
X = X ./ (maximum(X) - minimum(X))
m = BernoulliRBM(28*28, 300)
fit(m, X; persistent=true, lr=0.1, n_iter=10, batch_size=100, n_gibbs=1, dorate=0.5)
fit(m, X,true,0.1,10,1,0.5)
plot_weights(m.W[1:64, :], (28, 28))
return m
end
Expand Down

0 comments on commit 0dcbb67

Please sign in to comment.