예제 #1
0
파일: main.go 프로젝트: unixpickle/weakai
func createDBN() rbm.DBN {
	res := make(rbm.DBN, len(LayerSizes)-1)
	for i := 1; i < len(LayerSizes); i++ {
		res[i-1] = rbm.NewRBM(LayerSizes[i-1], LayerSizes[i])
	}
	return res
}
예제 #2
0
파일: main.go 프로젝트: unixpickle/weakai
func buildLayers() rbm.DBN {
	res := make(rbm.DBN, len(HiddenSizes))
	for i, size := range HiddenSizes {
		inputSize := ImageSize
		if i > 0 {
			inputSize = HiddenSizes[i-1]
		}
		res[i] = rbm.NewRBM(inputSize, size)
	}
	return res
}