tessem.cc

Go to the documentation of this file.

 
#include "tessem.h"
#include "file.h"
#include "matpack_data.h"
#include "mystring.h"
 
void tessem_read_ascii(std::ifstream& is, TessemNN& net) {
  is >> net.nb_inputs >> net.nb_cache >> net.nb_outputs;
 
  net.b1.resize(net.nb_cache);
  for (Index i = 0; i < net.nb_cache; i++) is >> double_imanip() >> net.b1[i];
 
  net.b2.resize(net.nb_outputs);
  for (Index i = 0; i < net.nb_outputs; i++) is >> double_imanip() >> net.b2[i];
 
  net.w1.resize(net.nb_cache, net.nb_inputs);
  for (Index i = 0; i < net.nb_cache; i++)
    for (Index j = 0; j < net.nb_inputs; j++) is >> double_imanip() >> net.w1(i, j);
 
  net.w2.resize(net.nb_outputs, net.nb_cache);
  for (Index i = 0; i < net.nb_outputs; i++)
    for (Index j = 0; j < net.nb_cache; j++) is >> double_imanip() >> net.w2(i, j);
 
  net.x_min.resize(net.nb_inputs);
  for (Index i = 0; i < net.nb_inputs; i++) is >> double_imanip() >> net.x_min[i];
 
  net.x_max.resize(net.nb_inputs);
  for (Index i = 0; i < net.nb_inputs; i++) is >> double_imanip() >> net.x_max[i];
 
  net.y_min.resize(net.nb_outputs);
  for (Index i = 0; i < net.nb_outputs; i++) is >> double_imanip() >> net.y_min[i];
 
  net.y_max.resize(net.nb_outputs);
  for (Index i = 0; i < net.nb_outputs; i++) is >> double_imanip() >> net.y_max[i];
}
 
void tessem_prop_nn(VectorView ny, const TessemNN& net, ConstVectorView nx) {
  ARTS_USER_ERROR_IF (nx.nelem() != net.nb_inputs,
    "Tessem NN requires ", net.nb_inputs,
    " values, but input vector has ", nx.nelem(), " element.")
 
  ARTS_USER_ERROR_IF (ny.nelem() != net.nb_outputs,
    "Tessem NN generates ", net.nb_outputs,
    " values, but output vector has ", ny.nelem(), " element.")
 
  // preprocessing
  Vector new_x(nx);
  new_x[0] *= 1e-9;
  new_x[4] *= 1e3;
  for (Index i = 0; i < net.nb_inputs; i++)
    new_x[i] =
        -1. + (new_x[i] - net.x_min[i]) / (net.x_max[i] - net.x_min[i]) * 2;
 
  // propagation
  Vector trans = net.b1;
  for (Index i = 0; i < net.nb_cache; i++) {
    for (Index j = 0; j < net.nb_inputs; j++)
      trans[i] += net.w1(i, j) * new_x[j];
 
    trans[i] = 2. / (1. + exp(-2. * trans[i])) - 1.;
  }
 
  Vector new_y = net.b2;
  for (Index i = 0; i < net.nb_outputs; i++) {
    for (Index j = 0; j < net.nb_cache; j++)
      new_y[i] += net.w2(i, j) * trans[j];
  }
 
  // postprocessing
  for (Index i = 0; i < net.nb_outputs; i++)
    ny[i] = net.y_min[i] + (new_y[i] + 1.) / 2. * (net.y_max[i] - net.y_min[i]);
}