|
import pandas as pd |
|
from arango import ArangoClient |
|
from tqdm import tqdm |
|
import numpy as np |
|
import itertools |
|
import requests |
|
import sys |
|
import oasis |
|
from arango import ArangoClient |
|
|
|
import torch |
|
import torch.nn.functional as F |
|
from torch.nn import Linear |
|
from arango import ArangoClient |
|
import torch_geometric.transforms as T |
|
from torch_geometric.nn import SAGEConv, to_hetero |
|
from torch_geometric.transforms import RandomLinkSplit, ToUndirected |
|
from sentence_transformers import SentenceTransformer |
|
from torch_geometric.data import HeteroData |
|
import yaml |
|
|
|
|
|
import pickle |
|
|
|
|
|
def load_hetero_data(): |
|
with open('Hgraph.pkl', 'rb') as file: |
|
data = pickle.load(file) |
|
return data |
|
|
