regress_linear.h

Go to the documentation of this file.

 
#ifndef LIBA_REGRESS_LINEAR_H
#define LIBA_REGRESS_LINEAR_H
 
#include "a.h"

 
/* clang-format off */
#define A_REGRESS_LINEAR_3(coef_p, coef_n, bias) {coef_p, coef_n, a_real_c(bias)}
/* clang-format on */
#define A_REGRESS_LINEAR_2(coef_p, coef_n) A_REGRESS_LINEAR_3(coef_p, coef_n, 0)
#define A_REGRESS_LINEAR_1(coef) A_REGRESS_LINEAR_2(coef, A_LEN(coef))
#define A_REGRESS_LINEAR_0() A_REGRESS_LINEAR_2(A_NULL, 0)
 
typedef struct a_regress_linear a_regress_linear;
 
#if defined(__cplusplus)
extern "C" {
#endif /* __cplusplus */

A_EXTERN void a_regress_linear_init(a_regress_linear *ctx, a_real *coef_p, a_size coef_n, a_real bias);

A_EXTERN a_real a_regress_linear_eval(a_regress_linear const *ctx, a_real const *val);

A_EXTERN void a_regress_linear_err(a_regress_linear const *ctx, a_size n, a_real const *x, a_real const *y, a_real *err);

A_EXTERN void a_regress_linear_pdm(a_regress_linear const *ctx, a_size n, a_real const *x, a_real *pdm, a_real y_mean);

A_EXTERN void a_regress_linear_gd(a_regress_linear *ctx, a_real const *input, a_real error, a_real alpha);

A_EXTERN void a_regress_linear_sgd(a_regress_linear *ctx, a_size n, a_real const *x, a_real const *y, a_real alpha);

A_EXTERN void a_regress_linear_bgd(a_regress_linear *ctx, a_size n, a_real const *x, a_real const *err, a_real alpha);

A_EXTERN a_real a_regress_linear_mgd(a_regress_linear *ctx, a_size n, a_real const *x, a_real const *y, a_real *err,
                                     a_real delta, a_real lrmax, a_real lrmin, a_size lrtim, a_size epoch, a_size batch);

A_EXTERN void a_regress_linear_zero(a_regress_linear *ctx);
 
#if defined(__cplusplus)
} /* extern "C" */
namespace a
{
typedef struct a_regress_linear regress_linear;
} /* namespace a */
#endif /* __cplusplus */

struct a_regress_linear
{
    a_real *coef_p; 
    a_size coef_n; 
    a_real bias; 
#if defined(__cplusplus)
    A_INLINE void init(a_real *p, a_size n, a_real b = 0)
    {
        a_regress_linear_init(this, p, n, b);
    }
    A_INLINE a_real eval(a_real const *val) const
    {
        return a_regress_linear_eval(this, val);
    }
    A_INLINE void err(a_size n, a_real const *x, a_real const *y, a_real *err) const
    {
        a_regress_linear_err(this, n, x, y, err);
    }
    A_INLINE void pdm(a_size n, a_real const *x, a_real *pdm, a_real y_mean) const
    {
        a_regress_linear_pdm(this, n, x, pdm, y_mean);
    }
    A_INLINE void gd(a_real const *input, a_real error, a_real alpha)
    {
        a_regress_linear_gd(this, input, error, alpha);
    }
    A_INLINE void sgd(a_size n, a_real const *x, a_real const *y, a_real alpha)
    {
        a_regress_linear_sgd(this, n, x, y, alpha);
    }
    A_INLINE void bgd(a_size n, a_real const *x, a_real const *err, a_real alpha)
    {
        a_regress_linear_bgd(this, n, x, err, alpha);
    }
    A_INLINE a_real mgd(a_size n, a_real const *x, a_real const *y, a_real *err, a_real delta,
                        a_real lrmax, a_real lrmin, a_size lrtim, a_size epoch, a_size batch)
    {
        return a_regress_linear_mgd(this, n, x, y, err, delta, lrmax, lrmin, lrtim, epoch, batch);
    }
    A_INLINE void zero() { a_regress_linear_zero(this); }
#endif /* __cplusplus */
};

 
#endif /* a/regress_linear.h */