Update gobable activations

samuelfneumann · Sep 24, 2021 · 5b8842c · 5b8842c
1 parent 8815fa3
commit 5b8842c
Showing 1 changed file with 25 additions and 1 deletion.
diff --git a/network/Activations.go b/network/Activations.go
@@ -108,12 +108,36 @@ func (a *Activation) GobDecode(encoded []byte) error {
 	switch decoded {
 	case relu:
 		*a = *ReLU()
+
 	case identity:
 		*a = *Identity()
+
 	case tanh:
 		*a = *TanH()
+
+	case sigmoid:
+		*a = *Sigmoid()
+
+	case sin:
+		*a = *Sin()
+
+	case cos:
+		*a = *Cos()
+
+	case sqrt:
+		*a = *Sqrt()
+
+	case mish:
+		*a = *Mish()
+
+	case logarithm:
+		*a = *Log()
+
+	case softplus:
+		*a = *Softplus()
+
 	default:
-		return fmt.Errorf("gobdecode: illegal Activation type")
+		return fmt.Errorf("unmarshalJSON: illegal Activation type")
 	}
 	return nil
 }