Forráskód Böngészése

Change network configuration

Alexey Edelev 5 éve
szülő
commit
c0cb4764d5
1 módosított fájl, 1 hozzáadás és 1 törlés
  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,