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