From 3a93734a5d91a439d03e14f89776c5776ba63e00 Mon Sep 17 00:00:00 2001 From: inter Date: Thu, 4 Sep 2025 14:09:09 +0800 Subject: [PATCH] Add File --- .../org/dromara/easyai/entity/WordMatrix.java | 69 +++++++++++++++++++ 1 file changed, 69 insertions(+) create mode 100644 src/main/java/org/dromara/easyai/entity/WordMatrix.java diff --git a/src/main/java/org/dromara/easyai/entity/WordMatrix.java b/src/main/java/org/dromara/easyai/entity/WordMatrix.java new file mode 100644 index 0000000..86766c7 --- /dev/null +++ b/src/main/java/org/dromara/easyai/entity/WordMatrix.java @@ -0,0 +1,69 @@ +package org.dromara.easyai.entity; + +import org.dromara.easyai.matrixTools.Matrix; +import org.dromara.easyai.i.OutBack; + +import java.util.ArrayList; +import java.util.List; + +/** + * @param + * @DATA + * @Author LiDaPeng + * @Description + */ +public class WordMatrix implements OutBack { + private final float[] vector; + + public WordMatrix(int size) { + vector = new float[size]; + } + + public Matrix getVector() throws Exception { + Matrix matrix = new Matrix(1, vector.length); + for (int i = 0; i < vector.length; i++) { + matrix.setNub(0, i, vector[i]); + } + return matrix; + } + + public List getList() { + List list = new ArrayList<>(); + for (float v : vector) { + list.add(v); + } + return list; + } + + @Override + public void getBack(float out, int id, long eventId) { + + } + + @Override + public void getSoftMaxBack(long eventId, List softMax) { + + } + + + @Override + public void backWord(String word, long eventId) { + + } + + + @Override + public void getBackMatrix(Matrix matrix, int id, long eventId) { + + } + + @Override + public void getWordVector(int id, float w) { + vector[id - 1] = w; + } + + @Override + public void getBackThreeChannelMatrix(ThreeChannelMatrix picture) { + + } +}