feat: add throw if insufficient number of points

mljs · May 10, 2024 · d85a924 · d85a924
1 parent c919fa2
commit d85a924
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/PolynomialRegression2D.ts b/src/PolynomialRegression2D.ts
@@ -183,7 +183,11 @@ function train(input: DataXY, y: NumberArray | Matrix, order: number) {
 
   const examples = x.rows;
   const nbCoefficients = ((order + 2) * (order + 1)) / 2;
-
+  if (examples < nbCoefficients) {
+    throw new TypeError(
+      'Insufficient number of points to create regression model.',
+    );
+  }
   const x1 = x.getColumnVector(0);
   const x2 = x.getColumnVector(1);