#include "gnn_momentum.h"Include dependency graph for gnn_momentum.c:

Go to the source code of this file.
Data Structures | |
| struct | _gnn_momentum |
Typedefs | |
| typedef _gnn_momentum | gnn_momentum |
Functions | |
| int | gnn_momentum_train (gnn_trainer *trainer) |
| Train function. | |
| void | gnn_momentum_destroy (gnn_trainer *trainer) |
| Destructor. | |
| int | gnn_momentum_reset (gnn_trainer *trainer) |
| Reset function. | |
| gnn_trainer * | gnn_momentum_new (gnn_node *node, gnn_criterion *crit, gnn_dataset *data, double mu, double eta) |
| Creates a new gradient descent with momentum trainer. | |
| double | gnn_momentum_get_mu (gnn_trainer *trainer) |
| Gets the learning rate. | |
| int | gnn_momentum_set_mu (gnn_trainer *trainer, double mu) |
| Sets the learning rate. | |
| double | gnn_momentum_get_eta (gnn_trainer *trainer) |
| Gets the momentum rate. | |
| int | gnn_momentum_set_eta (gnn_trainer *trainer, double eta) |
| Sets the momentum rate. | |
|
|
Definition at line 57 of file gnn_momentum.c. |
1.2.18