github/semantic
1
1
Fork 0
mirror of https://github.com/github/semantic.git synced 2024-12-19 21:01:35 +03:00
Code Issues Projects Releases Wiki Activity

Memoize the feature vector computation for bs.

Browse Source
This commit is contained in:
Rob Rix 2016-06-30 12:19:01 -04:00
parent 6e1a7957ec
commit 1a35975c4f
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/Data/RandomWalkSimilarity.hs
View File

@ -22,12 +22,13 @@ rws compare getLabel as bs
| null as, null bs = [] | null as, null bs = []
| null as = insert <$> bs | null as = insert <$> bs
| null bs = delete <$> as | null bs = delete <$> as
| otherwise = uncurry deleteRemaining . (`runState` []) $ traverse findNearestNeighbourTo (featurize <$> bs) | otherwise = uncurry deleteRemaining . (`runState` []) $ traverse findNearestNeighbourTo fbs
where insert = pure . Insert where insert = pure . Insert
delete = pure . Delete delete = pure . Delete
replace = (pure .) . Replace replace = (pure .) . Replace
(p, q, d) = (2, 2, 15) (p, q, d) = (2, 2, 15)
fas = featurize <$> as fas = featurize <$> as
fbs = featurize <$> bs
kdas = KdTree.build (Vector.toList . fst) fas kdas = KdTree.build (Vector.toList . fst) fas
featurize = featureVector d . pqGrams p q getLabel &&& identity featurize = featureVector d . pqGrams p q getLabel &&& identity
findNearestNeighbourTo kv@(_, v) = do findNearestNeighbourTo kv@(_, v) = do