Added sigmoid activation, and ability to do activation above linear layers.

This commit is contained in:
Alec Helbling
2023-01-25 17:36:44 -05:00
parent 4948c0ea4e
commit 301b230c73
7 changed files with 57 additions and 10 deletions

View File

@ -68,6 +68,11 @@ class Convolutional2DLayer(VGroupNeuralNetworkLayer, ThreeDLayer):
about_point=self.get_center(),
axis=ThreeDLayer.rotation_axis,
)
self.construct_activation_function()
def construct_activation_function(self):
"""Construct the activation function"""
# Add the activation function
if not self.activation_function is None:
# Check if it is a string