mirror of
https://github.com/UpsilonNumworks/Upsilon.git
synced 2026-01-19 16:57:31 +01:00
a8ead6b66e
It provides a better fit for:
x = {0.0, 1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0}
y = {5.0, 9.0, 40.0, 64.0, 144.0, 200.0, 269.0, 278.0, 290.0, 295.0}
(coeffs should be {64.9, 1.0, 297.4})
98 lines
3.4 KiB
C++
98 lines
3.4 KiB
C++
#include "logistic_model.h"
|
|
#include "../store.h"
|
|
#include <math.h>
|
|
#include <assert.h>
|
|
#include <poincare/code_point_layout.h>
|
|
#include <poincare/fraction_layout.h>
|
|
#include <poincare/horizontal_layout.h>
|
|
#include <poincare/vertical_offset_layout.h>
|
|
|
|
using namespace Poincare;
|
|
|
|
namespace Regression {
|
|
|
|
Layout LogisticModel::layout() {
|
|
if (m_layout.isUninitialized()) {
|
|
constexpr int exponentSize = 4;
|
|
Layout exponentLayoutChildren[exponentSize] = {
|
|
CodePointLayout::Builder('-', k_layoutFont),
|
|
CodePointLayout::Builder('b', k_layoutFont),
|
|
CodePointLayout::Builder(UCodePointMiddleDot, k_layoutFont),
|
|
CodePointLayout::Builder('X', k_layoutFont)
|
|
};
|
|
constexpr int denominatorSize = 6;
|
|
Layout layoutChildren[denominatorSize] = {
|
|
CodePointLayout::Builder('1', k_layoutFont),
|
|
CodePointLayout::Builder('+', k_layoutFont),
|
|
CodePointLayout::Builder('a', k_layoutFont),
|
|
CodePointLayout::Builder(UCodePointMiddleDot, k_layoutFont),
|
|
CodePointLayout::Builder('e', k_layoutFont),
|
|
VerticalOffsetLayout::Builder(
|
|
HorizontalLayout::Builder(exponentLayoutChildren, exponentSize),
|
|
VerticalOffsetLayoutNode::Position::Superscript
|
|
)
|
|
};
|
|
m_layout = FractionLayout::Builder(
|
|
CodePointLayout::Builder('c', k_layoutFont),
|
|
HorizontalLayout::Builder(layoutChildren, denominatorSize)
|
|
);
|
|
}
|
|
return m_layout;
|
|
}
|
|
|
|
double LogisticModel::evaluate(double * modelCoefficients, double x) const {
|
|
double a = modelCoefficients[0];
|
|
double b = modelCoefficients[1];
|
|
double c = modelCoefficients[2];
|
|
return c/(1.0+a*exp(-b*x));
|
|
}
|
|
|
|
double LogisticModel::levelSet(double * modelCoefficients, double xMin, double step, double xMax, double y, Poincare::Context * context) {
|
|
double a = modelCoefficients[0];
|
|
double b = modelCoefficients[1];
|
|
double c = modelCoefficients[2];
|
|
if (a == 0 || b == 0 || c == 0 || y == 0) {
|
|
return NAN;
|
|
}
|
|
double lnArgument = (c/y - 1)/a;
|
|
if (lnArgument <= 0) {
|
|
return NAN;
|
|
}
|
|
return -log(lnArgument)/b;
|
|
}
|
|
|
|
double LogisticModel::partialDerivate(double * modelCoefficients, int derivateCoefficientIndex, double x) const {
|
|
double a = modelCoefficients[0];
|
|
double b = modelCoefficients[1];
|
|
double c = modelCoefficients[2];
|
|
double denominator = 1.0+a*exp(-b*x);
|
|
if (derivateCoefficientIndex == 0) {
|
|
// Derivate: exp(-b*x)*(-1 * c/(1.0+a*exp(-b*x))^2)
|
|
return -exp(-b*x) * c/(denominator * denominator);
|
|
}
|
|
if (derivateCoefficientIndex == 1) {
|
|
// Derivate: (-x)*a*exp(-b*x)*(-1/(1.0+a*exp(-b*x))^2)
|
|
return x*a*exp(-b*x)*c/(denominator * denominator);
|
|
}
|
|
if (derivateCoefficientIndex == 2) {
|
|
// Derivate: (-x)*a*exp(-b*x)*(-1/(1.0+a*exp(-b*x))^2)
|
|
return 1.0/denominator;
|
|
}
|
|
assert(false);
|
|
return 0.0;
|
|
}
|
|
|
|
void LogisticModel::specializedInitCoefficientsForFit(double * modelCoefficients, double defaultValue, Store * store, int series) const {
|
|
assert(store != nullptr && series >= 0 && series < Store::k_numberOfSeries && !store->seriesIsEmpty(series));
|
|
modelCoefficients[0] = defaultValue;
|
|
modelCoefficients[1] = defaultValue;
|
|
/* If the data is a standard logistic function, the ordinates are between 0
|
|
* and c. Twice the standard vertical deviation is a rough estimate of c
|
|
* that is "close enough" to c to seed the coefficient, without being too
|
|
* dependent on outliers.*/
|
|
modelCoefficients[2] = 2.0*store->standardDeviationOfColumn(series, 1);
|
|
}
|
|
|
|
|
|
}
|