Создание комплексного вывода и ранжирование записей в Neo4j - PullRequest
Новая
       27

Создание комплексного вывода и ранжирование записей в Neo4j

0 голосов
/ 25 июня 2019

Эта проблема напрямую связана с предыдущим постом:

Как рассчитать ранг для значений с плавающей запятой в Neo4j?

Я пытаюсь объединить значения "rank" и "weight" с origin и path. Я мог бы успешно сделать это для происхождения: 

CALL 
 apoc.load.json("file:///.../input.json") YIELD value 
 UNWIND value.origin AS orig 
 MATCH(origin:concept{name:orig.label}) WITH value, collect(origin) as 
 origins 
 UNWIND value.target AS tar MATCH(target:concept{name:tar.label}) 
 UNWIND origins AS origin WITH origin, target 
 CALL apoc.algo.dijkstra(origin, target, 'link', 'Weight') yield path as 
 path, weight as weight 
 WITH origin, path, weight ORDER BY weight ASC WITH {origin: origin, weight: 
 collect(weight)} AS SuggestionForOrigin UNWIND [r in range(1, 
 SIZE(SuggestionForOrigin.weight)) | {origin: SuggestionForOrigin.origin, 
 rank:r, weight: SuggestionForOrigin.weight[r-1]}] AS suggestion RETURN 
 suggestion

Тогда я получаю следующий результат (который меня устраивает): 

{"origin": {"name": "A","type": "string"},"rank": 1,"weight": 0.0}
 {"origin": {"name": "A","type": "string"},"rank": 2,"weight": 
 0.6180339887498948}
 {"origin": {"name": "P1","type": "string"},"rank": 1,"weight": 
 0.6180339887498948}
 {"origin": {"name": "P1","type": "string"},"rank": 2,"weight": 
 1.2360679774997896}

Но когда я пытаюсь объединить параметр "путь", у меня возникают проблемы. Я думаю, я компенсирую вещи. Что-то, чего я хотел бы достичь, это (не совсем, но чтобы можно было комбинировать «путь» с соответствующим «весом»): 

{"origin": {....}, "path": {...}, "rank": 1,"weight": 0.0}

И это должно быть связано с конкретным узлом источника, если у меня есть 3 предложения путей для первого источника, они должны быть объединены вместе. Я пробовал, но это не работает так, как я хочу: 

...
 CALL apoc.algo.dijkstra(origin, target, 'link', 'Weight') yield path as 
 path, weight 
 WITH {origin: origin, path: collect(path), weight: collect(weight)} AS 
 SuggestionForOrigin 
 UNWIND [r in range(1, SIZE(SuggestionForOrigin.weight)) | {rank:r, weight: 
 SuggestionForOrigin.weight[r-1], path: SuggestionForOrigin}] AS suggestion 
 WITH {origin: SuggestionForOrigin.origin, suggestions: collect(suggestion) 
 [0..3]} AS output 
 RETURN output

Буду признателен, если вы поможете.

1 Ответ

1 голос
/ 25 июня 2019

Это должно работать: 

...
CALL apoc.algo.dijkstra(origin, target, 'link', 'Weight') YIELD path, weight
WITH origin, path, weight
ORDER BY weight
WITH origin, COLLECT(path) AS ps, COLLECT(weight) AS ws
UNWIND [r IN RANGE(1, SIZE(ws)) | {
  origin: origin,
  path: ps[r-1],
  rank: r,
  weight: ws[r-1]}] AS res
RETURN res;
...