ERROR: TypeError: fitdbm: in typeassert, expected Array{Int64,1}, got Array{Int64,2}
in trainpartdbm
function trainpartdbm(mat;partitions::Array{Array{Int64,1},1}=[collect(1:size(mat)[2])],
nhiddens::Array{Int64,1}=[size(mat)[2],div(size(mat)[2],10)],
epochs::Int64=20,
nparticles::Int64=100,
learningrate::Float64=0.005)
partDBMS = Vector{BoltzmannMachines.BasicDBM}()
nhiddensmat = partitionhiddens(partitions,nhiddens)
println(nhiddensmat)
for i=1:length(partitions)
partx = mat[:,partitions[i]]
pdbm = BoltzmannMachines.fitdbm(partx,
nhiddens=nhiddensmat[i,:],
epochs =epochs,
nparticles=nparticles,
learningrate=learningrate
)
push!(partDBMS,pdbm)
end
jointdbm = BoltzmannMachines.joindbms(partDBMS,partitions)
jointdbm,partDBMS
end
end
Kinsly suggest me how to solve it. Thanks in advance