small fixes

mateogianolio · Jun 7, 2015 · 1aa3730 · 1aa3730
1 parent 6cf6268
commit 1aa3730
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 4 deletions.
diff --git a/config.json b/config.json
@@ -1,12 +1,12 @@
 {
- "text": "0123456789",
- "training_set": 50000,
+ "text": "abcdefghijklmnopqrstuvwxyz",
+ "training_set": 20000,
  "testing_set": 500,
  "image_size": 16,
  "threshold": 400,
  "network": {
  "hidden": 40,
- "output": 10,
+ "output": 26,
  "learning_rate": 0.03
  }
 }
diff --git a/examples/0123456789.png b/examples/0123456789.png
diff --git a/examples/abcdefghijklmnopqrstuvwxyz.png b/examples/abcdefghijklmnopqrstuvwxyz.png
diff --git a/main.js b/main.js
@@ -22,6 +22,7 @@
  config.fonts = config.fonts || ['serif', 'sans-serif'];
  config.distortion = config.distortion === undefined ? true : config.distortion;
  config.network.hidden = config.network.hidden || 40;
+ config.network.output = config.network.output || 8;
  config.network.learning_rate = config.network.learning_rate || 0.1;
 
  log('... done');
@@ -171,7 +172,7 @@
 
  chunk = tools.center(chunk);
 
- var output = Array.apply(null, new Array(10)).map(Number.prototype.valueOf, 0);
+ var output = Array.apply(null, new Array(config.network.output)).map(Number.prototype.valueOf, 0);
  output[i] = 1;
 
  if(index < config.training_set) {