Skip to content

Commit

Permalink
rbm.vis -> vis (#1)
Browse files Browse the repository at this point in the history
Wanted to check the size of vis argument, not a parameter of the rbm
class…
  • Loading branch information
eric-tramel committed Sep 9, 2015
1 parent f5530ab commit 41fecfc
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/rbm.jl
Original file line number Diff line number Diff line change
Expand Up @@ -106,7 +106,7 @@ function gibbs(rbm::RBM, vis::Mat{Float64}; n_times=1,dorate=0.0)
#
# We need to remember that the visible units are being passed to
# here as a matrix, so we need to generate a matrix of dropout patterns.
suppressedUnits = rand(size(rbm.hbias,1),size(rbm.vis,2)) .< dorate
suppressedUnits = rand(size(rbm.hbias,1),size(vis,2)) .< dorate

# Every time we sample the visible units, we need to
# make sure that we do so only using the "active" hidden
Expand Down

0 comments on commit 41fecfc

Please sign in to comment.