Test Dense layer export

Kotlin · Apr 24, 2022 · 0491460 · 0491460
1 parent 7d1d372
commit 0491460
Showing 1 changed file with 27 additions and 1 deletion.
diff --git a/...rence/keras/InputLayerImportExportTest.kt → ...rence/keras/CoreLayersImportExportTest.kt b/...rence/keras/InputLayerImportExportTest.kt → ...rence/keras/CoreLayersImportExportTest.kt
@@ -7,10 +7,16 @@ package org.jetbrains.kotlinx.dl.api.inference.keras
 
 import org.jetbrains.kotlinx.dl.api.core.Functional
 import org.jetbrains.kotlinx.dl.api.core.Sequential
+import org.jetbrains.kotlinx.dl.api.core.activation.Activations
+import org.jetbrains.kotlinx.dl.api.core.initializer.HeNormal
+import org.jetbrains.kotlinx.dl.api.core.initializer.HeUniform
+import org.jetbrains.kotlinx.dl.api.core.layer.core.Dense
 import org.jetbrains.kotlinx.dl.api.core.layer.core.Input
+import org.jetbrains.kotlinx.dl.api.core.regularizer.L2
+import org.jetbrains.kotlinx.dl.api.core.regularizer.L2L1
 import org.junit.jupiter.api.Test
 
-class InputLayerImportExportTest {
+class CoreLayersImportExportTest {
     @Test
     fun inputLayerSequential() {
         LayerImportExportTest.run(Sequential.of(Input(4)))
@@ -26,4 +32,24 @@ class InputLayerImportExportTest {
         LayerImportExportTest.run(Functional.of(Input(128, 128, 3)))
         LayerImportExportTest.run(Functional.of(Input(10, 10, 10, 10)))
     }
+
+    @Test
+    fun denseLayer() {
+        LayerImportExportTest.run(
+            Sequential.of(
+                Input(10),
+                Dense(
+                    name = "test_dense",
+                    outputSize = 10,
+                    activation = Activations.Tanh,
+                    kernelInitializer = HeNormal(),
+                    biasInitializer = HeUniform(),
+                    kernelRegularizer = L2(),
+                    biasRegularizer = L2(),
+                    activityRegularizer = L2L1(),
+                    useBias = true
+                )
+            )
+        )
+    }
 }