rgcn

link-prediction/citation/citation.py

@@ -85,7 +85,7 @@ sampling_rate = 2
 train_neg = construct_negative_graph(train, sampling_rate, ('judgement', 'citation', 'judgement'))
 test_neg = construct_negative_graph(test,sampling_rate,('judgement', 'citation', 'judgement'))
 print(train.etypes)
-model = Model(13, 10, 2, train.etypes)
+model = Model(27, 10, 2, train.etypes)
 judgement_feats = train.nodes['judgement'].data['feature']
 test_judgement_feats = test.nodes['judgement'].data['feature']
 node_features = {'judgement': judgement_feats}#, 'court': court_feats}

link-prediction/citation/model.py

@@ -9,13 +9,12 @@ import json
 import csv
 from hetDotProduct import HeteroDotProductPredictor
 from rgcnClass import RGCN
-#from sageClass import GraphSAGE
 
 class Model(nn.Module):
     def __init__(self, in_features, hidden_features, out_features, rel_names):
         super().__init__()
-        self.sage = RGCN(in_features, hidden_features, out_features, rel_names)
+        self.rgcn = RGCN(in_features, hidden_features, out_features, rel_names)
         self.pred = HeteroDotProductPredictor()
     def forward(self, g, neg_g, x, etype):
-        h = self.sage(g, x)
+        h = self.rgcn(g, x)
         return h
\ No newline at end of file

link-prediction/citation/output/not_predicted_results.csv

 SUBJECT,RELATION,OBJECT,SCORE
