TheArtOfHPC_vol3_cppf08programming/Code/deep/net.cpp

/****************************************************************
 ****
 **** This file belongs with the course
 **** Introduction to Scientific Programming in C++/Fortran2003
 **** copyright 2020 Victor Eijkhout eijkhout@tacc.utexas.edu
 ****
 **** net.cxx : neural net tester
 ****
 ****************************************************************/

#include <iostream>
using std::cin;
using std::cout;

#include <vector>
using std::vector;

#include "ml.hpp"

int main() {

  const int insize = 4, outsize = 2;

  {
    auto layer = network_layer(insize,outsize,0.,1.);
    data_vector
      indata(insize),
      outdata(outsize);
    cout << "Layer with zero coefficients and 1 bias" << '\n';
    layer.apply(indata,outdata,true);
    cout << '\n';
  }

  {
    auto net = network( {insize,outsize} );
    cout << "Network with " << net.depth() << " layers" << '\n';
    cout << '\n';
    data_vector
      indata(insize),
      outdata(outsize);
    net.apply(indata,outdata,true);
  }

  return 0;
}