Fix basic tests

neuralnetwork/neuralnetworkbase/neuralnetwork_test.go

@@ -1,67 +1,67 @@
 package neuralnetworkbase
 
-// import (
-// 	"testing"
+import (
+	"testing"
 
-// 	"gonum.org/v1/gonum/mat"
-// )
+	"gonum.org/v1/gonum/mat"
+)
 
-// func TestNewNeuralNetwork(t *testing.T) {
-// 	nn, err := NewNeuralNetwork([]int{}, 0.1, 500)
-// 	if nn != nil || err == nil {
-// 		t.Error("nn initialized, but shouldn't ", err)
-// 	}
+func TestNewNeuralNetwork(t *testing.T) {
+	nn, err := NewNeuralNetwork([]int{}, 500, NewBackPropInitializer(0.1))
+	if nn != nil || err == nil {
+		t.Error("nn initialized, but shouldn't ", err)
+	}
 
-// 	nn, err = NewNeuralNetwork([]int{0, 0, 0, 0}, 0.1, 500)
-// 	if nn != nil || err == nil {
-// 		t.Error("nn initialized, but shouldn't ", err)
-// 	}
+	nn, err = NewNeuralNetwork([]int{0, 0, 0, 0}, 500, NewBackPropInitializer(0.1))
+	if nn != nil || err == nil {
+		t.Error("nn initialized, but shouldn't ", err)
+	}
 
-// 	nn, err = NewNeuralNetwork([]int{1, 1, 1, 1}, 0.1, 500)
-// 	if nn != nil || err == nil {
-// 		t.Error("nn initialized, but shouldn't ", err)
-// 	}
+	nn, err = NewNeuralNetwork([]int{1, 1, 1, 1}, 500, NewBackPropInitializer(0.1))
+	if nn != nil || err == nil {
+		t.Error("nn initialized, but shouldn't ", err)
+	}
 
-// 	nn, err = NewNeuralNetwork([]int{5, 5}, 0.1, 500)
-// 	if nn != nil || err == nil {
-// 		t.Error("nn initialized, but shouldn't ", err)
-// 	}
+	nn, err = NewNeuralNetwork([]int{5, 5}, 500, NewBackPropInitializer(0.1))
+	if nn != nil || err == nil {
+		t.Error("nn initialized, but shouldn't ", err)
+	}
 
-// 	nn, err = NewNeuralNetwork([]int{5, 1, 5, 5}, 0.1, 500)
-// 	if nn != nil || err == nil {
-// 		t.Error("nn initialized, but shouldn't ", err)
-// 	}
+	nn, err = NewNeuralNetwork([]int{5, 1, 5, 5}, 500, NewBackPropInitializer(0.1))
+	if nn != nil || err == nil {
+		t.Error("nn initialized, but shouldn't ", err)
+	}
 
-// 	nn, err = NewNeuralNetwork([]int{5, 4, 4, 5}, 0.1, 500)
-// 	if nn == nil || err != nil {
-// 		t.Error("nn is not initialized, but should be ", err)
-// 	}
-// }
+	nn, err = NewNeuralNetwork([]int{5, 4, 4, 5}, 500, NewBackPropInitializer(0.1))
+	if nn == nil || err != nil {
+		t.Error("nn is not initialized, but should be ", err)
+	}
+}
 
-// func TestNeuralNetworkPredict(t *testing.T) {
-// 	nn, _ := NewNeuralNetwork([]int{3, 4, 4, 2}, 0.1, 500)
+func TestNeuralNetworkPredict(t *testing.T) {
+	nn, _ := NewNeuralNetwork([]int{3, 4, 4, 2}, 500, NewBackPropInitializer(0.1))
 
-// 	aIn := &mat.Dense{}
-// 	index, max := nn.Predict(aIn)
-// 	if index != -1 || max != 0.0 {
-// 		t.Error("Prediction when empty aIn shouldn't be possibe but predicted", index, max)
-// 	}
+	aIn := &mat.Dense{}
+	index, max := nn.Predict(aIn)
+	if index != -1 || max != 0.0 {
+		t.Error("Prediction when empty aIn shouldn't be possibe but predicted", index, max)
+	}
 
-// 	aIn = mat.NewDense(2, 1, []float64{0.1, 0.2})
-// 	index, max = nn.Predict(aIn)
-// 	if index != -1 || max != 0.0 {
-// 		t.Error("Prediction aIn has invalid size shouldn't be possibe but predicted", index, max)
-// 	}
+	aIn = mat.NewDense(2, 1, []float64{0.1, 0.2})
+	index, max = nn.Predict(aIn)
+	if index != -1 || max != 0.0 {
+		t.Error("Prediction aIn has invalid size shouldn't be possibe but predicted", index, max)
+	}
 
-// 	aIn = mat.NewDense(3, 1, []float64{0.1, 0.2, 0.3})
-// 	index, max = nn.Predict(aIn)
-// 	if index == -1 || max == 0.0 {
-// 		t.Error("Prediction of aIn valid size should be predicted", index, max)
-// 	}
+	aIn = mat.NewDense(3, 1, []float64{0.1, 0.2, 0.3})
+	index, max = nn.Predict(aIn)
+	if index == -1 || max == 0.0 {
+		t.Error("Prediction of aIn valid size should be predicted", index, max)
+	}
 
-// 	aIn = mat.NewDense(4, 1, []float64{0.1, 0.2, 0.3, 0.4})
-// 	index, max = nn.Predict(aIn)
-// 	if index != -1 || max != 0.0 {
-// 		t.Error("Prediction aIn has invalid size shouldn't be possibe but predicted", index, max)
-// 	}
-// }
+	aIn = mat.NewDense(4, 1, []float64{0.1, 0.2, 0.3, 0.4})
+	index, max = nn.Predict(aIn)
+	if index != -1 || max != 0.0 {
+		t.Error("Prediction aIn has invalid size shouldn't be possibe but predicted", index, max)
+	}
+}