瀏覽代碼

Change network configuration

Alexey Edelev 5 年之前
父節點
當前提交
c0cb4764d5
共有 1 個文件被更改,包括 1 次插入1 次删除
  1. 1 1
      handwriting/handwriting/handwriting.go

+ 1 - 1
handwriting/handwriting/handwriting.go

@@ -20,7 +20,7 @@ type HandwritingService struct {
 
 func NewHandwritingService() (hws *HandwritingService) {
 	hws = &HandwritingService{}
-	hws.nn, _ = neuralnetwork.NewNeuralNetwork([]int{784, 16, 16, 10}, neuralnetwork.NewRPropInitializer(neuralnetwork.RPropConfig{
+	hws.nn, _ = neuralnetwork.NewNeuralNetwork([]int{784, 300, 200, 10}, neuralnetwork.NewRPropInitializer(neuralnetwork.RPropConfig{
 		NuPlus:   1.2,
 		NuMinus:  0.5,
 		DeltaMax: 50.0,