{"cells":[{"metadata":{},"cell_type":"markdown","source":"Add missing datasets to achiv high score."},{"metadata":{"_uuid":"051d70d956493feee0c6d64651c6a088724dca2a","_execution_state":"idle","trusted":true},"cell_type":"code","source":"library(data.table)\n\ntrain <- fread(\"../input/vrddata/challenge-2019-train-vrd-labels.csv\")\ntr <- fread(\"../input/vrddata/challenge-2019-train-vrd.csv\")\nhead(train)\nhead(tr)","execution_count":null,"outputs":[]},{"metadata":{"trusted":true},"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,300)","execution_count":null,"outputs":[]},{"metadata":{"trusted":true},"cell_type":"code","source":"dt <- dt[1:300]\ndt[, pred := paste( N/sum(N), LabelName1, XMin1, YMin1, XMax1, YMax1, LabelName2, XMin2, YMin2, XMax2, YMax2, RelationshipLabel  )  ]\nhead(dt,300)","execution_count":null,"outputs":[]},{"metadata":{"trusted":true},"cell_type":"code","source":"sub <- fread('../input/open-images-2019-visual-relationship/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  dt$pred[51], dt$pred[52], dt$pred[53], dt$pred[54],dt$pred[55], dt$pred[56], dt$pred[57], dt$pred[58], dt$pred[59], dt$pred[60],\n  dt$pred[61], dt$pred[62], dt$pred[63], dt$pred[64],dt$pred[65], dt$pred[66], dt$pred[67], dt$pred[68], dt$pred[69], dt$pred[70],\n  dt$pred[71], dt$pred[72], dt$pred[73], dt$pred[74],dt$pred[75], dt$pred[76], dt$pred[77], dt$pred[78], dt$pred[79], dt$pred[80],\n  dt$pred[81], dt$pred[82], dt$pred[83], dt$pred[84],dt$pred[85], dt$pred[86], dt$pred[87], dt$pred[88], dt$pred[89], dt$pred[90],\n  dt$pred[91], dt$pred[92], dt$pred[93], dt$pred[94],dt$pred[95], dt$pred[96], dt$pred[97], dt$pred[98], dt$pred[99], dt$pred[100]\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}