main.go 1.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263
  1. package main
  2. import (
  3. neuralnetwork "./neuralnetwork"
  4. remotecontrol "./remotecontrol"
  5. snakesimulator "./snakesimulator"
  6. )
  7. func main() {
  8. s := snakesimulator.NewSnakeSimulator()
  9. s.Run()
  10. // genetic.NewPopulation(nil, mutagen.NewDummyMutagen(50), 200, []int{13, 8, 12, 3})
  11. sizes := []int{13, 8, 12, 3}
  12. nn, _ := neuralnetwork.NewNeuralNetwork(sizes, neuralnetwork.NewRPropInitializer(neuralnetwork.RPropConfig{
  13. NuPlus: 1.2,
  14. NuMinus: 0.5,
  15. DeltaMax: 50.0,
  16. DeltaMin: 0.000001,
  17. }))
  18. rc := &remotecontrol.RemoteControl{}
  19. nn.SetStateWatcher(rc)
  20. // inFile, err := os.Open("./networkstate")
  21. // if err != nil {
  22. // log.Fatal(err)
  23. // }
  24. // defer inFile.Close()
  25. // nn.LoadState(inFile)
  26. // nn, _ := neuralnetwork.NewNeuralNetwork(sizes, neuralnetwork.NewBackPropInitializer(0.1))
  27. // for i := 0; i < nn.Count; i++ {
  28. // if i > 0 {
  29. // fmt.Printf("Weights before:\n%v\n\n", mat.Formatted(nn.Weights[i], mat.Prefix(""), mat.Excerpt(0)))
  30. // fmt.Printf("Biases before:\n%v\n\n", mat.Formatted(nn.Biases[i], mat.Prefix(""), mat.Excerpt(0)))
  31. // fmt.Printf("Z before:\n%v\n\n", mat.Formatted(nn.Z[i], mat.Prefix(""), mat.Excerpt(0)))
  32. // }
  33. // fmt.Printf("A before:\n%v\n\n", mat.Formatted(nn.A[i], mat.Prefix(""), mat.Excerpt(0)))
  34. // }
  35. // nn = &neuralnetwork.NeuralNetwork{}
  36. // inFile, err := os.Open("./data")
  37. // if err != nil {
  38. // log.Fatal(err)
  39. // }
  40. // defer inFile.Close()
  41. // nn.LoadState(inFile)
  42. // inFile.Close()
  43. // failCount = 0
  44. // training.Reset()
  45. // for training.NextValidator() {
  46. // dataSet, expect := training.GetValidator()
  47. // index, _ := nn.Predict(dataSet)
  48. // if expect.At(index, 0) != 1.0 {
  49. // failCount++
  50. // // fmt.Printf("Fail: %v, %v\n\n", training.ValidationIndex(), expect.At(index, 0))
  51. // }
  52. // }
  53. // fmt.Printf("Fail count: %v\n\n", failCount)
  54. rc.Run()
  55. }