-1258,citation,723,0.52
-799,citation,1117,0.577391304347826
-117,citation,1482,0.3834782608695652
-636,citation,1431,0.3565217391304348
-1098,citation,370,0.12782608695652173
-1394,citation,841,0.49391304347826087
-469,citation,1116,0.02
-209,citation,1188,0.13478260869565217
-51,citation,227,0.3234782608695652
-620,citation,401,0.5565217391304348
-972,citation,612,0.1208695652173913
-933,citation,796,0.40608695652173915
-236,citation,242,0.4860869565217391
-873,citation,718,0.28956521739130436
-871,citation,1367,0.11478260869565217
-958,citation,278,0.17739130434782607
-396,citation,679,0.09739130434782609
-717,citation,509,0.025217391304347827
-758,citation,1266,0.43826086956521737
-110,citation,1427,0.43826086956521737
-334,citation,431,0.20695652173913043
-626,citation,1099,0.1808695652173913
-937,citation,1108,0.5313043478260869
-758,citation,1301,0.38956521739130434
-570,citation,732,0.20173913043478262
-1099,citation,216,0.1808695652173913
-1184,citation,199,0.52
-567,citation,703,0.11826086956521739
-71,citation,1006,0.4008695652173913
-291,citation,1163,0.08956521739130435
-504,citation,819,0.43826086956521737
-73,citation,352,0.54
-622,citation,1067,0.528695652173913
-1147,citation,207,0.23304347826086957
-1050,citation,1209,0.4017391304347826
-375,citation,199,0.52
-507,citation,134,0.17739130434782607
-1403,citation,1376,0.042608695652173914
-657,citation,431,0.3652173913043478
-888,citation,393,0.21304347826086956
-330,citation,390,0.11478260869565217
-736,citation,604,0.42434782608695654
-205,citation,779,0.4860869565217391
-700,citation,930,0.11826086956521739
-551,citation,963,0.05391304347826087
-253,citation,196,0.20434782608695654
-725,citation,556,0.4339130434782609
-611,citation,486,0.3791304347826087
-1324,citation,1418,0.24608695652173912
-249,citation,984,0.2808695652173913
-1420,citation,787,0.23130434782608697
-1251,citation,333,0.16608695652173913
-1260,citation,263,0.06434782608695652
-1155,citation,1128,0.43826086956521737
-292,citation,872,0.4147826086956522
-1069,citation,134,0.0026086956521739132
-1389,citation,796,0.11478260869565217
-28,citation,796,0.15478260869565216
-294,citation,1157,0.36
-759,citation,551,0.08347826086956522
-590,citation,1341,0.2565217391304348
-928,citation,551,0.10347826086956521
-218,citation,752,0.25478260869565217
-611,citation,1368,0.3791304347826087
-303,citation,509,0.059130434782608696
-551,citation,187,0.16260869565217392
-605,citation,1234,0.5469565217391305
-611,citation,1074,0.3791304347826087
-1197,citation,123,0.3278260869565217
-1096,citation,522,0.06695652173913043
-438,citation,235,0.1356521739130435
-1369,citation,310,0.5
-1483,citation,1453,0.2530434782608696
-1272,citation,106,0.5443478260869565
-1268,citation,521,0.5182608695652174
-124,citation,336,0.36695652173913046
-1316,citation,60,0.21913043478260869
-605,citation,1255,0.5469565217391305
-1235,citation,322,0.21913043478260869
-1041,citation,431,0.3652173913043478
-1049,citation,515,0.31565217391304345
-691,citation,1064,0.07304347826086957
-810,citation,1047,0.13478260869565217
-698,citation,1482,0.52
-95,citation,796,0.3565217391304348
-303,citation,483,0.2991304347826087
-1196,citation,602,0.54
-285,citation,683,0.2582608695652174
-930,citation,700,0.11826086956521739
-1449,citation,289,0.2643478260869565
-403,citation,1035,0.10608695652173913
-1110,citation,549,0.25565217391304346
-955,citation,864,0.37217391304347824
-1005,citation,431,0.1443478260869565
-787,citation,884,0.3278260869565217
-295,citation,869,0.528695652173913
-1098,citation,1479,0.16956521739130434
-1234,citation,107,0.5452173913043479
-614,citation,1256,0.2747826086956522
-380,citation,215,0.577391304347826
-924,citation,746,0.43826086956521737
-936,citation,245,0.4860869565217391
-168,citation,1201,0.1782608695652174
-712,citation,721,0.21565217391304348
-336,citation,102,0.26956521739130435
-43,citation,1225,0.17130434782608694
-754,citation,1210,0.04695652173913043
-298,citation,1369,0.42869565217391303
-871,citation,1105,0.11478260869565217
-437,citation,862,0.21304347826086956
-309,citation,629,0.09217391304347826
-126,citation,171,0.5504347826086956
-1220,citation,1028,0.1634782608695652
-611,citation,693,0.10695652173913044
-799,citation,1106,0.577391304347826
-611,citation,796,0.05217391304347826
-778,citation,274,0.10434782608695652
-1050,citation,476,0.36
-1099,citation,626,0.1808695652173913
-395,citation,727,0.32521739130434785
-1386,citation,1294,0.43826086956521737
-1431,citation,574,0.09478260869565218
-1312,citation,25,0.14347826086956522
-221,citation,182,0.5069565217391304
-1134,citation,594,0.15391304347826087
-982,citation,1013,0.3165217391304348
-577,citation,231,0.21913043478260869
-1286,citation,1378,0.54
-79,citation,1344,0.43826086956521737
-666,citation,521,0.5182608695652174
-171,citation,576,0.5504347826086956
-1386,citation,674,0.43826086956521737
-937,citation,232,0.5069565217391304
-210,citation,842,0.43826086956521737
-671,citation,796,0.3565217391304348
-577,citation,1306,0.05217391304347826
-487,citation,585,0.23304347826086957
-920,citation,1330,0.577391304347826
-414,citation,316,0.528695652173913
-149,citation,97,0.5452173913043479
-290,citation,950,0.5730434782608695
-725,citation,378,0.4339130434782609
-873,citation,179,0.28956521739130436
-1265,citation,675,0.54
-478,citation,1389,0.021739130434782608
-226,citation,796,0.05391304347826087
-0,citation,390,0.10782608695652174
-1268,citation,431,0.3652173913043478
-367,citation,1064,0.14956521739130435
-1421,citation,615,0.40608695652173915
-203,citation,1150,0.31565217391304345
-1304,citation,218,0.21913043478260869
-1302,citation,354,0.5095652173913043
-1060,citation,66,0.20521739130434782
-912,citation,969,0.12782608695652173
-533,citation,771,0.2747826086956522
-537,citation,476,0.36
-74,citation,332,0.17739130434782607
-1272,citation,88,0.20782608695652174
-188,citation,45,0.08782608695652173
-443,citation,522,0.07652173913043478
-719,citation,390,0.042608695652173914
-562,citation,431,0.17739130434782607
-1475,citation,693,0.2608695652173913
-1128,citation,1155,0.43826086956521737
-1381,citation,1261,0.43826086956521737
-116,citation,336,0.36695652173913046
-1329,citation,1431,0.5452173913043479
-1228,citation,339,0.22956521739130434
-1360,citation,75,0.3217391304347826
-228,citation,509,0.01217391304347826
-1386,citation,1327,0.43826086956521737
-1324,citation,472,0.24608695652173912
-622,citation,740,0.528695652173913
-203,citation,788,0.31565217391304345
-152,citation,200,0.1426086956521739
-1278,citation,1224,0.16608695652173913
-298,citation,310,0.5243478260869565
-914,citation,874,0.48782608695652174
-1346,citation,947,0.24608695652173912
-1485,citation,389,0.33217391304347826
-120,citation,709,0.4026086956521739
-1174,citation,787,0.23130434782608697
-725,citation,81,0.4339130434782609
-785,citation,650,0.14956521739130435
-574,citation,938,0.025217391304347827
-1277,citation,1435,0.54
-378,citation,1129,0.3408695652173913
-464,citation,1227,0.4182608695652174
-1067,citation,941,0.3217391304347826
-28,citation,333,0.051304347826086956
-522,citation,1465,0.3226086956521739
-203,citation,1108,0.18608695652173912
-1105,citation,1227,0.4182608695652174
-471,citation,431,0.2408695652173913
-563,citation,245,0.43043478260869567
-717,citation,1336,0.2391304347826087
-615,citation,1432,0.21913043478260869
-1031,citation,612,0.1208695652173913
-889,citation,97,0.5443478260869565
-760,citation,1174,0.43826086956521737
-559,citation,118,0.10869565217391304
-1164,citation,1225,0.08434782608695653
-1105,citation,1314,0.4
-969,citation,279,0.33304347826086955
-892,citation,796,0.1208695652173913
-1073,citation,236,0.4860869565217391
-900,citation,1420,0.43826086956521737
-799,citation,586,0.577391304347826
-1277,citation,165,0.4860869565217391
-171,citation,126,0.5504347826086956
-1270,citation,694,0.43826086956521737
-955,citation,796,0.05217391304347826
-96,citation,431,0.3652173913043478
-1219,citation,1360,0.16695652173913045
-1092,citation,406,0.5617391304347826
-628,citation,960,0.16260869565217392
-465,citation,1360,0.20782608695652174
-469,citation,239,0.08434782608695653
-355,citation,1313,0.28347826086956524
-966,citation,1173,0.3408695652173913
-1110,citation,1346,0.25565217391304346
-188,citation,629,0.05391304347826087
-1277,citation,375,0.54
-1152,citation,227,0.3234782608695652
-1476,citation,445,0.43826086956521737
-868,citation,1324,0.05391304347826087
-932,citation,887,0.43826086956521737
-657,citation,221,0.43826086956521737
-1270,citation,149,0.43826086956521737
-1386,citation,597,0.43826086956521737
-760,citation,215,0.5182608695652174
-350,citation,112,0.16608695652173913
-1326,citation,921,0.5313043478260869
-1337,citation,921,0.43826086956521737
-1060,citation,290,0.29478260869565215
-1260,citation,134,0.05304347826086957
-779,citation,431,0.5443478260869565
-1469,citation,1287,0.3834782608695652
-378,citation,1451,0.4295652173913043
-902,citation,134,0.17739130434782607
-1147,citation,646,0.23304347826086957
-598,citation,699,0.2991304347826087
-1139,citation,313,0.43826086956521737
-242,citation,832,0.43826086956521737
-1041,citation,1207,0.5095652173913043
-1160,citation,58,0.43043478260869567
-1229,citation,1172,0.3591304347826087
-577,citation,1230,0.21913043478260869
-549,citation,431,0.3652173913043478
-276,citation,17,0.17130434782608694
-204,citation,263,0.08782608695652173
-1047,citation,447,0.36782608695652175
-1127,citation,255,0.40608695652173915
-68,citation,509,0.03826086956521739
+1258,citation,723,0.14695652173913043
+187,citation,928,0.4965217391304348
+1148,citation,122,0.14695652173913043
+752,citation,1095,0.5339130434782609
+1394,citation,841,0.07391304347826087
+662,citation,733,0.14695652173913043
+209,citation,1188,0.0791304347826087
+51,citation,227,0.14695652173913043
+620,citation,401,0.12347826086956522
+205,citation,131,0.14695652173913043
+1164,citation,994,0.14695652173913043
+758,citation,1266,0.14695652173913043
+110,citation,1427,0.14695652173913043
+461,citation,97,0.13217391304347825
+937,citation,1108,0.14695652173913043
+758,citation,1301,0.14695652173913043
+1220,citation,491,0.14521739130434783
+910,citation,425,0.11391304347826087
+198,citation,637,0.14695652173913043
+1184,citation,199,0.14695652173913043
+44,citation,759,0.49478260869565216
+71,citation,1006,0.14695652173913043
+291,citation,1163,0.05565217391304348
+504,citation,819,0.14695652173913043
+73,citation,352,0.14695652173913043
+183,citation,1172,0.14695652173913043
+622,citation,1067,0.14695652173913043
+1071,citation,798,0.5165217391304348
+477,citation,1436,0.1208695652173913
+375,citation,199,0.14695652173913043
+989,citation,886,0.11826086956521739
+244,citation,1015,0.5139130434782608
+1403,citation,1376,0.14695652173913043
+705,citation,728,0.14695652173913043
+888,citation,393,0.06695652173913043
+1411,citation,622,0.14695652173913043
+728,citation,705,0.14695652173913043
+736,citation,604,0.1
+700,citation,930,0.13826086956521738
+253,citation,196,0.14695652173913043
+1324,citation,1418,0.14695652173913043
+519,citation,1015,0.5139130434782608
+196,citation,839,0.14695652173913043
+68,citation,1211,0.14695652173913043
+280,citation,560,0.12173913043478261
+1155,citation,1128,0.14695652173913043
+1084,citation,833,0.5356521739130434
+292,citation,872,0.09478260869565218
+1198,citation,1375,0.07565217391304348
+42,citation,1359,0.14695652173913043
+218,citation,752,0.5339130434782609
+1173,citation,750,0.5539130434782609
+1197,citation,123,0.49043478260869566
+1086,citation,884,0.5469565217391305
+242,citation,217,0.14695652173913043
+265,citation,1054,0.07391304347826087
+728,citation,1111,0.14695652173913043
+438,citation,235,0.5304347826086957
+1280,citation,742,0.05652173913043478
+1369,citation,310,0.551304347826087
+1483,citation,1453,0.12521739130434784
+1232,citation,682,0.0782608695652174
+1272,citation,106,0.14695652173913043
+610,citation,119,0.14695652173913043
+124,citation,336,0.14695652173913043
+1316,citation,60,0.14695652173913043
+1235,citation,322,0.10173913043478261
+1029,citation,501,0.0782608695652174
+764,citation,558,0.14695652173913043
+810,citation,1047,0.13826086956521738
+698,citation,1482,0.14695652173913043
+764,citation,315,0.14695652173913043
+303,citation,483,0.08521739130434783
+1196,citation,602,0.14695652173913043
+1278,citation,350,0.14695652173913043
+930,citation,700,0.13826086956521738
+614,citation,570,0.5565217391304348
+884,citation,587,0.5234782608695652
+1449,citation,289,0.14695652173913043
+1110,citation,549,0.14695652173913043
+955,citation,864,0.13391304347826086
+655,citation,1472,0.011304347826086957
+1340,citation,909,0.4930434782608696
+652,citation,1034,0.5078260869565218
+295,citation,869,0.14695652173913043
+1234,citation,107,0.14695652173913043
+578,citation,262,0.14695652173913043
+614,citation,1256,0.48869565217391303
+978,citation,501,0.0782608695652174
+1015,citation,1241,0.5139130434782608
+1249,citation,55,0.14695652173913043
+380,citation,215,0.14695652173913043
+924,citation,746,0.14695652173913043
+1260,citation,783,0.020869565217391306
+448,citation,199,0.08347826086956522
+712,citation,721,0.09217391304347826
+336,citation,102,0.14695652173913043
+298,citation,1369,0.05652173913043478
+437,citation,862,0.11565217391304349
+1385,citation,86,0.14695652173913043
+1220,citation,1028,0.14173913043478262
+949,citation,1291,0.12
+728,citation,238,0.14695652173913043
+395,citation,1295,0.4956521739130435
+1403,citation,1477,0.14695652173913043
+395,citation,727,0.14695652173913043
+1386,citation,1294,0.14695652173913043
+221,citation,182,0.14695652173913043
+1134,citation,594,0.14695652173913043
+982,citation,1013,0.06782608695652174
+577,citation,231,0.14695652173913043
+1309,citation,272,0.14695652173913043
+1286,citation,1378,0.14695652173913043
+79,citation,1344,0.14695652173913043
+1188,citation,1396,0.023478260869565216
+570,citation,614,0.5565217391304348
+1161,citation,1236,0.14695652173913043
+1386,citation,674,0.14695652173913043
+937,citation,232,0.14695652173913043
+539,citation,153,0.14695652173913043
+210,citation,842,0.14695652173913043
+577,citation,1306,0.11304347826086956
+1188,citation,437,0.0791304347826087
+487,citation,585,0.5052173913043478
+736,citation,431,0.0782608695652174
+839,citation,196,0.14695652173913043
+414,citation,316,0.14695652173913043
+149,citation,97,0.13217391304347825
+1416,citation,752,0.5339130434782609
+290,citation,950,0.5408695652173913
+1149,citation,1193,0.057391304347826085
+1265,citation,675,0.14695652173913043
+714,citation,752,0.5339130434782609
+1059,citation,247,0.14695652173913043
+671,citation,733,0.14695652173913043
+712,citation,573,0.029565217391304348
+1421,citation,615,0.12956521739130436
+203,citation,1150,0.14695652173913043
+1302,citation,354,0.14695652173913043
+557,citation,151,0.14695652173913043
+667,citation,435,0.0008695652173913044
+912,citation,969,0.14695652173913043
+533,citation,771,0.14695652173913043
+911,citation,877,0.14695652173913043
+1164,citation,211,0.14695652173913043
+443,citation,522,0.00782608695652174
+1128,citation,1155,0.14695652173913043
+1381,citation,1261,0.14695652173913043
+116,citation,336,0.14695652173913043
+733,citation,199,0.14695652173913043
+1476,citation,1103,0.11826086956521739
+261,citation,851,0.49391304347826087
+1375,citation,1211,0.07565217391304348
+662,citation,199,0.14695652173913043
+1149,citation,67,0.10260869565217391
+1386,citation,1327,0.14695652173913043
+1324,citation,472,0.14695652173913043
+622,citation,740,0.14695652173913043
+203,citation,788,0.14695652173913043
+152,citation,200,0.5252173913043479
+465,citation,349,0.14695652173913043
+1278,citation,1224,0.14521739130434783
+914,citation,874,0.14695652173913043
+120,citation,709,0.11565217391304349
+578,citation,99,0.14695652173913043
+448,citation,869,0.08347826086956522
+785,citation,650,0.0782608695652174
+1277,citation,1435,0.14695652173913043
+1401,citation,1311,0.542608695652174
+1009,citation,1088,0.14695652173913043
+1397,citation,228,0.14695652173913043
+203,citation,1108,0.14695652173913043
+717,citation,1336,0.14695652173913043
+615,citation,1432,0.12956521739130436
+889,citation,97,0.13217391304347825
+760,citation,1174,0.14695652173913043
+47,citation,970,0.5808695652173913
+1105,citation,1314,0.5417391304347826
+969,citation,279,0.14695652173913043
+505,citation,859,0.11739130434782609
+900,citation,1420,0.14695652173913043
+729,citation,742,0.05652173913043478
+1270,citation,694,0.14695652173913043
+526,citation,351,0.11130434782608696
+73,citation,514,0.09391304347826086
+469,citation,239,0.02
+355,citation,1313,0.14695652173913043
+1389,citation,502,0.042608695652173914
+966,citation,1173,0.5652173913043478
+710,citation,46,0.5391304347826087
+1110,citation,1346,0.14695652173913043
+1277,citation,375,0.14695652173913043
+1152,citation,227,0.14695652173913043
+555,citation,1306,0.04869565217391304
+1476,citation,445,0.14695652173913043
+868,citation,1324,0.14695652173913043
+634,citation,570,0.5565217391304348
+932,citation,887,0.14695652173913043
+657,citation,221,0.14695652173913043
+1270,citation,149,0.14695652173913043
+1243,citation,752,0.5339130434782609
+351,citation,553,0.021739130434782608
+1386,citation,597,0.14695652173913043
+1015,citation,687,0.5139130434782608
+760,citation,215,0.14695652173913043
+350,citation,112,0.14695652173913043
+1326,citation,921,0.14695652173913043
+570,citation,307,0.5565217391304348
+1337,citation,921,0.14695652173913043
+989,citation,55,0.5852173913043478
+1469,citation,1287,0.14695652173913043
+21,citation,422,0.1356521739130435
+1139,citation,313,0.14695652173913043
+1412,citation,693,0.00608695652173913
+242,citation,832,0.14695652173913043
+1041,citation,1207,0.14695652173913043
+1160,citation,58,0.14695652173913043
+1229,citation,1172,0.14695652173913043
+577,citation,1230,0.14695652173913043
+276,citation,17,0.14521739130434783
+1047,citation,447,0.14695652173913043

