{"cells":[
 {
  "cell_type": "code",
  "execution_count": null,
  "metadata": {
   "collapsed": false
  },
  "outputs": [],
  "source": "library(data.table)"
 },
 {
  "cell_type": "code",
  "execution_count": null,
  "metadata": {
   "collapsed": false
  },
  "outputs": [],
  "source": "train <- fread(\"../input/train.csv\", header = TRUE, select = c(\"user_location_city\", \"orig_destination_distance\", \"hotel_cluster\"))\ntest <- fread(\"../input/test.csv\", header = TRUE, select = c(\"user_location_city\", \"orig_destination_distance\"))"
 },
 {
  "cell_type": "code",
  "execution_count": null,
  "metadata": {
   "collapsed": false
  },
  "outputs": [],
  "source": "setkey(train, user_location_city, orig_destination_distance)\nsetkey(test,  user_location_city, orig_destination_distance)"
 },
 {
  "cell_type": "code",
  "execution_count": null,
  "metadata": {
   "collapsed": false
  },
  "outputs": [],
  "source": "train <- train[user_location_city %in% test$user_location_city, .(user_location_city, orig_destination_distance, hotel_cluster)]\nhead(train)"
 },
 {
  "cell_type": "code",
  "execution_count": null,
  "metadata": {
   "collapsed": false
  },
  "outputs": [],
  "source": "train <- train[orig_destination_distance %in% test$orig_destination_distance]\nhead(train)"
 },
 {
  "cell_type": "code",
  "execution_count": null,
  "metadata": {
   "collapsed": false
  },
  "outputs": [],
  "source": "setkey(train, user_location_city, orig_destination_distance)\nsetkey(test,  user_location_city, orig_destination_distance)"
 },
 {
  "cell_type": "code",
  "execution_count": null,
  "metadata": {
   "collapsed": false
  },
  "outputs": [],
  "source": "test$hotel_cluster <- NA\ntest[user_location_city]"
 },
 {
  "cell_type": "code",
  "execution_count": null,
  "metadata": {
   "collapsed": false
  },
  "outputs": [],
  "source": "str(train)\nstr(test)"
 }
],"metadata":{"kernelspec":{"display_name":"R","language":"R","name":"ir"}}, "nbformat": 4, "nbformat_minor": 0}