---

gnn_trainer.c File Reference

#include <string.h>
#include "gnn_trainer.h"
#include "gnn_utilities.h"

Include dependency graph for gnn_trainer.c:

Go to the source code of this file.

Functions
int	gnn_trainer_default_reset (gnn_trainer *trainer)
	Default "reset" function for a trainer.
void	gnn_trainer_default_destroy (gnn_trainer *trainer)
	Default "destroy" function for a trainer.
const char *	gnn_trainer_get_type (gnn_trainer *trainer)
	Get the trainer's type.
int	gnn_trainer_init (gnn_trainer *trainer, const char *type, gnn_node *node, gnn_criterion *crit, gnn_dataset *data, gnn_trainer_reset_type reset, gnn_trainer_train_type train, gnn_trainer_destroy_type destroy)
	Initializes a gnn_trainer : Trainers for Models. structure.
void	gnn_trainer_destroy (gnn_trainer *trainer)
	Destroys the trainer.
int	gnn_trainer_reset (gnn_trainer *trainer)
	Resets the trainer.
double	gnn_trainer_train (gnn_trainer *trainer)
	Trains the node.
gnn_dataset *	gnn_trainer_get_dataset (gnn_trainer *trainer)
	Get the dataset.
gnn_node *	gnn_trainer_get_node (gnn_trainer *trainer)
	Get the model.
gnn_criterion *	gnn_trainer_get_criterion (gnn_trainer *trainer)
	Get the criterion.
int	gnn_trainer_batch_set_size (gnn_trainer *trainer, size_t size)
	Set the size for minibatches.
size_t	gnn_trainer_batch_get_size (gnn_trainer *trainer)
	Get the size of the minibatches.
size_t	gnn_trainer_get_pattern_index (gnn_trainer *trainer)
	Returns the index of the next minibatch.
size_t	gnn_trainer_get_epoch (gnn_trainer *trainer)
	Get the number of the current epoch.
double	gnn_trainer_get_epoch_cost (gnn_trainer *trainer)
	Get the mean cost.
gnn_grad *	gnn_trainer_batch_get_grad (gnn_trainer *trainer)
	Returns a pointer to the internal Evaluations. buffer.
double	gnn_trainer_batch_get_e (gnn_trainer *trainer)
	Returns the last evaluated batch's mean error.
gsl_vector *	gnn_trainer_batch_get_dx (gnn_trainer *trainer)
	Returns the last batch's evaluated gradient dx.
gsl_vector *	gnn_trainer_batch_get_dw (gnn_trainer *trainer)
	Returns the last batch's evaluated gradient dw.
int	gnn_trainer_batch_process (gnn_trainer *trainer)
	Processes a batch.
int	gnn_trainer_batch_next (gnn_trainer *trainer)
	Moves onto the next batch.

---