@thi.ng/math 5.13.6 → 5.14.0

package/README.md

@@ -7,7 +7,7 @@
 [![Mastodon Follow](https://img.shields.io/mastodon/follow/109331703950160316?domain=https%3A%2F%2Fmastodon.thi.ng&style=social)](https://mastodon.thi.ng/@toxi)
 
 > [!NOTE]
-> This is one of 210 standalone projects, maintained as part
+> This is one of 211 standalone projects, maintained as part
 > of the [@thi.ng/umbrella](https://github.com/thi-ng/umbrella/) monorepo
 > and anti-framework.
 >
@@ -77,7 +77,7 @@ For Node.js REPL:
 const math = await import("@thi.ng/math");
 ```
 
-Package sizes (brotli'd, pre-treeshake): ESM: 5.04 KB
+Package sizes (brotli'd, pre-treeshake): ESM: 5.29 KB
 
 ## Dependencies
 

package/index.d.ts

@@ -12,6 +12,7 @@ export * from "./libc.js";
 export * from "./min-error.js";
 export * from "./mix.js";
 export * from "./permutations.js";
+export * from "./polynomial.js";
 export * from "./prec.js";
 export * from "./prime.js";
 export * from "./ratio.js";

package/index.js

@@ -12,6 +12,7 @@ export * from "./libc.js";
 export * from "./min-error.js";
 export * from "./mix.js";
 export * from "./permutations.js";
+export * from "./polynomial.js";
 export * from "./prec.js";
 export * from "./prime.js";
 export * from "./ratio.js";

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@thi.ng/math",
-	"version": "5.13.6",
+	"version": "5.14.0",
 	"description": "Assorted common math functions & utilities",
 	"type": "module",
 	"module": "./index.js",
@@ -42,7 +42,7 @@
 		"tool:tangle": "../../node_modules/.bin/tangle src/**/*.ts"
 	},
 	"dependencies": {
-		"@thi.ng/api": "^8.12.9"
+		"@thi.ng/api": "^8.12.10"
 	},
 	"devDependencies": {
 		"esbuild": "^0.27.0",
@@ -122,6 +122,9 @@
 		"./permutations": {
 			"default": "./permutations.js"
 		},
+		"./polynomial": {
+			"default": "./polynomial.js"
+		},
 		"./prec": {
 			"default": "./prec.js"
 		},
@@ -144,5 +147,5 @@
 	"thi.ng": {
 		"year": 2013
 	},
-	"gitHead": "fdca77cabf47dd23a9ab17a5ca13e3060075c12c\n"
+	"gitHead": "824bf9047b5a10f777c5c5b4aeecf0c750a22c75\n"
 }

package/polynomial.d.ts

@@ -0,0 +1,49 @@
+/**
+ * Computes polynomial for `x` and given `coeffs` (in order of increasing
+ * exponents).
+ *
+ * @remarks
+ * See {@link polynomialRegression} for computing coefficients.
+ *
+ * The number of given coefficients defines the degree (+1) of the polynomial,
+ * i.e. a cubic function will require 4 coeffs, with the y-intercept being the
+ * first coeff.
+ *
+ * @example
+ * ```ts tangle:../export/polynomial.ts
+ * import { polynomial } from "@thi.ng/math";
+ *
+ * const coeffs = [-5, -4, 3, 2];
+ *
+ * for(let x = -2; x <= 2; x += 0.5) {
+ *   console.log(`f(${x}) = ${polynomial(x, coeffs)}`);
+ * }
+ * // f(-2) = -1
+ * // f(-1.5) = 1
+ * // f(-1) = 0
+ * // f(-0.5) = -2.5
+ * // f(0) = -5
+ * // f(0.5) = -6
+ * // f(1) = -4
+ * // f(1.5) = 2.5
+ * // f(2) = 15
+ * ```
+ *
+ * @param x
+ * @param coeffs
+ */
+export declare const polynomial: (x: number, coeffs: number[]) => number;
+/**
+ * Computes the coefficients of a polynomial regression for the given `samples`
+ * (each a [x,y] tuple). The `degree` param defines the degree of the polynomial
+ * and the number of returned coefficients (+1).
+ *
+ * @remarks
+ * The resulting coeffs can be then used with {@link polynomial} to evaluate the
+ * curve (i.e. used to make predictions).
+ *
+ * @param samples
+ * @param degree
+ */
+export declare const polynomialRegression: (samples: number[][], degree: number) => number[];
+//# sourceMappingURL=polynomial.d.ts.map

package/polynomial.js

@@ -0,0 +1,20 @@
+import { gaussianElimination } from "./solve.js";
+const polynomial = (x, coeffs) => coeffs.reduce((sum, c, i) => sum + c * x ** i, 0);
+const polynomialRegression = (samples, degree) => {
+  const mat = [];
+  const aug = [];
+  for (let i = 0; i <= degree; i++) {
+    const col = [];
+    for (let j = 0; j <= degree; j++) {
+      col.push(samples.reduce((acc, x) => acc + x[0] ** (i + j), 0));
+    }
+    mat.push(col);
+    aug.push(samples.reduce((acc, x) => acc + x[0] ** i * x[1], 0));
+  }
+  mat.push(aug);
+  return gaussianElimination(mat, degree + 1);
+};
+export {
+  polynomial,
+  polynomialRegression
+};

package/regression.d.ts

@@ -0,0 +1,43 @@
+/**
+ * Computes the coefficients of a polynomial regression for the given `samples`
+ * (each a [x,y] tuple). The `degree` param defines the degree of the polynomial
+ * and the number of returned coefficients (+1).
+ *
+ * @remarks
+ * The resulting coeffs can be then used with {@link polynomial} to evaluate the
+ * curve (i.e. used to make predictions).
+ *
+ * @param samples
+ * @param degree
+ */
+export declare const polynomialRegression: (samples: number[][], degree: number) => number[];
+/**
+ * Computes polynomial for `x` and given `coeffs` (in order of increasing
+ * exponents).
+ *
+ * @remarks
+ * See {@link polynomialRegression} for computing coefficients.
+ *
+ * The number of given coefficients defines the degree (+1) of the polynomial,
+ * i.e. a cubic function will require 4 coeffs, with the y-intercept being the
+ * first coeff.
+ *
+ * @param x
+ * @param coeffs
+ */
+export declare const polynomial: (x: number, coeffs: number[]) => number;
+/**
+ * Takes an augmented matrix (in column-major order) and uses Gaussian
+ * elimination to compute solution coefficients for its system of linear
+ * equations `Ax = b`.
+ *
+ * @remarks
+ * References:
+ * - https://en.wikipedia.org/wiki/Gaussian_elimination
+ * - https://www.geeksforgeeks.org/dsa/gaussian-elimination/
+ *
+ * @param mat
+ * @param degree
+ */
+export declare const gaussianElimination: (mat: number[][], degree: number) => number[];
+//# sourceMappingURL=regression.d.ts.map

package/regression.js

@@ -0,0 +1,51 @@
+const polynomialRegression = (samples, degree) => {
+  const mat = [];
+  const aug = [];
+  for (let i = 0; i <= degree; i++) {
+    aug.push(samples.reduce((acc, x) => acc + x[0] ** i * x[1], 0));
+    const col = [];
+    for (let j = 0; j <= degree; j++) {
+      col.push(samples.reduce((acc, x) => acc + x[0] ** (i + j), 0));
+    }
+    mat.push(col);
+  }
+  mat.push(aug);
+  return gaussianElimination(mat, degree + 1);
+};
+const polynomial = (x, coeffs) => coeffs.reduce((sum, c, i) => sum + c * x ** i, 0);
+const gaussianElimination = (mat, degree) => {
+  const n = mat.length - 1;
+  const coeffs = [degree];
+  for (let i = 0; i < n; i++) {
+    let max = i;
+    const col = mat[i];
+    for (let j = i + 1; j < n; j++) {
+      if (Math.abs(col[j]) > Math.abs(col[max])) max = j;
+    }
+    for (let k = i; k <= n; k++) {
+      const col2 = mat[k];
+      const tmp = col2[i];
+      col2[i] = col2[max];
+      col2[max] = tmp;
+    }
+    for (let j = i + 1; j < n; j++) {
+      for (let k = n; k >= i; k--) {
+        mat[k][j] -= mat[k][i] * mat[i][j] / mat[i][i];
+      }
+    }
+  }
+  for (let j = n; j-- > 0; ) {
+    const d = mat[j][j];
+    let sum = 0;
+    for (let k = j + 1; k < n; k++) {
+      sum += mat[k][j] * coeffs[k];
+    }
+    coeffs[j] = (mat[n][j] - sum) / d;
+  }
+  return coeffs;
+};
+export {
+  gaussianElimination,
+  polynomial,
+  polynomialRegression
+};

package/solve.d.ts

@@ -75,4 +75,34 @@ export declare const solveCubic: (a: number, b: number, c: number, d: number, ep
  * @param d - input coefficients & output solutions [0,N-1]
  */
 export declare const solveTridiagonal: (a: NumericArray, b: NumericArray, c: NumericArray, d: NumericArray) => NumericArray;
+/**
+ * Takes an augmented matrix (in column-major order) and uses Gaussian
+ * elimination to compute solution coefficients for its system of linear
+ * equations `Ax = b`.
+ *
+ * @remarks
+ * References:
+ * - https://en.wikipedia.org/wiki/Gaussian_elimination
+ * - https://www.geeksforgeeks.org/dsa/gaussian-elimination/
+ *
+ * @example
+ * ```ts tangle:../export/gaussian-elimination.ts
+ * import { gaussianElimination } from "@thi.ng/math";
+ *
+ * // in column-major order
+ * const matrix = [
+ *   [3, 2, 5],
+ *   [2, 3, -3],
+ *   [-4, 3, 1],
+ *   [3, 15, 14]
+ * ];
+ *
+ * console.log(gaussianElimination(matrix, 1));
+ * // [3, 1, 2] (rounded)
+ * ```
+ *
+ * @param mat
+ * @param degree
+ */
+export declare const gaussianElimination: (mat: number[][], degree: number) => number[];
 //# sourceMappingURL=solve.d.ts.map

package/solve.js

@@ -51,8 +51,40 @@ const solveTridiagonal = (a, b, c, d) => {
   }
   return d;
 };
+const gaussianElimination = (mat, degree) => {
+  const n = mat.length - 1;
+  const coeffs = [degree];
+  for (let i = 0; i < n; i++) {
+    let max = i;
+    const col = mat[i];
+    for (let j = i + 1; j < n; j++) {
+      if (Math.abs(col[j]) > Math.abs(col[max])) max = j;
+    }
+    for (let k = i; k <= n; k++) {
+      const col2 = mat[k];
+      const tmp = col2[i];
+      col2[i] = col2[max];
+      col2[max] = tmp;
+    }
+    for (let j = i + 1; j < n; j++) {
+      for (let k = n; k >= i; k--) {
+        mat[k][j] -= mat[k][i] * mat[i][j] / mat[i][i];
+      }
+    }
+  }
+  for (let j = n; j-- > 0; ) {
+    const d = mat[j][j];
+    let sum = 0;
+    for (let k = j + 1; k < n; k++) {
+      sum += mat[k][j] * coeffs[k];
+    }
+    coeffs[j] = (mat[n][j] - sum) / d;
+  }
+  return coeffs;
+};
 export {
   derivative,
+  gaussianElimination,
   solveCubic,
   solveLinear,
   solveQuadratic,