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]

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

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

"""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

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

import os
import random

import pandas as pd

from shutil import copyfile

class DataSplitter:
    
    def __init__(self, split_ratio):