ml.h

Go to the documentation of this file.

 
#ifndef ML_H
#define ML_H
 
#ifdef __cplusplus
extern "C" {
#endif
 
#include "matrix.h"
#include "vec.h"
 
// Max layers for the fixed-size network
#define ML_MAX_LAYERS 8
 
typedef struct ML_Layer {
    Mat4 weights;     // Weights matrix (4x4)
    Vec4 bias;        // Bias vector (4)
    
    // Cache for backpropagation
    Vec4 input;       // Input to this layer
    Vec4 z;           // Pre-activation output (Wx + b)
    Vec4 output;      // Post-activation output (Sigmoid(z))
    
    // Gradients
    Mat4 d_weights;
    Vec4 d_bias;
} ML_Layer;
 
typedef struct ML_Network {
    ML_Layer layers[ML_MAX_LAYERS];
    int layer_count;
    float learning_rate;
} ML_Network;
 
void ml_init(ML_Network* net, int num_layers, float learning_rate);
 
Vec4 ml_forward(ML_Network* net, Vec4 input);
 
float ml_train_step(ML_Network* net, Vec4 input, Vec4 target);
 
#ifdef __cplusplus
}
#endif
 
#endif // ML_H