Bläddra i källkod

Few UI improvements

- Add delays to newural network prediction
Alexey Edelev 5 år sedan
förälder
incheckning
37cd824031
2 ändrade filer med 12 tillägg och 10 borttagningar
  1. 8 7
      gui/main.qml
  2. 4 3
      neuralnetwork/main.go

+ 8 - 7
gui/main.qml

@@ -63,18 +63,19 @@ ApplicationWindow {
                     Repeater {
                         id: neuronRepeater
                         model: layerSize
-                        delegate: Rectangle {
-                            id: neuron
-                            property ValueIndicator activation: visualizerModel.activation(layerIndex, model.index)
+                        delegate:Rectangle {
                             width: 30
                             height: 30
                             radius: 15
-                            color: {
-                                if (activation.value > 0) {
+                            color: "#00ffffff"
+                            Rectangle {
+                                id: neuron
+                                anchors.fill: parent
+                                property ValueIndicator activation: visualizerModel.activation(layerIndex, model.index)
+                                radius: 15
+                                color: {
                                     var alpha = activation.value
                                     Qt.rgba(0, 1, 0, alpha)
-                                } else {
-                                    "#ffffffff"
                                 }
                             }
                         }

+ 4 - 3
neuralnetwork/main.go

@@ -12,7 +12,7 @@ import (
 )
 
 func main() {
-	sizes := []int{13, 14, 14, 3}
+	sizes := []int{13, 12, 12, 3}
 	nn, _ := neuralnetwork.NewNeuralNetwork(sizes, neuralnetwork.NewRPropInitializer(neuralnetwork.RPropConfig{
 		NuPlus:   1.2,
 		NuMinus:  0.5,
@@ -42,8 +42,8 @@ func main() {
 
 	go func() {
 		// teacher := teach.NewMNISTReader("./minst.data", "./mnist.labels")
-		teacher := teach.NewTextDataReader("wine.data", 7)
-		nn.Teach(teacher, 1000)
+		teacher := teach.NewTextDataReader("wine.data", 5)
+		nn.Teach(teacher, 1500)
 
 		// for i := 0; i < nn.Count; i++ {
 		// 	if i > 0 {
@@ -62,6 +62,7 @@ func main() {
 		nn.SaveState(outFile)
 		outFile.Close()
 
+		time.Sleep(5 * time.Second)
 		failCount := 0
 		teacher.Reset()
 		for teacher.NextValidator() {