add regularization params to ts

denosaurs · Sep 4, 2024 · 14093db · 14093db
1 parent a201646
commit 14093db
Showing 1 changed file with 30 additions and 0 deletions.
diff --git a/src/core/api/layer.ts b/src/core/api/layer.ts
@@ -31,6 +31,16 @@ export type DenseLayerConfig = {
    * The size of the layer.
    */
   size: Shape1D;
+
+  /**
+   * Inverse of regularization strength.
+   */
+  c?: number;
+
+  /**
+   * Ratio of l1:l2.
+   */
+  l1Ratio?: number;
 };
 
 /**
@@ -86,6 +96,16 @@ export type Conv2DLayerConfig = {
    * The optional strides to use.
    */
   strides?: Shape2D;
+
+  /**
+   * Inverse of regularization strength.
+   */
+  c?: number;
+
+  /**
+   * Ratio of l1:l2.
+   */
+  l1Ratio?: number;
 };
 
 /**
@@ -111,6 +131,16 @@ export type ConvTranspose2DLayerConfig = {
    * The optional strides to use.
    */
   strides?: Shape2D;
+
+  /**
+   * Inverse of regularization strength.
+   */
+  c?: number;
+
+  /**
+   * Ratio of l1:l2.
+   */
+  l1Ratio?: number;
 };
 
 export enum PoolMode {