{"cells":[{"metadata":{"_uuid":"67ffeb8b5c04758bd6d37f0c4068a44c3d4446e3","_execution_state":"idle","trusted":true,"scrolled":true},"cell_type":"code","source":"library(data.table)\n\ntrain <- fread(\"../input/challenge-2018-train-image-relationship/challenge-2018-train-vrd-labels.csv\")\ntr <- fread(\"../input/challenge-2018-train-image-relationship/challenge-2018-train-vrd.csv\")\nhead(train)\nhead(tr)","execution_count":null,"outputs":[]},{"metadata":{"trusted":true,"_uuid":"1387aee7612edcb862593ba30e6a20a55e90b5e9"},"cell_type":"code","source":"rowPaste <- function(X) apply( X,1,paste, collapse=\" \"  )\ntr[, label := rowPaste( tr[,c(\"LabelName1\",\"RelationshipLabel\",\"LabelName2\"),with=F]  )  ]\ndt <- tr[, .N , keyby=c(\"ImageID\")]\n\ndt <- tr[, list(\n  XMin1=median(XMin1),\n  XMax1=median(XMax1),\n  YMin1=median(YMin1),\n  YMax1=median(YMax1),\n  XMin2=median(XMin2),\n  XMax2=median(XMax2),\n  YMin2=median(YMin2),\n  YMax2=median(YMax2),\n  N=.N\n) , keyby=c(\"LabelName1\",\"RelationshipLabel\",\"LabelName2\")  ]\n\nsetorderv( dt, \"N\",-1  )\n\nhead(dt,50)","execution_count":null,"outputs":[]},{"metadata":{"trusted":true,"_uuid":"b3786079b0badcd80246c837b8c5968acd879552"},"cell_type":"code","source":"dt <- dt[1:50]\ndt[, pred := paste( N/sum(N), LabelName1, XMin1, YMin1, XMax1, YMax1, LabelName2, XMin2, YMin2, XMax2, YMax2, RelationshipLabel  )  ]\nhead(dt,50)","execution_count":null,"outputs":[]},{"metadata":{"trusted":true,"_uuid":"01c64376c3b4604d3e26416ea17fe691559a1a08"},"cell_type":"code","source":"sub <- fread('../input/google-ai-open-images-visual-relationship-track/VRD_sample_submission.csv')\nsub[, PredictionString := paste( \n  dt$pred[ 1], dt$pred[ 2], dt$pred[ 3], dt$pred[ 4],dt$pred[ 5], dt$pred[ 6], dt$pred[ 7], dt$pred[ 8], dt$pred[ 9], dt$pred[10],\n  dt$pred[11], dt$pred[12], dt$pred[13], dt$pred[14],dt$pred[15], dt$pred[16], dt$pred[17], dt$pred[18], dt$pred[19], dt$pred[20],\n  dt$pred[21], dt$pred[22], dt$pred[23], dt$pred[24],dt$pred[25], dt$pred[26], dt$pred[27], dt$pred[28], dt$pred[29], dt$pred[30],\n  dt$pred[31], dt$pred[32], dt$pred[33], dt$pred[34],dt$pred[35], dt$pred[36], dt$pred[37], dt$pred[38], dt$pred[39], dt$pred[40],\n  dt$pred[41], dt$pred[42], dt$pred[43], dt$pred[44],dt$pred[45], dt$pred[46], dt$pred[47], dt$pred[48], dt$pred[49], dt$pred[50]\n  ) ]\nhead(sub)\nfwrite(sub,\"sub1.csv\")","execution_count":null,"outputs":[]}],"metadata":{"kernelspec":{"display_name":"R","language":"R","name":"ir"},"language_info":{"mimetype":"text/x-r-source","name":"R","pygments_lexer":"r","version":"3.4.2","file_extension":".r","codemirror_mode":"r"}},"nbformat":4,"nbformat_minor":1}