Explorar el Código

Change network configuration

Alexey Edelev hace 5 años
padre
commit
c0cb4764d5
Se han modificado 1 ficheros con 1 adiciones y 1 borrados
  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,