GstTensorTrainer's properties for neural network framework (internal data structure) More...
#include <nnstreamer_plugin_api_trainer.h>
Public Attributes | |
GstTensorsInfo | input_meta |
const char * | model_config |
const char * | model_save_path |
const char * | model_load_path |
unsigned int | num_inputs |
unsigned int | num_labels |
unsigned int | num_training_samples |
unsigned int | num_validation_samples |
unsigned int | num_epochs |
unsigned int | epoch_count |
double | training_loss |
double | training_accuracy |
double | validation_loss |
double | validation_accuracy |
GstTensorTrainer's properties for neural network framework (internal data structure)
Internal data of GstTensorTrainer required by tensor_trainer's custom subplugin.
Definition at line 31 of file nnstreamer_plugin_api_trainer.h.
unsigned int _GstTensorTrainerProperties::epoch_count |
Number of currently completed epochs
Definition at line 43 of file nnstreamer_plugin_api_trainer.h.
GstTensorsInfo _GstTensorTrainerProperties::input_meta |
configured input tensor info
Definition at line 33 of file nnstreamer_plugin_api_trainer.h.
const char* _GstTensorTrainerProperties::model_config |
The configuration file path for creating model
Definition at line 34 of file nnstreamer_plugin_api_trainer.h.
const char* _GstTensorTrainerProperties::model_load_path |
The file path to load an existing model to use for training a new model
Definition at line 36 of file nnstreamer_plugin_api_trainer.h.
const char* _GstTensorTrainerProperties::model_save_path |
The file path to save the new model
Definition at line 35 of file nnstreamer_plugin_api_trainer.h.
unsigned int _GstTensorTrainerProperties::num_epochs |
The number of repetition of total training and validation sample. subplugin must receive total samples((num_training_samples + num_validation_samples) * num_epochs)
Definition at line 41 of file nnstreamer_plugin_api_trainer.h.
unsigned int _GstTensorTrainerProperties::num_inputs |
The number of input lists, the input is where framework receive the features to train the model, num_inputs indicates how many inputs there are.
Definition at line 37 of file nnstreamer_plugin_api_trainer.h.
unsigned int _GstTensorTrainerProperties::num_labels |
The number of label lists, the label is where framework receive the class to train the model, num_labels indicates how many labels there are.
Definition at line 38 of file nnstreamer_plugin_api_trainer.h.
unsigned int _GstTensorTrainerProperties::num_training_samples |
The number of training sample used to train the model.
Definition at line 39 of file nnstreamer_plugin_api_trainer.h.
unsigned int _GstTensorTrainerProperties::num_validation_samples |
The number of validation sample used to valid the model.
Definition at line 40 of file nnstreamer_plugin_api_trainer.h.
double _GstTensorTrainerProperties::training_accuracy |
Training accuracy of the model being trained in the subplugin
Definition at line 45 of file nnstreamer_plugin_api_trainer.h.
double _GstTensorTrainerProperties::training_loss |
Training loss of the model being trained in the subplugin
Definition at line 44 of file nnstreamer_plugin_api_trainer.h.
double _GstTensorTrainerProperties::validation_accuracy |
Validation accuracy of the model being trained in the subplugin
Definition at line 47 of file nnstreamer_plugin_api_trainer.h.
double _GstTensorTrainerProperties::validation_loss |
Validation loss of the model being trained in the subplugin
Definition at line 46 of file nnstreamer_plugin_api_trainer.h.