From 3ad9a7e62bb89a5482366b7b9f3029b1b9370846 Mon Sep 17 00:00:00 2001 From: inter Date: Thu, 4 Sep 2025 14:09:20 +0800 Subject: [PATCH] Add File --- .../model/TransWordVectorModel.java | 56 +++++++++++++++++++ 1 file changed, 56 insertions(+) create mode 100644 src/main/java/org/dromara/easyai/transFormer/model/TransWordVectorModel.java diff --git a/src/main/java/org/dromara/easyai/transFormer/model/TransWordVectorModel.java b/src/main/java/org/dromara/easyai/transFormer/model/TransWordVectorModel.java new file mode 100644 index 0000000..111ce93 --- /dev/null +++ b/src/main/java/org/dromara/easyai/transFormer/model/TransWordVectorModel.java @@ -0,0 +1,56 @@ +package org.dromara.easyai.transFormer.model; + +import java.util.List; + +/** + * @author lidapeng + * @time 2025/4/3 15:27 + * @des transFormer词向量模型 + */ +public class TransWordVectorModel { + private List wordList;//词离散id + private List wordVectorModel;//词向量 + private Float[] positionMatrix;//位置编码 + private int x;//词向量行数 + private int y;//词向量列数 + + public Float[] getPositionMatrix() { + return positionMatrix; + } + + public void setPositionMatrix(Float[] positionMatrix) { + this.positionMatrix = positionMatrix; + } + + public int getX() { + return x; + } + + public void setX(int x) { + this.x = x; + } + + public int getY() { + return y; + } + + public void setY(int y) { + this.y = y; + } + + public List getWordList() { + return wordList; + } + + public void setWordList(List wordList) { + this.wordList = wordList; + } + + public List getWordVectorModel() { + return wordVectorModel; + } + + public void setWordVectorModel(List wordVectorModel) { + this.wordVectorModel = wordVectorModel; + } +}