alex/restaurant-search
1
Fork 0
Code Releases Activity

Various improvements

Browse Source
This commit is contained in:
Alex Yatskov 2015-03-26 12:51:49 +09:00
parent c2997ff3a1
commit 474d02ebcb
1 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
util.go
View File

@ -58,9 +58,9 @@ func innerProduct(features1 featureMap, features2 featureMap) float64 {
}
func walkMatches(entries records, features featureMap, minScore float64, callback func(record, float64)) {
for _, record := range entries {
if score := innerProduct(features, record.features); score >= minScore {
callback(record, score)
for _, entry := range entries {
if score := innerProduct(features, entry.features); score >= minScore {
callback(entry, score)
}
}
}
@ -69,8 +69,8 @@ func statRecords(entries records, features featureMap, minScore float64) (float6
var compatibility float64
var count int
walkMatches(entries, features, minScore, func(record record, score float64) {
compatibility += record.compatibility
walkMatches(entries, features, minScore, func(entry record, score float64) {
compatibility += entry.compatibility
count++
})
@ -92,9 +92,9 @@ func stepRange(min, max float64, steps int, callback func(float64)) {
func findRecords(entries records, features featureMap, minScore float64) records {
var foundEntries records
walkMatches(entries, features, minScore, func(record record, score float64) {
record.score = score
foundEntries = append(foundEntries, record)
walkMatches(entries, features, minScore, func(entry record, score float64) {
entry.score = score
foundEntries = append(foundEntries, entry)
})
sort.Sort(foundEntries)