Add File

2025-09-04 14:08:46 +08:00
parent 7671645f42
commit 5aa6e850b2
1 changed files with 103 additions and 0 deletions
--- a/src/main/java/org/dromara/easyai/nerveEntity/SoftMax.java
+++ b/src/main/java/org/dromara/easyai/nerveEntity/SoftMax.java
@@ -0,0 +1,103 @@
+package org.dromara.easyai.nerveEntity;
+
+import org.dromara.easyai.config.RZ;
+import org.dromara.easyai.i.OutBack;
+
+import java.util.ArrayList;
+import java.util.List;
+import java.util.Map;
+
+public class SoftMax extends Nerve {
+    private final List<OutNerve> outNerves;
+    private final boolean isShowLog;
+
+    public SoftMax(int upNub, boolean isDynamic, List<OutNerve> outNerves, boolean isShowLog, int coreNumber) throws Exception {
+        super(0, upNub, "softMax", 0, 0, false, null, isDynamic
+                , RZ.NOT_RZ, 0, 0, 0, 0, 0, coreNumber, 0
+                , 0, false, null, 0.9f, 0.9f, false, 1);
+        this.outNerves = outNerves;
+        this.isShowLog = isShowLog;
+    }
+
+    @Override
+    protected void input(long eventId, float parameter, boolean isStudy, Map<Integer, Float> E, OutBack outBack) throws Exception {
+        boolean allReady = insertParameter(eventId, parameter);
+        if (allReady) {
+            Mes mes = softMax(eventId);//输出值
+            int key = 0;
+            if (isStudy) {//学习
+                for (Map.Entry<Integer, Float> entry : E.entrySet()) {
+                    if (entry.getValue() > 0.9) {
+                        key = entry.getKey();
+                        break;
+                    }
+                }
+                if (isShowLog) {
+                    System.out.println("softMax==" + key + ",out==" + mes.poi + ",nerveId==" + mes.typeID);
+                }
+                List<Float> errors = error(mes, key);
+                features.remove(eventId); //清空当前上层输入参数参数
+                int size = outNerves.size();
+                for (int i = 0; i < size; i++) {
+                    outNerves.get(i).getGBySoftMax(errors.get(i), eventId);
+                }
+            } else {//输出
+                destoryParameter(eventId);
+                if (outBack != null) {
+                    outBack.getBack(mes.poi, mes.typeID, eventId);
+                    outBack.getSoftMaxBack(eventId, mes.softMax);
+                } else {
+                    throw new Exception("not find outBack");
+                }
+            }
+        }
+    }
+
+    private List<Float> error(Mes mes, int key) {
+        int t = key - 1;
+        List<Float> softMax = mes.softMax;
+        List<Float> error = new ArrayList<>();
+        for (int i = 0; i < softMax.size(); i++) {
+            float self = softMax.get(i);
+            float myError;
+            if (i != t) {
+                myError = -self;
+            } else {
+                myError = 1 - self;
+            }
+            error.add(myError);
+        }
+        return error;
+    }
+
+    private Mes softMax(long eventId) {//计算当前输出结果
+        float sigma = 0;
+        int id = 0;
+        float poi = 0;
+        Mes mes = new Mes();
+        List<Float> featuresList = features.get(eventId);
+        for (float value : featuresList) {
+            sigma = (float) Math.exp(value) + sigma;
+        }
+        List<Float> softMax = new ArrayList<>();
+        for (int i = 0; i < featuresList.size(); i++) {
+            float eSelf = (float) Math.exp(featuresList.get(i));
+            float value = eSelf / sigma;
+            softMax.add(value);
+            if (value > poi) {
+                poi = value;
+                id = i + 1;
+            }
+        }
+        mes.softMax = softMax;
+        mes.typeID = id;
+        mes.poi = poi;
+        return mes;
+    }
+
+    static class Mes {
+        int typeID;
+        float poi;
+        List<Float> softMax;
+    }
+}