Add File

src/main/java/org/dromara/easyai/naturalLanguage/WordModel.java

@@ -0,0 +1,67 @@
+package org.dromara.easyai.naturalLanguage;
+
+import org.dromara.easyai.randomForest.RfModel;
+
+import java.util.List;
+
+/**
+ * @author lidapeng
+ * @description中文语言分类模型
+ * @date 11:24 上午 2020/2/29
+ */
+public class WordModel {
+    private RfModel rfModel;//随机森林模型
+    private List<WorldBody> allWorld;//所有词集合
+    private List<List<String>> wordTimes;//所有分词编号
+    private float garbageTh;//垃圾分类的阈值默认0.5
+    private float trustPunishment;//信任惩罚
+    private float trustTh;//信任阈值
+
+    public float getGarbageTh() {
+        return garbageTh;
+    }
+
+    public void setGarbageTh(float garbageTh) {
+        this.garbageTh = garbageTh;
+    }
+
+    public float getTrustPunishment() {
+        return trustPunishment;
+    }
+
+    public void setTrustPunishment(float trustPunishment) {
+        this.trustPunishment = trustPunishment;
+    }
+
+    public float getTrustTh() {
+        return trustTh;
+    }
+
+    public void setTrustTh(float trustTh) {
+        this.trustTh = trustTh;
+    }
+
+    public RfModel getRfModel() {
+        return rfModel;
+    }
+
+    public void setRfModel(RfModel rfModel) {
+        this.rfModel = rfModel;
+    }
+
+    public List<WorldBody> getAllWorld() {
+        return allWorld;
+    }
+
+    public void setAllWorld(List<WorldBody> allWorld) {
+        this.allWorld = allWorld;
+    }
+
+    public List<List<String>> getWordTimes() {
+        return wordTimes;
+    }
+
+    public void setWordTimes(List<List<String>> wordTimes) {
+        this.wordTimes = wordTimes;
+    }
+}