diff --git a/src/main/java/org/dromara/easyai/entity/WordTwoVectorModel.java b/src/main/java/org/dromara/easyai/entity/WordTwoVectorModel.java
new file mode 100644
index 0000000..e9f137d
--- /dev/null
+++ b/src/main/java/org/dromara/easyai/entity/WordTwoVectorModel.java
@@ -0,0 +1,26 @@
+package org.dromara.easyai.entity;
+
+import org.dromara.easyai.rnnNerveCenter.ModelParameter;
+
+import java.util.List;
+
+public class WordTwoVectorModel {
+    private ModelParameter modelParameter;
+    private List<String> wordList;
+
+    public ModelParameter getModelParameter() {
+        return modelParameter;
+    }
+
+    public void setModelParameter(ModelParameter modelParameter) {
+        this.modelParameter = modelParameter;
+    }
+
+    public List<String> getWordList() {
+        return wordList;
+    }
+
+    public void setWordList(List<String> wordList) {
+        this.wordList = wordList;
+    }
+}