[DGL-KE] Add license to every file header (dmlc#1368)

* update metis * update * update dataloader * update dataloader * new script * update * update * update * update * update * update * update * update dataloader * update * update * update * update * update * update * update * Add license to every filer header
asaluja · Mar 17, 2020 · 635dfb4 · 635dfb4
1 parent f8ae635
commit 635dfb4
Show file tree

Hide file tree

Showing 26 changed files with 397 additions and 3 deletions.
diff --git a/apps/kg/dataloader/KGDataset.py b/apps/kg/dataloader/KGDataset.py
@@ -1,3 +1,22 @@
+# -*- coding: utf-8 -*-
+#
+# setup.py
+#
+# Copyright 2018 Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
 import os
 import numpy as np
 
@@ -513,9 +532,12 @@ def get_partition_dataset(data_path, data_name, part_id):
     if data_name == 'Freebase':
         relation_path = os.path.join(path, 'relation2id.txt')
         skip_first_line = True
-    else:
+    elif data_name in ['FB15k', 'FB15k-237', 'wn18', 'wn18rr']:
         relation_path = os.path.join(path, 'relations.dict')
         skip_first_line = False
+    else:
+        relation_path = os.path.join(path, 'relation.tsv')
+        skip_first_line = False
 
     dataset = PartitionKGDataset(relation_path, 
                                  train_path, 
@@ -550,9 +572,12 @@ def get_server_partition_dataset(data_path, data_name, part_id):
     if data_name == 'Freebase':
         relation_path = os.path.join(path, 'relation2id.txt')
         skip_first_line = True
-    else:
+    elif data_name in ['FB15k', 'FB15k-237', 'wn18', 'wn18rr']:
         relation_path = os.path.join(path, 'relations.dict')
         skip_first_line = False
+    else:
+        relation_path = os.path.join(path, 'relation.tsv')
+        skip_first_line = False
 
     dataset = PartitionKGDataset(relation_path,
                                  train_path,

diff --git a/apps/kg/dataloader/__init__.py b/apps/kg/dataloader/__init__.py
@@ -1,2 +1,21 @@
+# -*- coding: utf-8 -*-
+#
+# setup.py
+#
+# Copyright 2018 Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
 from .KGDataset import *
 from .sampler import *
diff --git a/apps/kg/dataloader/sampler.py b/apps/kg/dataloader/sampler.py
@@ -1,3 +1,22 @@
+# -*- coding: utf-8 -*-
+#
+# setup.py
+#
+# Copyright 2018 Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
 import math
 import numpy as np
 import scipy as sp

diff --git a/apps/kg/distributed/fb15k_transe_l2.sh b/apps/kg/distributed/fb15k_transe_l2.sh
@@ -1,3 +1,5 @@
+#!/bin/bash
+
 ##################################################################################
 # This script runing distmult model on Freebase dataset in distributed setting.
 # You can change the hyper-parameter in this file but DO NOT run script manually

diff --git a/apps/kg/distributed/freebase_complex.sh b/apps/kg/distributed/freebase_complex.sh
@@ -1,3 +1,5 @@
+#!/bin/bash
+
 ##################################################################################
 # This script runing ComplEx model on Freebase dataset in distributed setting.
 # You can change the hyper-parameter in this file but DO NOT run script manually

diff --git a/apps/kg/distributed/freebase_distmult.sh b/apps/kg/distributed/freebase_distmult.sh
@@ -1,3 +1,5 @@
+#!/bin/bash
+
 ##################################################################################
 # This script runing distmult model on Freebase dataset in distributed setting.
 # You can change the hyper-parameter in this file but DO NOT run script manually

diff --git a/apps/kg/distributed/freebase_transe_l2.sh b/apps/kg/distributed/freebase_transe_l2.sh
@@ -1,3 +1,5 @@
+#!/bin/bash
+
 ##################################################################################
 # This script runing distmult model on Freebase dataset in distributed setting.
 # You can change the hyper-parameter in this file but DO NOT run script manually

diff --git a/apps/kg/distributed/launch.sh b/apps/kg/distributed/launch.sh
@@ -1,3 +1,5 @@
+#!/bin/bash
+
 ##################################################################################
 # User runs this script to launch distrobited jobs on cluster
 ##################################################################################
@@ -25,7 +27,6 @@ do
     else
         ssh -i $ssh_key $user_name@$ip 'cd '$script_path'; '$script_file' '$s_id' '$server_count' '$machine_count'' &
     fi
-
 done
 
 # run command on local machine

diff --git a/apps/kg/distributed/partition.sh b/apps/kg/distributed/partition.sh
@@ -1,3 +1,5 @@
+#!/bin/bash
+
 ##################################################################################
 # User runs this script to partition a graph using METIS
 ##################################################################################

diff --git a/apps/kg/eval.py b/apps/kg/eval.py
@@ -1,3 +1,22 @@
+# -*- coding: utf-8 -*-
+#
+# setup.py
+#
+# Copyright 2018 Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
 from dataloader import EvalDataset, TrainDataset
 from dataloader import get_dataset
 

diff --git a/apps/kg/kvclient.py b/apps/kg/kvclient.py
@@ -1,3 +1,22 @@
+# -*- coding: utf-8 -*-
+#
+# setup.py
+#
+# Copyright 2018 Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
 import os
 import argparse
 import time

diff --git a/apps/kg/kvserver.py b/apps/kg/kvserver.py
@@ -1,3 +1,22 @@
+# -*- coding: utf-8 -*-
+#
+# setup.py
+#
+# Copyright 2018 Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
 import os
 import argparse
 import time

diff --git a/apps/kg/models/__init__.py b/apps/kg/models/__init__.py
@@ -1 +1,20 @@
+# -*- coding: utf-8 -*-
+#
+# setup.py
+#
+# Copyright 2018 Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
 from .general_models import KEModel
diff --git a/apps/kg/models/general_models.py b/apps/kg/models/general_models.py
@@ -1,3 +1,21 @@
+# -*- coding: utf-8 -*-
+#
+# setup.py
+#
+# Copyright 2018 Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
 """
 Graph Embedding Model
 1. TransE

diff --git a/apps/kg/models/mxnet/__init__.py b/apps/kg/models/mxnet/__init__.py
@@ -0,0 +1,18 @@
+# -*- coding: utf-8 -*-
+#
+# setup.py
+#
+# Copyright 2018 Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
diff --git a/apps/kg/models/mxnet/score_fun.py b/apps/kg/models/mxnet/score_fun.py
@@ -1,3 +1,22 @@
+# -*- coding: utf-8 -*-
+#
+# setup.py
+#
+# Copyright 2018 Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
 import numpy as np
 import mxnet as mx
 from mxnet import gluon

diff --git a/apps/kg/models/mxnet/tensor_models.py b/apps/kg/models/mxnet/tensor_models.py
@@ -1,3 +1,22 @@
+# -*- coding: utf-8 -*-
+#
+# setup.py
+#
+# Copyright 2018 Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
 """
 KG Sparse embedding
 """

diff --git a/apps/kg/models/pytorch/__init__.py b/apps/kg/models/pytorch/__init__.py
@@ -0,0 +1,18 @@
+# -*- coding: utf-8 -*-
+#
+# setup.py
+#
+# Copyright 2018 Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
diff --git a/apps/kg/models/pytorch/score_fun.py b/apps/kg/models/pytorch/score_fun.py
@@ -1,3 +1,22 @@
+# -*- coding: utf-8 -*-
+#
+# setup.py
+#
+# Copyright 2018 Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
 import torch as th
 import torch.nn as nn
 import torch.nn.functional as functional

diff --git a/apps/kg/models/pytorch/tensor_models.py b/apps/kg/models/pytorch/tensor_models.py
@@ -1,3 +1,22 @@
+# -*- coding: utf-8 -*-
+#
+# setup.py
+#
+# Copyright 2018 Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
 """
 KG Sparse embedding
 """