keras::LayerActivation Class Reference

#include "keras_model.h"

Inheritance diagram for keras::LayerActivation:

Public Member Functions
	LayerActivation ()
void	load_weights (std::ifstream &fin)
keras::DataChunk *	compute_output (keras::DataChunk *)
virtual unsigned int	get_input_rows () const
virtual unsigned int	get_input_cols () const
virtual unsigned int	get_output_units () const
std::string	get_name ()

Public Attributes
std::string	m_activation_type
std::string	m_name

Detailed Description

Definition at line 164 of file keras_model.h.

Constructor & Destructor Documentation

keras::LayerActivation::LayerActivation ( )

inline

Definition at line 166 of file keras_model.h.

References fin.

: Layer("Activation") {}

Member Function Documentation

keras::DataChunk * keras::LayerActivation::compute_output ( keras::DataChunk *  dc )

virtual

Implements keras::Layer.

Definition at line 184 of file keras_model.cc.

References keras::DataChunk::get_1d(), keras::DataChunk::get_3d(), keras::DataChunk::get_data_dim(), keras::missing_activation_impl(), keras::DataChunk::set_data(), and y.

                                                                       {

  if (dc->get_data_dim() == 3) {
    vector<vector<vector<float> > > y = dc->get_3d();
    if(m_activation_type == "relu") {
      for(unsigned int i = 0; i < y.size(); ++i) {
        for(unsigned int j = 0; j < y[0].size(); ++j) {
          for(unsigned int k = 0; k < y[0][0].size(); ++k) {
            if(y[i][j][k] < 0) y[i][j][k] = 0;
          }
        }
      }
    } else if(m_activation_type == "tanh") {
      for(unsigned int i = 0; i < y.size(); ++i) {
        for(unsigned int j = 0; j < y[0].size(); ++j) {
          for(unsigned int k = 0; k < y[0][0].size(); ++k) {
            y[i][j][k] = tanh(y[i][j][k]);
          }
        }
      }
    } else {
      keras::missing_activation_impl(m_activation_type);
    }

    keras::DataChunk *out = new keras::DataChunk2D();
    out->set_data(y);
    return out;

  } else if (dc->get_data_dim() == 1) { // flat data, use 1D
    vector<float> y = dc->get_1d();
    if(m_activation_type == "relu") {
      for(unsigned int k = 0; k < y.size(); ++k) {
        if(y[k] < 0) y[k] = 0;
      }
    } else if(m_activation_type == "softmax") {
      float sum = 0.0;
      for(unsigned int k = 0; k < y.size(); ++k) {
        y[k] = exp(y[k]);
        sum += y[k];
      }
      for(unsigned int k = 0; k < y.size(); ++k) {
        y[k] /= sum;
      }
    } else if(m_activation_type == "tanh") {
      for(unsigned int k = 0; k < y.size(); ++k) {
        y[k] = tanh(y[k]);
      }
    } else if(m_activation_type == "sigmoid") {
      for(unsigned int k = 0; k < y.size(); ++k) {
        y[k] = 1.0F / (1.0F + exp(-y[k]));
      }
    } else {
      keras::missing_activation_impl(m_activation_type);
    }

    keras::DataChunk *out = new DataChunkFlat();
    out->set_data(y);
    return out;

  } else { throw "data dim not supported"; }

  return dc;
}

virtual unsigned int keras::LayerActivation::get_input_cols ( ) const

inlinevirtual

Implements keras::Layer.

Definition at line 171 of file keras_model.h.

{ return 0; } // same as for rows

virtual unsigned int keras::LayerActivation::get_input_rows ( ) const

inlinevirtual

Implements keras::Layer.

Definition at line 170 of file keras_model.h.

{ return 0; } // look for the value in the preceding layer

std::string keras::Layer::get_name ( )

inlineinherited

Definition at line 131 of file keras_model.h.

{ return m_name; }

virtual unsigned int keras::LayerActivation::get_output_units ( ) const

inlinevirtual

Implements keras::Layer.

Definition at line 172 of file keras_model.h.

{ return 0; }

void keras::LayerActivation::load_weights ( std::ifstream &  fin )

virtual

Implements keras::Layer.

Definition at line 84 of file keras_model.cc.

                                                        {
  fin >> m_activation_type;
  cout << "Activation type " << m_activation_type << endl;
}

Member Data Documentation

std::string keras::LayerActivation::m_activation_type

Definition at line 174 of file keras_model.h.

std::string keras::Layer::m_name

inherited

Definition at line 132 of file keras_model.h.

---

The documentation for this class was generated from the following files:

• larreco/v06_64_02/source/larreco/RecoAlg/ImagePatternAlgs/Keras/keras_model.h
• larreco/v06_64_02/source/larreco/RecoAlg/ImagePatternAlgs/Keras/keras_model.cc