link-prediction/citation/test_gsData.py

@@ -23,7 +23,9 @@ def return_edges():
 def return_features_labels():
     data = pd.read_csv("input/OE_LABELS.csv") #labels.csv
     data = data.drop(columns=['DATETIME'])
-    data = data[['tid','LAW','INFRINGEMENT','PERSON','NATION/RELIGIOUS/POL_GROUP','GEOGRAPHICAL_LOC','LANGUAGE','LOCATION_STRUCTURE','LOCATION','CONTACTNAME','ORGANIZATION','ORGANIZATION2','GEOGRAPHY2','PERCENT']]
+
+    #uncomment below for 13 features
+    #data = data[['tid','LAW','INFRINGEMENT','PERSON','NATION/RELIGIOUS/POL_GROUP','GEOGRAPHICAL_LOC','LANGUAGE','LOCATION_STRUCTURE','LOCATION','CONTACTNAME','ORGANIZATION','ORGANIZATION2','GEOGRAPHY2','PERCENT']]
     with open('input/MAPPING.json') as json_file: 
         j = json.load(json_file)
     for i in range(len(data)):

link-prediction/citation/train_gsData.py

@@ -23,7 +23,9 @@ def return_edges():
 def return_features_labels():
     data = pd.read_csv("input/OE_LABELS.csv") #labels.csv
     data = data.drop(columns=['DATETIME'])
