aifes_express_f32_fnn.h

Go to the documentation of this file.

 
#ifndef EXPRESS_FNN_F32
#define EXPRESS_FNN_F32
 
#include "core/aifes_math.h"
 
typedef enum
{
    AIfES_E_relu,           
    AIfES_E_sigmoid,        
    AIfES_E_softmax,        
    AIfES_E_leaky_relu,     
    AIfES_E_elu,            
    AIfES_E_tanh,           
    AIfES_E_softsign,       
    AIfES_E_linear          
}AIFES_E_activations;
 
typedef enum
{
    AIfES_E_mse,            
    AIfES_E_crossentropy    
}AIFES_E_loss;
 
typedef enum
{
    AIfES_E_adam,           
    AIfES_E_sgd             
}AIFES_E_optimizer;
 
typedef enum
{
    AIfES_E_init_uniform,           
    AIfES_E_init_glorot_uniform,    
    AIfES_E_init_no_init            
}AIFES_E_init_weights_method;
 
typedef enum
{
    AIfES_E_early_stopping_off,
    AIfES_E_early_stopping_on
 
}AIFES_E_early_stopping;
 
struct AIFES_E_model_parameter_fnn_f32{
   uint32_t             layer_count;        
   uint32_t             *fnn_structure;     
   AIFES_E_activations  *fnn_activations;   
   void *               flat_weights;       
};
 
struct AIFES_E_training_parameter_fnn_f32{
   AIFES_E_loss                 loss;   
   AIFES_E_optimizer            optimizer;  
   float                        learn_rate; 
   float                        sgd_momentum; 
   uint32_t                     batch_size; 
   uint32_t                     epochs; 
   uint32_t                     epochs_loss_print_interval; 
   void                         (*loss_print_function)(float);
   AIFES_E_early_stopping       early_stopping; 
   float                        early_stopping_target_loss; 
};
 
struct AIFES_E_init_weights_parameter_fnn_f32{
   AIFES_E_init_weights_method  init_weights_method;    
   float                        min_init_uniform;       
   float                        max_init_uniform;       
};
 
typedef struct AIFES_E_model_parameter_fnn_f32  AIFES_E_model_parameter_fnn_f32;
 
typedef struct AIFES_E_training_parameter_fnn_f32   AIFES_E_training_parameter_fnn_f32;
 
typedef struct AIFES_E_init_weights_parameter_fnn_f32   AIFES_E_init_weights_parameter_fnn_f32;
 
 
uint32_t AIFES_E_flat_weights_number_fnn_f32(uint32_t *fnn_structure,uint32_t layer_count);
 
 
 
int8_t AIFES_E_inference_fnn_f32(aitensor_t *input_tensor,
                                 AIFES_E_model_parameter_fnn_f32 *AIFES_E_fnn,
                                 aitensor_t *output_tensor);
 
 
int8_t AIFES_E_training_fnn_f32(aitensor_t *input_tensor,
                                aitensor_t *target_tensor,
                                AIFES_E_model_parameter_fnn_f32 *AIFES_E_fnn,
                                AIFES_E_training_parameter_fnn_f32 *AIFES_E_fnn_training,
                                AIFES_E_init_weights_parameter_fnn_f32 *AIFES_E_fnn_init_weights,
                                aitensor_t *output_tensor);
 
 
 
#endif // EXPRESS_FNN_F32