| 13 | Pruner::Pruner() { } |
| 14 | |
| 15 | void Pruner::pruneRandomnly(NeuralNet *network) { |
| 16 | int hiddenLayerIndex = rand() % network->numberOfHiddenLayers(); |
| 17 | network->removeNeuron(hiddenLayerIndex, rand() % network->net[hiddenLayerIndex].neurons.size()); |
| 18 | } |
| 19 | |
| 20 | void Pruner::prune(NeuralNet *network, Trainer *networkTrainer) { |
| 21 | prune(network, networkTrainer->getInitialWeights(), networkTrainer->getWeightChanges(), networkTrainer->getGradients()); |
no test coverage detected