Palashio / libra

Ergonomic machine learning for everyone.
http://libradocs.org/
MIT License
1.91k stars 109 forks source link

correcting CNN queries output layer activation function #328

Closed abx393 closed 4 years ago

abx393 commented 4 years ago

This pull request closes #327 and #267 .

If num_classes = 2, sigmoid activation function should be used on the last layer of CNN; otherwise, softmax.

Previously, by default, the softmax activation function was used on the last layer.