ricsi98

import os
import random

import pandas as pd

from shutil import copyfile

class DataSplitter:
    
    def __init__(self, split_ratio):

ricsi98

from typing import Dict, List


def match_length(data: bytearray, ptr: int, word: bytearray):
    offset = 0
    while offset < len(word) and offset + ptr < len(data):
        if data[ptr + offset] != word[offset]: break
        offset += 1
    return offset

ricsi98

"""Find all number of combinations that formed a sum (N) amount with given denominations (A)"""
"""The combinations can be indexed with I and for the same I the same permutation is returned every time"""

from typing import List, Dict, Set, Tuple
from tqdm import tqdm
import sys

MAX_CACHE_SIZE = 1000000
USE_PRUNE = True

ricsi98

"""PyTorch implementation of the Graph2Gauss algorithm (https://arxiv.org/pdf/1707.03815.pdf)"""

import walker
import random
import numpy as np
import networkx as nx

import torch
from torch.nn import Linear, Module, Sequential, ReLU

ricsi98

ricsi98

ricsi98

"""Create Freebase ID to node index mapping for FB15k-237 dataset."""

import json
from pathlib import Path
from typing import Dict

from torch_geometric.datasets import FB15k_237

NODE_PATH = Path("nodes.json")
EDGE_PATH = Path("edges.json")

ricsi98

ricsi98

from typing import Literal

import numpy as np
import scipy
from scipy.sparse import csr_array
from torch_geometric.datasets import Planetoid
from torch_geometric.utils import to_scipy_sparse_matrix

NormType = Literal["sym", "row", None]