#include <ctime></ctime>

#include "layer.all.inc.cpp"

#include "layer.all.test.inc.cpp"

#include "train.digit.inc.cpp"

bool runTests() {

 if (!AllTest::test()) return false;

 return printf("success\n"), true;

}

int main() {

  srand(time(NULL));

  //return !runTests();

  #define FILENAME "data/output/weights.bin" // 28x28

  printf("create neural network\n");

  Layer l(nullptr, Layout(28, 28)); l.filename                = FILENAME "1";

  //(new LayerSimple<funcsigmoidexp>(l, Layout(256)))->filename = FILENAME "2";</funcsigmoidexp>

  //(new LayerSimple<funcsigmoidexp>(l, Layout(64)))->filename = FILENAME "3";</funcsigmoidexp>

  //(new LayerSimple<funcsigmoidexp>(l, Layout(128)))->filename = FILENAME "4";</funcsigmoidexp>

  //(new LayerSimple<funcsigmoidexp>(l, Layout(64)))->filename  = FILENAME "5";</funcsigmoidexp>

  //(new LayerSimple<funcsigmoidexp>(l, Layout(128)))->filename = FILENAME "5";</funcsigmoidexp>

  (new LayerSimple<funcsigmoidexp>(l, Layout(32)))->filename  = FILENAME "6";</funcsigmoidexp>

  //(new LayerSimple<funcsigmoidexp>(l, Layout(16)))->filename  = FILENAME "7";</funcsigmoidexp>

  (new LayerSimple<funcsigmoidexp>(l, Layout(10)))->filename  = FILENAME "8";</funcsigmoidexp>

  l.sumStat().print();

  printf("load training data\n");

  TrainerDigit t;

  if (!t.loadSymbolMap("data/symbols-data.bin")) return 1;

  //printf("try load previously saved network\n"); l.load();

  printf("train\n");

  t.configure(l, 0.5, 4, 1000000, 0, 0, 0.0000001).run();

  return 0;

}