Skip to content

Latest commit

 

History

History
31 lines (27 loc) · 1.58 KB

README.md

File metadata and controls

31 lines (27 loc) · 1.58 KB

Hello Tensorspace


Click me to explore the model structure.

Resources

Model Structure

The default view of the model structure is as shown below with corresponding labeled layers.

  const model = new TSP.models.Sequential(modelContainer);
 
  //Order corresponds with the image above default view
  model.add( new TSP.layers.GreyscaleInput({ shape: [28, 28, 1] }) );            //1
  model.add( new TSP.layers.Padding2d({ padding: [2, 2] }) );                    //2
  model.add( new TSP.layers.Conv2d({ kernelSize: 5, filters: 6, strides: 1 }) ); //3
  model.add( new TSP.layers.Pooling2d({ poolSize: [2, 2], strides: [2, 2] }) );  //4
  model.add( new TSP.layers.Conv2d({ kernelSize: 5, filters: 16, strides: 1 }) );//5
  model.add( new TSP.layers.Pooling2d({ poolSize: [2, 2], strides: [2, 2] }) );  //6
  model.add( new TSP.layers.Dense({ units: 120 }) );                             //7
  model.add( new TSP.layers.Dense({ units: 84 }) );                              //8
  model.add( new TSP.layers.Output1d({                                           //9
    units: 10, 
    outputs: ["0", "1", "2", "3", "4", "5", "6", "7", "8", "9"]
  }) );