Different activation function to use with Convolutional Neural Network
2 次查看(过去 30 天)
显示 更早的评论
Hi there,
I am trying to use "Step Function" as the activation function for one of the convolutional layers in the CNN, though the only option Matlab provides is ReLU Layer, how can I add a different activation function?
Also what if the output of the CNN is another image not a class! What should I use as the final layer: Classification Layer or Regression Layer?
1 个评论
Karthiga Mahalingam
2018-7-9
You can create custom layers for neural network models in matlab. This link has examples for creating custom deep learning, classification and regression layers:
https://www.mathworks.com/help/nnet/ug/define-custom-deep-learning-layers.html
The Deep learning layer example explains how to construct a PReLU layer. You could modify it accordingly for a step function.
To use a classification or a regression layer as the final layer would depend on your application. The above link also has examples to creating custom classification and regression layers.
回答(0 个)
另请参阅
类别
在 Help Center 和 File Exchange 中查找有关 Get Started with Deep Learning Toolbox 的更多信息
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!