|
@@ -11,7 +11,7 @@ func main() {
|
|
|
|
|
|
dataSet, result := readData("./iris.data")
|
|
|
|
|
|
- sizes := []int{4, 8, 3}
|
|
|
+ sizes := []int{4, 8, 8, 3}
|
|
|
nn := neuralnetwork.NewNeuralNetwork(sizes)
|
|
|
|
|
|
for i := 0; i < nn.Count; i++ {
|
|
@@ -23,8 +23,8 @@ func main() {
|
|
|
fmt.Printf("A before:\n%v\n\n", mat.Formatted(nn.A[i], mat.Prefix(""), mat.Excerpt(0)))
|
|
|
}
|
|
|
|
|
|
- for j := 0; j < 150; j++ {
|
|
|
- for i := 0; i < len(dataSet); i++ {
|
|
|
+ for j := 0; j < 481; j++ {
|
|
|
+ for i := len(dataSet) - 1; i >= 0; i-- {
|
|
|
// fmt.Printf("Dataset[%d]:\n%v\n\n", i, mat.Formatted(dataSet[i], mat.Prefix(""), mat.Excerpt(0)))
|
|
|
// fmt.Printf("Result[%d]:\n%v\n\n", i, mat.Formatted(result[i], mat.Prefix(""), mat.Excerpt(0)))
|
|
|
nn.Backward(dataSet[i], result[i])
|