Commit ee6e7da3 authored by Jaspreet Singh's avatar Jaspreet Singh
Browse files

rgcn

parent a1d5d534
......@@ -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}
......
......@@ -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
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
SUBJECT,RELATION,OBJECT,SCORE
1195,citation,963,0.9243478260869565
377,citation,333,0.7521739130434782
1464,citation,796,0.9539130434782609
187,citation,928,0.8704347826086957
555,citation,431,0.9773913043478261
704,citation,844,0.811304347826087
1355,citation,528,0.5939130434782609
1148,citation,122,0.8165217391304348
752,citation,1095,0.9547826086956521
662,citation,733,0.8356521739130435
960,citation,431,0.9626086956521739
130,citation,796,0.7165217391304348
952,citation,134,0.9330434782608695
1111,citation,1000,0.7026086956521739
1069,citation,462,0.6982608695652174
1464,citation,635,0.6956521739130435
988,citation,1487,0.957391304347826
1090,citation,359,0.5973913043478261
205,citation,131,0.8947826086956522
1164,citation,994,0.6469565217391304
829,citation,134,0.9982608695652174
1485,citation,787,0.8791304347826087
486,citation,796,0.7165217391304348
461,citation,97,0.7608695652173914
144,citation,796,0.7165217391304348
356,citation,817,0.8913043478260869
1002,citation,390,0.7321739130434782
1245,citation,17,0.8843478260869565
78,citation,747,0.9260869565217391
1468,citation,263,0.8139130434782609
708,citation,1064,0.76
1151,citation,844,0.811304347826087
1220,citation,491,0.9965217391304347
1066,citation,193,0.9878260869565217
910,citation,425,0.6730434782608695
136,citation,703,0.9965217391304347
198,citation,637,0.7643478260869565
248,citation,1064,0.9330434782608695
330,citation,796,0.9147826086956522
382,citation,644,0.908695652173913
894,citation,865,0.6191304347826087
78,citation,1334,0.8539130434782609
44,citation,759,0.8095652173913044
350,citation,644,0.837391304347826
183,citation,1172,0.8608695652173913
1071,citation,798,0.8791304347826087
596,citation,1177,0.9391304347826087
727,citation,997,0.6730434782608695
1358,citation,693,0.6478260869565218
146,citation,390,0.9243478260869565
1154,citation,1145,0.9965217391304347
477,citation,1436,0.8139130434782609
989,citation,886,0.9243478260869565
244,citation,1015,0.7773913043478261
705,citation,728,0.717391304347826
1411,citation,622,0.7382608695652174
960,citation,592,0.6756521739130434
1138,citation,824,0.9626086956521739
728,citation,705,0.717391304347826
1185,citation,796,0.9965217391304347
291,citation,1175,0.9782608695652174
381,citation,1126,0.9547826086956521
250,citation,1201,0.8721739130434782
519,citation,1015,0.8652173913043478
1304,citation,1044,0.9817391304347826
196,citation,839,0.731304347826087
944,citation,131,0.9843478260869565
68,citation,1211,0.9843478260869565
280,citation,560,0.5991304347826087
426,citation,114,0.6408695652173914
491,citation,364,0.9817391304347826
1014,citation,686,0.8808695652173913
1010,citation,431,0.8260869565217391
1237,citation,220,0.691304347826087
942,citation,136,0.9121739130434783
1084,citation,833,0.7443478260869565
1418,citation,509,0.7913043478260869
1466,citation,686,0.908695652173913
1191,citation,787,0.9556521739130435
35,citation,1396,0.6878260869565217
1313,citation,339,0.8547826086956521
1198,citation,1375,0.8530434782608696
680,citation,923,0.8591304347826086
1467,citation,379,0.5965217391304348
42,citation,1359,0.7521739130434782
644,citation,1383,0.6426086956521739
416,citation,796,0.9878260869565217
75,citation,796,0.7165217391304348
1070,citation,390,0.9930434782608696
1085,citation,997,0.837391304347826
1092,citation,1064,0.6095652173913043
1173,citation,750,0.5904347826086956
309,citation,1033,0.6582608695652173
1311,citation,960,0.74
1086,citation,884,0.6643478260869565
451,citation,796,0.951304347826087
725,citation,431,0.717391304347826
242,citation,217,0.6130434782608696
1202,citation,796,0.7165217391304348
739,citation,693,0.6478260869565218
265,citation,1054,0.7608695652173914
790,citation,509,0.9460869565217391
1192,citation,1242,0.8947826086956522
728,citation,1111,0.717391304347826
1280,citation,742,0.7852173913043479
1070,citation,693,0.9721739130434782
415,citation,31,0.8904347826086957
459,citation,1376,0.9565217391304348
1232,citation,682,0.92
610,citation,119,0.6521739130434783
1029,citation,501,0.7165217391304348
808,citation,509,0.9565217391304348
815,citation,123,0.8330434782608696
473,citation,1064,0.8034782608695652
824,citation,477,0.717391304347826
369,citation,1157,0.9260869565217391
764,citation,558,0.6495652173913044
868,citation,629,0.9721739130434782
764,citation,315,0.6495652173913044
570,citation,431,0.8930434782608696
1126,citation,852,0.9626086956521739
1141,citation,390,0.7965217391304348
1278,citation,350,0.7
753,citation,918,0.9921739130434782
614,citation,570,0.6121739130434782
998,citation,881,0.9660869565217391
884,citation,587,0.6956521739130435
740,citation,551,0.5947826086956521
655,citation,1472,0.8843478260869565
597,citation,923,0.6582608695652173
1319,citation,431,0.8765217391304347
1340,citation,909,0.7182608695652174
134,citation,903,1.0
1190,citation,844,0.811304347826087
1398,citation,287,0.6269565217391304
960,citation,1449,0.6756521739130434
46,citation,431,0.951304347826087
652,citation,1034,0.6321739130434783
919,citation,225,0.768695652173913
1006,citation,325,0.6382608695652174
578,citation,262,0.6382608695652174
374,citation,1225,0.8660869565217392
978,citation,501,0.8930434782608696
665,citation,469,0.8486956521739131
1305,citation,92,0.7495652173913043
1006,citation,796,0.7165217391304348
432,citation,339,0.9869565217391304
1015,citation,1241,0.8582608695652174
1249,citation,55,0.9826086956521739
1080,citation,58,0.7739130434782608
1456,citation,97,0.7443478260869565
1260,citation,783,0.6956521739130435
1074,citation,796,0.7165217391304348
448,citation,199,0.6408695652173914
1437,citation,1059,0.7208695652173913
1196,citation,1351,0.7
11,citation,973,0.6869565217391305
15,citation,509,0.9678260869565217
136,citation,263,0.9956521739130435
1444,citation,844,0.9817391304347826
1184,citation,975,0.7147826086956521
1307,citation,1064,0.92
1047,citation,1098,0.7182608695652174
108,citation,693,0.6478260869565218
1474,citation,796,0.9947826086956522
1385,citation,86,0.668695652173913
15,citation,1113,0.8878260869565218
949,citation,1291,0.628695652173913
728,citation,238,0.717391304347826
903,citation,490,0.7773913043478261
1052,citation,1201,0.9443478260869566
395,citation,1295,0.6747826086956522
529,citation,134,0.9982608695652174
1403,citation,1477,0.7626086956521739
259,citation,390,0.8530434782608696
1062,citation,1147,0.9547826086956521
1309,citation,272,0.6721739130434783
1022,citation,118,0.8686956521739131
298,citation,1306,0.8608695652173913
1407,citation,1147,0.9547826086956521
1188,citation,1396,0.6121739130434782
846,citation,1343,0.9721739130434782
570,citation,614,0.6121739130434782
1161,citation,1236,0.8808695652173913
1067,citation,509,0.7913043478260869
274,citation,1129,0.928695652173913
539,citation,153,0.951304347826087
392,citation,338,0.82
568,citation,796,0.7165217391304348
724,citation,1487,0.6643478260869565
881,citation,441,0.5886956521739131
871,citation,1066,0.9982608695652174
136,citation,191,0.9843478260869565
1188,citation,437,0.6330434782608696
736,citation,431,0.9078260869565218
881,citation,30,0.5886956521739131
930,citation,29,0.668695652173913
839,citation,196,0.731304347826087
903,citation,660,0.7773913043478261
843,citation,882,0.6269565217391304
906,citation,787,0.7669565217391304
1416,citation,752,0.8913043478260869
1149,citation,1193,0.8426086956521739
714,citation,752,0.8973913043478261
1059,citation,247,0.6478260869565218
671,citation,733,0.8356521739130435
644,citation,1094,0.8260869565217391
589,citation,629,0.8173913043478261
712,citation,573,0.931304347826087
470,citation,390,0.8730434782608696
1076,citation,1487,0.6991304347826087
894,citation,681,0.888695652173913
253,citation,431,0.9095652173913044
828,citation,114,0.7321739130434782
551,citation,251,0.5947826086956521
873,citation,890,0.9556521739130435
557,citation,151,0.7486956521739131
1296,citation,1042,0.7486956521739131
667,citation,435,0.9965217391304347
1229,citation,967,0.7965217391304348
1052,citation,1398,0.8739130434782608
64,citation,1094,0.82
1052,citation,1083,0.6547826086956522
555,citation,118,0.9060869565217391
911,citation,877,0.668695652173913
1164,citation,211,0.6469565217391304
54,citation,324,0.9904347826086957
455,citation,144,0.7078260869565217
266,citation,309,0.8982608695652174
1094,citation,644,0.8260869565217391
541,citation,796,0.9930434782608696
1085,citation,693,0.9243478260869565
295,citation,390,0.7321739130434782
733,citation,199,0.7391304347826086
196,citation,1487,0.9295652173913044
875,citation,796,0.7982608695652174
1063,citation,92,0.9565217391304348
1058,citation,680,0.8591304347826086
1476,citation,1103,0.6504347826086957
261,citation,851,0.8547826086956521
1375,citation,1211,0.9295652173913044
662,citation,199,0.6895652173913044
1149,citation,67,0.6782608695652174
353,citation,1314,0.9826086956521739
464,citation,1284,0.8139130434782609
1167,citation,531,0.6956521739130435
465,citation,349,0.6269565217391304
1151,citation,967,0.7965217391304348
454,citation,693,0.8878260869565218
578,citation,99,0.8060869565217391
945,citation,1209,0.8930434782608696
448,citation,869,0.6478260869565218
1054,citation,147,0.6121739130434782
1401,citation,1311,0.6565217391304348
120,citation,76,0.7095652173913043
1009,citation,1088,0.7165217391304348
545,citation,747,0.5973913043478261
1397,citation,228,0.9695652173913043
1440,citation,844,0.811304347826087
1425,citation,26,0.92
1467,citation,17,0.8130434782608695
47,citation,970,0.5947826086956521
298,citation,747,0.828695652173913
1013,citation,796,0.7165217391304348
57,citation,1110,0.9826086956521739
1256,citation,944,0.9852173913043478
1195,citation,963,0.6347826086956522
799,citation,1117,0.6504347826086957
377,citation,333,0.9008695652173913
1464,citation,796,0.8756521739130435
117,citation,1482,0.7417391304347826
636,citation,1431,0.5921739130434782
555,citation,431,0.8991304347826087
704,citation,844,0.8052173913043478
1355,citation,528,0.7486956521739131
1098,citation,370,0.9669565217391304
469,citation,1116,0.7339130434782609
960,citation,431,0.9991304347826087
130,citation,796,0.8756521739130435
972,citation,612,0.9008695652173913
933,citation,796,0.9965217391304347
236,citation,242,0.6156521739130435
952,citation,134,0.8834782608695653
873,citation,718,0.8530434782608696
1111,citation,1000,0.7478260869565218
1069,citation,462,0.6
1464,citation,635,0.7904347826086957
988,citation,1487,0.96
1090,citation,359,0.7730434782608696
871,citation,1367,0.6817391304347826
958,citation,278,0.9043478260869565
396,citation,679,0.7965217391304348
829,citation,134,0.991304347826087
717,citation,509,0.8130434782608695
1485,citation,787,0.76
486,citation,796,0.8756521739130435
334,citation,431,0.8486956521739131
144,citation,796,0.8756521739130435
356,citation,817,0.9582608695652174
1002,citation,390,0.8417391304347827
626,citation,1099,0.7773913043478261
1245,citation,17,0.5956521739130435
78,citation,747,0.7704347826086957
1468,citation,263,0.8791304347826087
708,citation,1064,0.8330434782608696
1151,citation,844,0.8052173913043478
1066,citation,193,0.7834782608695652
570,citation,732,0.6182608695652174
1099,citation,216,0.7773913043478261
136,citation,703,0.9930434782608696
248,citation,1064,0.9443478260869566
330,citation,796,0.9991304347826087
382,citation,644,0.9721739130434782
894,citation,865,0.611304347826087
78,citation,1334,0.7147826086956521
567,citation,703,0.8782608695652174
350,citation,644,0.88
596,citation,1177,0.7643478260869565
727,citation,997,0.6739130434782609
1358,citation,693,0.711304347826087
1147,citation,207,0.7956521739130434
146,citation,390,0.9678260869565217
1154,citation,1145,0.8417391304347827
1050,citation,1209,0.6339130434782608
507,citation,134,0.8834782608695653
657,citation,431,0.8486956521739131
960,citation,592,0.9182608695652174
330,citation,390,0.9939130434782608
1138,citation,824,0.7643478260869565
1185,citation,796,0.9147826086956522
205,citation,779,0.8278260869565217
551,citation,963,0.7704347826086957
291,citation,1175,0.8295652173913044
381,citation,1126,0.9165217391304348
725,citation,556,0.6243478260869565
611,citation,486,0.7860869565217391
249,citation,984,0.9817391304347826
250,citation,1201,0.8565217391304348
1420,citation,787,0.6043478260869565
1304,citation,1044,0.9669565217391304
1251,citation,333,0.9634782608695652
944,citation,131,0.7382608695652174
1260,citation,263,0.9582608695652174
426,citation,114,0.9069565217391304
491,citation,364,0.7321739130434782
1014,citation,686,0.9678260869565217
1010,citation,431,0.8486956521739131
1237,citation,220,0.9165217391304348
942,citation,136,0.9791304347826087
1418,citation,509,0.8130434782608695
1466,citation,686,0.9721739130434782
1069,citation,134,0.9339130434782609
1191,citation,787,0.9034782608695652
1389,citation,796,0.9817391304347826
35,citation,1396,0.8347826086956521
28,citation,796,0.9626086956521739
294,citation,1157,0.8156521739130435
759,citation,551,0.7434782608695653
1313,citation,339,0.8582608695652174
680,citation,923,0.9686956521739131
1467,citation,379,0.6226086956521739
590,citation,1341,0.9139130434782609
644,citation,1383,0.9104347826086957
416,citation,796,0.9721739130434782
75,citation,796,0.8756521739130435
928,citation,551,0.7147826086956521
1070,citation,390,0.9878260869565217
611,citation,1368,0.7860869565217391
1085,citation,997,0.9773913043478261
1092,citation,1064,0.8330434782608696
303,citation,509,0.86
309,citation,1033,0.7443478260869565
551,citation,187,0.7704347826086957
605,citation,1234,0.6921739130434783
611,citation,1074,0.7860869565217391
1311,citation,960,0.9182608695652174
451,citation,796,0.9895652173913043
725,citation,431,0.98
1096,citation,522,0.8356521739130435
1202,citation,796,0.8756521739130435
739,citation,693,0.711304347826087
790,citation,509,0.8130434782608695
1192,citation,1242,0.811304347826087
1070,citation,693,0.9878260869565217
415,citation,31,0.8730434782608696
459,citation,1376,0.7730434782608696
1268,citation,521,0.5904347826086956
605,citation,1255,0.6921739130434783
808,citation,509,0.9660869565217391
815,citation,123,0.6086956521739131
1041,citation,431,0.8486956521739131
1049,citation,515,0.6017391304347826
691,citation,1064,0.86
473,citation,1064,0.8330434782608696
824,citation,477,0.7643478260869565
369,citation,1157,0.9686956521739131
868,citation,629,0.8582608695652174
95,citation,796,0.8756521739130435
570,citation,431,0.8973913043478261
1126,citation,852,0.8243478260869566
1141,citation,390,0.8417391304347827
285,citation,683,0.9660869565217391
753,citation,918,0.7834782608695652
998,citation,881,0.64
403,citation,1035,0.6582608695652173
740,citation,551,0.7704347826086957
1005,citation,431,0.8486956521739131
597,citation,923,0.688695652173913
1319,citation,431,0.86
787,citation,884,0.6043478260869565
134,citation,903,0.9939130434782608
1190,citation,844,0.8052173913043478
1398,citation,287,0.9026086956521739
960,citation,1449,0.9182608695652174
46,citation,431,0.8486956521739131
919,citation,225,0.6739130434782609
1098,citation,1479,0.851304347826087
1006,citation,325,0.86
374,citation,1225,0.7817391304347826
665,citation,469,0.7339130434782609
1305,citation,92,0.7408695652173913
1006,citation,796,0.8756521739130435
432,citation,339,0.9773913043478261
1080,citation,58,0.8582608695652174
1456,citation,97,0.8617391304347826
1074,citation,796,0.8756521739130435
936,citation,245,0.731304347826087
168,citation,1201,0.8565217391304348
1437,citation,1059,0.6582608695652173
1196,citation,1351,0.7843478260869565
43,citation,1225,0.7817391304347826
11,citation,973,0.9686956521739131
15,citation,509,0.9582608695652174
754,citation,1210,0.808695652173913
136,citation,263,0.9930434782608696
1444,citation,844,0.8052173913043478
1184,citation,975,0.7104347826086956
871,citation,1105,0.6817391304347826
1307,citation,1064,0.8330434782608696
1047,citation,1098,0.851304347826087
108,citation,693,0.711304347826087
1474,citation,796,0.9852173913043478
309,citation,629,0.9808695652173913
126,citation,171,0.5973913043478261
15,citation,1113,0.9634782608695652
903,citation,490,0.8852173913043478
1052,citation,1201,0.9582608695652174
611,citation,693,0.9921739130434782
799,citation,1106,0.6504347826086957
529,citation,134,0.928695652173913
611,citation,796,0.9973913043478261
778,citation,274,0.5947826086956521
1050,citation,476,0.6739130434782609
1099,citation,626,0.7773913043478261
1431,citation,574,0.94
1312,citation,25,0.9243478260869565
259,citation,390,0.871304347826087
1062,citation,1147,0.7956521739130434
1022,citation,118,0.7417391304347826
298,citation,1306,0.9034782608695652
1407,citation,1147,0.7956521739130434
846,citation,1343,0.9582608695652174
666,citation,521,0.5904347826086956
171,citation,576,0.5973913043478261
1067,citation,509,0.8130434782608695
274,citation,1129,0.9539130434782609
392,citation,338,0.9182608695652174
568,citation,796,0.8756521739130435
724,citation,1487,0.8295652173913044
881,citation,441,0.8878260869565218
671,citation,796,0.8756521739130435
871,citation,1066,0.9026086956521739
136,citation,191,0.9243478260869565
920,citation,1330,0.9591304347826087
881,citation,30,0.8878260869565218
930,citation,29,0.591304347826087
903,citation,660,0.8852173913043478
843,citation,882,0.7678260869565218
906,citation,787,0.9182608695652174
725,citation,378,0.6243478260869565
873,citation,179,0.8530434782608696
478,citation,1389,0.9773913043478261
644,citation,1094,0.9678260869565217
589,citation,629,0.8582608695652174
226,citation,796,0.8756521739130435
0,citation,390,0.8417391304347827
470,citation,390,0.9878260869565217
1268,citation,431,0.8486956521739131
1076,citation,1487,0.8295652173913044
367,citation,1064,0.8330434782608696
894,citation,681,0.9904347826086957
253,citation,431,0.8486956521739131
828,citation,114,0.9852173913043478
551,citation,251,0.7704347826086957
873,citation,890,0.8530434782608696
1304,citation,218,0.6182608695652174
1296,citation,1042,0.8
1229,citation,967,0.7834782608695652
1052,citation,1398,0.9130434782608695
1060,citation,66,0.9852173913043478
537,citation,476,0.6739130434782609
64,citation,1094,0.9643478260869566
74,citation,332,0.9208695652173913
1052,citation,1083,0.7252173913043478
1272,citation,88,0.5956521739130435
555,citation,118,0.7678260869565218
54,citation,324,0.9165217391304348
455,citation,144,0.8991304347826087
188,citation,45,0.86
266,citation,309,0.9956521739130435
1094,citation,644,0.9678260869565217
719,citation,390,0.908695652173913
562,citation,431,0.9895652173913043
541,citation,796,0.991304347826087
1085,citation,693,0.9878260869565217
295,citation,390,0.8417391304347827
1475,citation,693,0.971304347826087
1329,citation,1431,0.9043478260869565
1228,citation,339,0.8582608695652174
1360,citation,75,0.7243478260869565
196,citation,1487,0.8295652173913044
875,citation,796,0.88
1063,citation,92,0.7408695652173913
1058,citation,680,0.9634782608695652
228,citation,509,0.8130434782608695
353,citation,1314,0.6539130434782608
464,citation,1284,0.8556521739130435
1167,citation,531,0.671304347826087
298,citation,310,0.6747826086956522
1346,citation,947,0.7434782608695653
1485,citation,389,0.6060869565217392
1151,citation,967,0.7834782608695652
454,citation,693,0.9843478260869565
945,citation,1209,0.6339130434782608
1174,citation,787,0.6043478260869565
1054,citation,147,0.6860869565217391
725,citation,81,0.6243478260869565
574,citation,938,0.9530434782608695
378,citation,1129,0.7052173913043478
464,citation,1227,0.6495652173913044
1067,citation,941,0.6956521739130435
120,citation,76,0.8356521739130435
545,citation,747,0.7704347826086957
28,citation,333,0.9556521739130435
522,citation,1465,0.8356521739130435
1105,citation,1227,0.6495652173913044
471,citation,431,0.8486956521739131
563,citation,245,0.9443478260869566
1440,citation,844,0.8052173913043478
1425,citation,26,0.971304347826087
1031,citation,612,0.9008695652173913
1467,citation,17,0.5956521739130435
298,citation,747,0.9721739130434782
1013,citation,796,0.8756521739130435
57,citation,1110,0.8017391304347826
559,citation,118,0.7730434782608696
1164,citation,1225,0.7817391304347826
1256,citation,944,0.7486956521739131
194,citation,390,0.9634782608695652
505,citation,859,0.7408695652173913
1455,citation,1064,0.9026086956521739
606,citation,431,0.8913043478260869
700,citation,703,0.9217391304347826
272,citation,287,0.7486956521739131
729,citation,742,0.7852173913043479
1003,citation,134,0.711304347826087
346,citation,263,0.9721739130434782
1371,citation,1360,0.9060869565217391
494,citation,19,0.7947826086956522
1178,citation,487,0.9739130434782609
907,citation,844,0.811304347826087
1177,citation,903,0.9904347826086957
526,citation,351,0.7486956521739131
4,citation,796,0.7252173913043478
73,citation,514,0.8678260869565217
1445,citation,629,0.84
19,citation,494,0.7947826086956522
1312,citation,1208,0.9991304347826087
1389,citation,502,0.7034782608695652
346,citation,136,0.9121739130434783
644,citation,382,0.908695652173913
1265,citation,796,0.7321739130434782
710,citation,46,0.828695652173913
1088,citation,134,0.711304347826087
848,citation,454,0.8269565217391305
555,citation,1306,0.9826086956521739
190,citation,796,0.7930434782608695
89,citation,213,0.6521739130434783
1441,citation,787,0.9982608695652174
184,citation,195,0.9930434782608696
634,citation,570,0.768695652173913
1150,citation,844,0.811304347826087
1243,citation,752,0.6121739130434782
278,citation,612,0.9182608695652174
351,citation,553,0.9095652173913044
1015,citation,687,0.6539130434782608
1080,citation,796,0.9895652173913043
570,citation,307,0.6121739130434782
397,citation,165,0.8260869565217391
989,citation,55,0.9547826086956521
494,citation,992,0.7947826086956522
486,citation,1351,0.6530434782608696
21,citation,422,0.6460869565217391
228,citation,1242,0.8739130434782608
1485,citation,701,0.7217391304347827
773,citation,1094,0.9069565217391304
1412,citation,693,0.9721739130434782
650,citation,129,0.9113043478260869
1065,citation,967,0.7965217391304348
513,citation,844,0.82
1134,citation,1487,0.7782608695652173
309,citation,440,0.6582608695652173
1455,citation,1064,0.8330434782608696
892,citation,796,0.8756521739130435
1073,citation,236,0.6156521739130435
606,citation,431,0.8156521739130435
799,citation,586,0.6504347826086957
700,citation,703,0.811304347826087
272,citation,287,0.671304347826087
1003,citation,134,0.8834782608695653
346,citation,263,0.9556521739130435
1371,citation,1360,0.9834782608695652
1277,citation,165,0.6243478260869565
494,citation,19,0.871304347826087
1178,citation,487,0.9278260869565217
907,citation,844,0.8052173913043478
1177,citation,903,0.971304347826087
171,citation,126,0.5973913043478261
955,citation,796,0.9078260869565218
96,citation,431,0.8486956521739131
4,citation,796,0.8756521739130435
1219,citation,1360,0.7243478260869565
1092,citation,406,0.8052173913043478
628,citation,960,0.9182608695652174
465,citation,1360,0.7243478260869565
1445,citation,629,0.8582608695652174
19,citation,494,0.871304347826087
1312,citation,1208,0.9347826086956522
346,citation,136,0.9791304347826087
644,citation,382,0.9721739130434782
1265,citation,796,0.8756521739130435
188,citation,629,0.8582608695652174
1088,citation,134,0.8834782608695653
848,citation,454,0.9869565217391304
190,citation,796,0.9956521739130435
89,citation,213,0.8339130434782609
1441,citation,787,0.7747826086956522
184,citation,195,0.922608695652174
1150,citation,844,0.8052173913043478
278,citation,612,0.9991304347826087
1080,citation,796,0.9834782608695652
397,citation,165,0.6243478260869565
1060,citation,290,0.7069565217391305
1260,citation,134,0.9582608695652174
779,citation,431,0.9843478260869565
494,citation,992,0.871304347826087
378,citation,1451,0.6017391304347826
486,citation,1351,0.7843478260869565
228,citation,1242,0.811304347826087
902,citation,134,0.8834782608695653
1147,citation,646,0.7956521739130434
1485,citation,701,0.6060869565217392
773,citation,1094,0.62
598,citation,699,0.7660869565217391
549,citation,431,0.8486956521739131
650,citation,129,0.9860869565217392
1065,citation,967,0.7834782608695652
513,citation,844,0.8052173913043478
204,citation,263,0.8791304347826087
1134,citation,1487,0.8295652173913044
1127,citation,255,0.6130434782608696
309,citation,440,0.7443478260869565
68,citation,509,0.8130434782608695
......@@ -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)):
......
......@@ -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)
......
......@@ -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
......@@ -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:
......
......@@ -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:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment