knowledge-graph-hub / neat-ml-schema Goto Github PK

  bert_params:
    node_columns:  # ideally these would have more text than category or ID (description)
      - category
      - id

Source:
  source_paths:
    - source1: https://someremoteurl.com/graph.tar.gz
    - source2: https://someremoteurl.com/graph2.tar.gz
    - source3: https://someremoteurl.com/graph3.tar.gz
  source_desc: 
    - source1: "This file contains the necessary graph files."
    - source2: "This file may be necessary."
    - source3: "This file may not be necessary."

grape.get_available_models_for_node_embedding()
             model_name       task_name library_name  available  ... can_use_edge_types requires_edge_weights can_use_edge_weights requires_positive_edge_weights
0              SocioDim  Node Embedding    Ensmallen       True  ...              False                 False                False                          False
1              SocioDim  Node Embedding  Karate Club       True  ...               None                  None                 None                           None
2        DeepWalk GloVe  Node Embedding    Ensmallen       True  ...               True                 False                 True                           True
3                  GLEE  Node Embedding    Ensmallen       True  ...              False                 False                False                          False
4                  GLEE  Node Embedding  Karate Club       True  ...               None                  None                 None                           None
5         Node2Vec CBOW  Node Embedding    Ensmallen       True  ...               True                 False                 True                           True
6         Node2Vec CBOW  Node Embedding   TensorFlow       True  ...               True                 False                 True                           True
7         Walklets CBOW  Node Embedding    Ensmallen       True  ...               True                 False                 True                           True
8                TransE  Node Embedding    Ensmallen       True  ...               True                 False                False                          False
9                TransE  Node Embedding   TensorFlow       True  ...               True                 False                False                          False
10               TransE  Node Embedding       PyKeen       True  ...               None                  None                 None                           None
11    Node2Vec SkipGram  Node Embedding    Ensmallen       True  ...               True                 False                 True                           True
12    Node2Vec SkipGram  Node Embedding   TensorFlow       True  ...               True                 False                 True                           True
13    Node2Vec SkipGram  Node Embedding  Karate Club       True  ...               None                  None                 None                           None
14        Nod2Vec GloVe  Node Embedding    Ensmallen       True  ...               True                 False                 True                           True
15                NetMF  Node Embedding    Ensmallen       True  ...              False                 False                False                          False
16                NetMF  Node Embedding  Karate Club       True  ...               None                  None                 None                           None
17    Walklets SkipGram  Node Embedding    Ensmallen       True  ...               True                 False                 True                           True
18    Walklets SkipGram  Node Embedding  Karate Club       True  ...               None                  None                 None                           None
19                 HOPE  Node Embedding    Ensmallen       True  ...              False                 False                False                          False
20                 HOPE  Node Embedding  Karate Club       True  ...               None                  None                 None                           None
21                SPINE  Node Embedding    Ensmallen       True  ...              False                 False                False                          False
22        WeightedSPINE  Node Embedding    Ensmallen       True  ...              False                  True                 True                           True
23    DeepWalk SkipGram  Node Embedding    Ensmallen       True  ...               True                 False                 True                           True
24        DeepWalk CBOW  Node Embedding    Ensmallen       True  ...               True                 False                 True                           True
25    Second Order LINE  Node Embedding   TensorFlow       True  ...              False                 False                False                          False
26     First Order LINE  Node Embedding   TensorFlow       True  ...              False                 False                False                          False
27               TransF  Node Embedding       PyKeen       True  ...               None                  None                 None                           None
28                QuatE  Node Embedding       PyKeen       True  ...               None                  None                 None                           None
29               AutoSF  Node Embedding       PyKeen       True  ...               None                  None                 None                           None
30             DistMult  Node Embedding       PyKeen       True  ...               None                  None                 None                           None
31                ERMLP  Node Embedding       PyKeen       True  ...               None                  None                 None                           None
32               TransH  Node Embedding       PyKeen       True  ...               None                  None                 None                           None
33                   CP  Node Embedding       PyKeen       True  ...               None                  None                 None                           None
34               RotatE  Node Embedding       PyKeen       True  ...               None                  None                 None                           None
35               CrossE  Node Embedding       PyKeen       True  ...               None                  None                 None                           None
36               DistMA  Node Embedding       PyKeen       True  ...               None                  None                 None                           None
37               RESCAL  Node Embedding       PyKeen       True  ...               None                  None                 None                           None
38               TuckER  Node Embedding       PyKeen       True  ...               None                  None                 None                           None
39               TorusE  Node Embedding       PyKeen       True  ...               None                  None                 None                           None
40               PairRE  Node Embedding       PyKeen       True  ...               None                  None                 None                           None
41                ProjE  Node Embedding       PyKeen       True  ...               None                  None                 None                           None
42               ERMLPE  Node Embedding       PyKeen       True  ...               None                  None                 None                           None
43                ConvE  Node Embedding       PyKeen       True  ...               None                  None                 None                           None
44               TransD  Node Embedding       PyKeen       True  ...               None                  None                 None                           None
45              ComplEx  Node Embedding       PyKeen       True  ...               None                  None                 None                           None
46                 HolE  Node Embedding       PyKeen       True  ...               None                  None                 None                           None
47               TransR  Node Embedding       PyKeen       True  ...               None                  None                 None                           None
48                 BoxE  Node Embedding       PyKeen       True  ...               None                  None                 None                           None
49             Role2Vec  Node Embedding  Karate Club       True  ...               None                  None                 None                           None
50               GraRep  Node Embedding  Karate Club       True  ...               None                  None                 None                           None
51  Laplacian Eigenmaps  Node Embedding  Karate Club       True  ...               None                  None                 None                           None
52               RandNE  Node Embedding  Karate Club       True  ...               None                  None                 None                           None
53             DeepWalk  Node Embedding  Karate Club       True  ...               None                  None                 None                           None
54              NMFADMM  Node Embedding  Karate Club       True  ...               None                  None                 None                           None
55              BoostNE  Node Embedding  Karate Club       True  ...               None                  None                 None                           None
56           NodeSketch  Node Embedding  Karate Club       True  ...               None                  None                 None                           None

grape.get_available_models_for_edge_prediction()
                      model_name        task_name  library_name  available  ... can_use_edge_types requires_edge_weights can_use_edge_weights requires_positive_edge_weights
58      Decision Tree Classifier  Edge Prediction  scikit-learn       True  ...              False                 False                False                          False
61        Extra Trees Classifier  Edge Prediction  scikit-learn       True  ...              False                 False                False                          False
64  Gradient Boosting Classifier  Edge Prediction  scikit-learn       True  ...              False                 False                False                          False
67      Random Forest Classifier  Edge Prediction  scikit-learn       True  ...              False                 False                False                          False
70                MLP Classifier  Edge Prediction  scikit-learn       True  ...              False                 False                False                          False
72                     GraphSAGE  Edge Prediction    TensorFlow       True  ...              False                 False                 True                          False
75                      Kipf GCN  Edge Prediction    TensorFlow       True  ...              False                 False                 True                          False
78                           GNN  Edge Prediction    TensorFlow       True  ...              False                 False                 True                          False
80                    Perceptron  Edge Prediction     Ensmallen       True  ...              False                 False                False                          False