libeblearn
|
#include <ebl_arch.h>
Public Member Functions | |
ebm_module_1_1 (module_1_1< T, Tin, Tout > *m, ebm_1< T, Ten > *e, const char *name="ebm_module_1_1") | |
virtual void | fprop (Tin &in, Tout &out) |
virtual void | bprop (Tin &in, Tout &out) |
virtual void | bbprop (Tin &in, Tout &out) |
virtual void | forget (forget_param_linear &fp) |
virtual Ten & | get_energy () |
Returns a reference to the energy output of fprop throught the ebm_1. | |
virtual fidxdim | fprop_size (fidxdim &isize) |
virtual fidxdim | bprop_size (const fidxdim &osize) |
virtual std::string | describe () |
Returns a string describing this module and its parameters. | |
Protected Attributes | |
module_1_1< T, Tin, Tout > * | module |
ebm_1< T, Ten > * | ebm |
Ten | energy |
A module containing both a module_1_1 that takes 1 input and produces 1 output but that also an ebm_1 that produces an energy given module_1_1's output.
ebl::ebm_module_1_1< T, Tin, Tout, Ten >::ebm_module_1_1 | ( | module_1_1< T, Tin, Tout > * | m, |
ebm_1< T, Ten > * | e, | ||
const char * | name = "ebm_module_1_1< T, Tin, Tout, Ten >" |
||
) |
Construct with module 'm' and ebm 'e'. This module is responsible for deleting 'm' and 'e' at destruction.
fidxdim ebl::ebm_module_1_1< T, Tin, Tout, Ten >::bprop_size | ( | const fidxdim & | osize | ) | [virtual] |
Returns input dimensions corresponding to output dimensions 'osize'. Implementation of this method helps automatic scaling of input data but is optional.
Reimplemented from ebl::module_1_1< T, Tin, Tout >.
fidxdim ebl::ebm_module_1_1< T, Tin, Tout, Ten >::fprop_size | ( | fidxdim & | isize | ) | [virtual] |
Modifies input dimensions 'isize' to be compliant with module's architecture, and returns corresponding output dimensions. Implementation of this method helps automatic scaling of input data but is optional.
Reimplemented from ebl::module_1_1< T, Tin, Tout >.