Set/get command for output neurons via UART.

nthnn · Apr 20, 2024 · 9f19be3 · 9f19be3
1 parent 6d51ce0
commit 9f19be3
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/main.go b/main.go
@@ -143,6 +143,20 @@ func main() {
 			}
 			break
 
+		case N2CMU_SET_OUTPUT_NEURON:
+			for i := 0; i < int(network.OutputCount); i++ {
+				network.OutputNeuron[i] = uart.ReadFloat32()
+			}
+
+			uart.WriteOk()
+			break
+
+		case N2CMU_GET_OUTPUT_NEURON:
+			for i := 0; i < int(network.OutputCount); i++ {
+				uart.WriteFloat32(network.OutputNeuron[i])
+			}
+			break
+
 		case N2CMU_SET_EPOCH_COUNT:
 			epoch = uart.ReadUint16()
 			break