-    data = data[['tid','LAW','INFRINGEMENT','PERSON','NATION/RELIGIOUS/POL_GROUP','GEOGRAPHICAL_LOC','LANGUAGE','LOCATION_STRUCTURE','LOCATION','CONTACTNAME','ORGANIZATION','ORGANIZATION2','GEOGRAPHY2','PERCENT']]
+
+    #uncomment below for 13 features
+    #data = data[['tid','LAW','INFRINGEMENT','PERSON','NATION/RELIGIOUS/POL_GROUP','GEOGRAPHICAL_LOC','LANGUAGE','LOCATION_STRUCTURE','LOCATION','CONTACTNAME','ORGANIZATION','ORGANIZATION2','GEOGRAPHY2','PERCENT']]
     
     with open('input/MAPPING.json') as json_file: 
         j = json.load(json_file)

link-prediction/similarity/model.py

@@ -9,13 +9,12 @@ import json
 import csv
 from hetDotProduct import HeteroDotProductPredictor
 from rgcnClass import RGCN
-#from sageClass import GraphSAGE
 
 class Model(nn.Module):
     def __init__(self, in_features, hidden_features, out_features, rel_names):
         super().__init__()
-        self.sage = RGCN(in_features, hidden_features, out_features, rel_names)
+        self.rgcn = RGCN(in_features, hidden_features, out_features, rel_names)
         self.pred = HeteroDotProductPredictor()
     def forward(self, g, neg_g, x, etype):
