diff --git a/src/main/java/org/dromara/easyai/config/SentenceConfig.java b/src/main/java/org/dromara/easyai/config/SentenceConfig.java new file mode 100644 index 0000000..7df7d8a --- /dev/null +++ b/src/main/java/org/dromara/easyai/config/SentenceConfig.java @@ -0,0 +1,146 @@ +package org.dromara.easyai.config; + + +/** + * @param + * @DATA + * @Author LiDaPeng + * @Description 参数配置 + */ +public class SentenceConfig { + private int typeNub = 11;//分类数量 + private int wordVectorDimension = 21;//语义分类网络词向量嵌入维度 + private int qaWordVectorDimension = 66;//问答模型此项链嵌入维度 + private int maxWordLength = 20;//最长字数 + private float weStudyPoint = 0.01f;//词向量学习学习率 + private float weLParam = 0.01f;//正则系数 + private int rzModel = RZ.NOT_RZ;//正则模式 + private boolean showLog = true; + private int minLength = 5; + private float trustPowerTh = 0.5f;//权重可信任阈值 + private int maxAnswerLength = 20;//最大回答长度 + private float sentenceTrustPowerTh = 0.2f;//语句生成可信赖阈值 + private int times = 10;//增加训练数量 + private float param = 0.01f;//线性层正则系数 + private int keyWordNerveDeep = 3; + + public int getRzModel() { + return rzModel; + } + + public void setRzModel(int rzModel) { + this.rzModel = rzModel; + } + + public int getQaWordVectorDimension() { + return qaWordVectorDimension; + } + + public void setQaWordVectorDimension(int qaWordVectorDimension) { + this.qaWordVectorDimension = qaWordVectorDimension; + } + + public int getKeyWordNerveDeep() { + return keyWordNerveDeep; + } + + public void setKeyWordNerveDeep(int keyWordNerveDeep) { + this.keyWordNerveDeep = keyWordNerveDeep; + } + + public float getParam() { + return param; + } + + public void setParam(float param) { + this.param = param; + } + + public int getTimes() { + return times; + } + + public void setTimes(int times) { + this.times = times; + } + + public float getSentenceTrustPowerTh() { + return sentenceTrustPowerTh; + } + + public void setSentenceTrustPowerTh(float sentenceTrustPowerTh) { + this.sentenceTrustPowerTh = sentenceTrustPowerTh; + } + + public int getMaxAnswerLength() { + return maxAnswerLength; + } + + public void setMaxAnswerLength(int maxAnswerLength) { + this.maxAnswerLength = maxAnswerLength; + } + + public float getTrustPowerTh() { + return trustPowerTh; + } + + public void setTrustPowerTh(float trustPowerTh) { + this.trustPowerTh = trustPowerTh; + } + + public int getMinLength() { + return minLength; + } + + public void setMinLength(int minLength) { + this.minLength = minLength; + } + + public int getMaxWordLength() { + return maxWordLength; + } + + public void setMaxWordLength(int maxWordLength) { + this.maxWordLength = maxWordLength; + } + + public boolean isShowLog() { + return showLog; + } + + public void setShowLog(boolean showLog) { + this.showLog = showLog; + } + + public int getTypeNub() { + return typeNub; + } + + public void setTypeNub(int typeNub) { + this.typeNub = typeNub; + } + + public int getWordVectorDimension() { + return wordVectorDimension; + } + + public void setWordVectorDimension(int wordVectorDimension) { + this.wordVectorDimension = wordVectorDimension; + } + + public float getWeStudyPoint() { + return weStudyPoint; + } + + public void setWeStudyPoint(float weStudyPoint) { + this.weStudyPoint = weStudyPoint; + } + + public float getWeLParam() { + return weLParam; + } + + public void setWeLParam(float weLParam) { + this.weLParam = weLParam; + } +}