forked from habeanf/yap
-
Notifications
You must be signed in to change notification settings - Fork 20
/
varbeam.go
61 lines (47 loc) · 1.08 KB
/
varbeam.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
package transition
import (
"yap/alg/perceptron"
// "yap/nlp/parser/dependency"
"yap/alg/search"
)
type VarBeam struct {
search.Beam
}
var _ search.Interface = &VarBeam{}
var _ perceptron.EarlyUpdateInstanceDecoder = &VarBeam{}
// var _ dependency.DependencyParser = &VarBeam{}
type NoCandidate struct{}
var _ search.Candidate = &NoCandidate{}
func (c *NoCandidate) Copy() search.Candidate {
return c
}
func (c *NoCandidate) Score() float64 {
return 0.0
}
func (c *NoCandidate) Equal(other search.Candidate) bool {
_, ok := other.(*NoCandidate)
return ok
}
func (c *NoCandidate) Len() int {
return 0
}
func (c *NoCandidate) Terminal() bool {
return false
}
func (v *VarBeam) Top(a search.Agenda) search.Candidate {
agenda := a.(*search.BaseAgenda)
for _, conf := range agenda.Confs {
if !conf.C.Terminal() {
return &NoCandidate{}
}
}
return v.Beam.Top(a)
}
func (v *VarBeam) GoalTest(p search.Problem, c search.Candidate, rounds int) bool {
_, isNoCandidate := c.(*NoCandidate)
if isNoCandidate {
return false
} else {
return v.Beam.GoalTest(p, c, rounds)
}
}