-        h = self.sage(g, x)
+        h = self.rgcn(g, x)
         return h
\ No newline at end of file

link-prediction/similarity/test_gsData.py

@@ -23,6 +23,8 @@ def return_edges():
 def return_features_labels():
     data = pd.read_csv("input/OE_LABELS.csv") #labels.csv
     data = data.drop(columns=['DATETIME'])
+
+    #uncomment below for 13 features
     #data = data[['tid','LAW','INFRINGEMENT','PERSON','NATION/RELIGIOUS/POL_GROUP','GEOGRAPHICAL_LOC','LANGUAGE','LOCATION_STRUCTURE','LOCATION','CONTACTNAME','ORGANIZATION','ORGANIZATION2','GEOGRAPHY2','PERCENT']]
     
     with open('input/MAPPING.json') as json_file: 

link-prediction/similarity/train_gsData.py

@@ -23,6 +23,8 @@ def return_edges():
 def return_features_labels():
     data = pd.read_csv("input/OE_LABELS.csv") #labels.csv
     data = data.drop(columns=['DATETIME'])
+
+    #uncomment below for 13 features
     #data = data[['tid','LAW','INFRINGEMENT','PERSON','NATION/RELIGIOUS/POL_GROUP','GEOGRAPHICAL_LOC','LANGUAGE','LOCATION_STRUCTURE','LOCATION','CONTACTNAME','ORGANIZATION','ORGANIZATION2','GEOGRAPHY2','PERCENT']]
     
     with open('input/MAPPING.json') as json_file: