Update logic2.py

logic2.py

@@ -0,0 +1,30 @@
+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
+