changes to cnn defaults for task image classification

ml5js · Nov 4, 2020 · d226e64 · d226e64
1 parent 97a4478
commit d226e64
Showing 1 changed file with 16 additions and 9 deletions.
diff --git a/src/NeuralNetwork/index.js b/src/NeuralNetwork/index.js
@@ -22,6 +22,13 @@ const DEFAULTS = {
 class DiyNeuralNetwork {
  constructor(options, cb) {
  this.callback = cb;
+
+ // Is there a better way to handle a different 
+ // default learning rate for image classification tasks?
+ if (options.task === 'imageClassification') {
+ DEFAULTS.learningRate = 0.02;
+ }
+
  this.options =
  {
  ...DEFAULTS,
@@ -687,29 +694,29 @@ class DiyNeuralNetwork {
  layers = [
  {
  type: 'conv2d',
- filters: 2,
- kernelSize: 2,
- strides: 2,
+ filters: 8,
+ kernelSize: 5,
+ strides: 1,
  activation: 'relu',
  kernelInitializer: 'varianceScaling',
  },
  {
  type: 'maxPooling2d',
- poolSize: [1, 1],
- strides: [1, 1],
+ poolSize: [2, 2],
+ strides: [2, 2],
  },
  {
  type: 'conv2d',
- filters: 1,
- kernelSize: 1,
+ filters: 16,
+ kernelSize: 5,
  strides: 1,
  activation: 'relu',
  kernelInitializer: 'varianceScaling',
  },
  {
  type: 'maxPooling2d',
- poolSize: [1, 1],
- strides: [1, 1],
+ poolSize: [2, 2],
+ strides: [2, 2],
  },
  {
  type: 'flatten',