|
@@ -25,7 +25,7 @@ type recomKws struct {
|
|
|
model *Model
|
|
|
}
|
|
|
|
|
|
-func (rk *recomKws) GetRecomKws(value string, count int) []*map[string]interface{} {
|
|
|
+func (rk *recomKws) GetRecomKws(value string, count int, recommendThreshold float32) []*map[string]interface{} {
|
|
|
rk.reqPool <- true
|
|
|
defer func() {
|
|
|
<-rk.reqPool
|
|
@@ -78,6 +78,9 @@ func (rk *recomKws) GetRecomKws(value string, count int) []*map[string]interface
|
|
|
pw, _ := rk.model.MostSimilar(strings.Split(v, "+"), []string{}, maxCount)
|
|
|
for k, p := range pw {
|
|
|
p.Word = strings.TrimSpace(p.Word)
|
|
|
+ if sim := p.Sim; sim < recommendThreshold {
|
|
|
+ continue
|
|
|
+ }
|
|
|
if p.Word == "" || !randomNum[k] || allKeyMap[p.Word] || len([]rune(p.Word)) == 1 {
|
|
|
continue
|
|
|
}
|