-module(gleastsq).
-compile([no_auto_import, nowarn_unused_vars, nowarn_unused_function, nowarn_nomatch]).

-export([levenberg_marquardt/10, least_squares/8]).

-spec levenberg_marquardt(
    list(float()),
    list(float()),
    fun((float(), list(float())) -> float()),
    list(float()),
    gleam@option:option(integer()),
    gleam@option:option(float()),
    gleam@option:option(float()),
    gleam@option:option(float()),
    gleam@option:option(float()),
    gleam@option:option(float())
) -> {ok, list(float())} | {error, gleastsq@errors:fit_errors()}.
levenberg_marquardt(
    X,
    Y,
    Func,
    Initial_params,
    Iterations,
    Epsilon,
    Tolerance,
    Damping,
    Damping_increase,
    Damping_decrease
) ->
    gleastsq@levenberg_marquardt:levenberg_marquardt(
        X,
        Y,
        Func,
        Initial_params,
        Iterations,
        Epsilon,
        Tolerance,
        Damping,
        Damping_increase,
        Damping_decrease
    ).

-spec least_squares(
    list(float()),
    list(float()),
    fun((float(), list(float())) -> float()),
    list(float()),
    gleam@option:option(integer()),
    gleam@option:option(float()),
    gleam@option:option(float()),
    gleam@option:option(float())
) -> {ok, list(float())} | {error, gleastsq@errors:fit_errors()}.
least_squares(
    X,
    Y,
    Func,
    Initial_params,
    Iterations,
    Epsilon,
    Tolerance,
    Lambda_reg
) ->
    gleastsq@least_squares:least_squares(
        X,
        Y,
        Func,
        Initial_params,
        Iterations,
        Epsilon,
        Tolerance,
        Lambda_reg
    ).