Removes unnecessary neurons from a NeuralNet. More...
#include <Pruner.h>
Public Member Functions | |
Pruner () | |
Initialize the Pruner object. More... | |
void | pruneRandomnly (NeuralNet *network) |
void | prune (NeuralNet *network, Trainer *networkTrainer) |
void | prune (NeuralNet *network, const std::vector< std::vector< std::vector< double > > > &initialWeights, const std::vector< std::vector< std::vector< std::vector< double > > > > &weightChanges, const std::vector< std::vector< std::vector< std::vector< double > > > > &gradients) |
Removes unnecessary neurons from a NeuralNet.
Pruner::Pruner | ( | ) |
Initialize the Pruner object.
void Pruner::prune | ( | NeuralNet * | network, |
const std::vector< std::vector< std::vector< double > > > & | initialWeights, | ||
const std::vector< std::vector< std::vector< std::vector< double > > > > & | weightChanges, | ||
const std::vector< std::vector< std::vector< std::vector< double > > > > & | gradients | ||
) |
void Pruner::pruneRandomnly | ( | NeuralNet * | network | ) |