Извлекайте или сохраняйте результаты запросов Gremlin в AWS Neptune ML, в блокноте IPython.

Question

У меня есть графическая база данных, хранящаяся на AWS Neptune, которую мне нужно запросить с помощью гремлина в блокноте Jupyter IPython. Я применяю функциональные возможности графических нейронных сетей, предлагаемые Neptune ML, для задачи прогнозирования ссылок. В частности, я хочу предсказать, какие узлы «TYPE_X» связаны с узлами, сохраненными в моей переменной «id_variable».

Мой запрос выглядит так:

      %%gremlin
g.with("Neptune#ml.endpoint","${endpoint}").
    V(${id_variable}).
    project('name', 'related to').
        by('name').
        by( out('RELATED_TO').with("Neptune#ml.prediction").
            hasLabel('TYPE_X').values('name') ).
    order(local).by(keys, desc)

который возвращает следующий вывод:

      {'name': 'AANAT', 'related to': 'WDR7'}
{'name': 'ACACA', 'related to': 'BTN1A1'}
{'name': 'ACTA1', 'related to': 'MDH'}
{'name': 'ALAS1', 'related to': 'WDR7'}
{'name': 'ALAS2', 'related to': 'TAC3'}
{'name': 'ALDH2', 'related to': 'SOCS2'}
{'name': 'ALDOA', 'related to': 'PRKAB2'}
{'name': 'AKR1B1', 'related to': 'ODF2L'}
{'name': 'ALOX15', 'related to': 'BMP15'}

Моя проблема в том, что этот вывод отображается как встроенный в вывод ячейки ноутбука; однако я хотел бы либо назначить его переменной, либо сохранить в файле, например, в формате JSON. На самом деле, я не могу выполнять присваивание переменной с помощьюклеточная магия, и пока я не нашел способа записать вывод в файл.

Обратите внимание, что мне не удалось выполнить этот запрос в обычном сценарии .py с помощью библиотеки gremlin_python, так как она, похоже, не поддерживает функции машинного обучения Neptune (в частности, она выдает ошибку насинтаксис).

Любое предложение более чем приветствуется!

Заранее спасибо.