diff --git a/src/main/java/org/dromara/easyai/function/Sigmoid.java b/src/main/java/org/dromara/easyai/function/Sigmoid.java new file mode 100644 index 0000000..ff6f51f --- /dev/null +++ b/src/main/java/org/dromara/easyai/function/Sigmoid.java @@ -0,0 +1,22 @@ +package org.dromara.easyai.function; + + +import org.dromara.easyai.i.ActiveFunction; + +/** + * @param + * @DATA + * @Author LiDaPeng + * @Description + */ +public class Sigmoid implements ActiveFunction { + @Override + public float function(float x) { + return 1 / (1 + (float)Math.exp(-x)); + } + + @Override + public float functionG(float out) { + return out * (1 - out); + } +}