Commit dd62e660 by Sanjay Krishnan

Updated with inclass material

parent 48ecc71e
{
"cells": [
{
"cell_type": "markdown",
"metadata": {},
"source": [
"# Python Document Search Engine\n",
"\n",
"Now, we will start to put together all of the topics that we have studied so far into a series of \"Python Recipes\"---coding examples that illustrate the power of thinking hard about how data is organized and structured. In the first example, we will consider a \"Python Search Engine\" that will identify relevant items given a query string.\n",
"\n",
"We're going to start with a dataset of tweets about airlines:"
]
},
{
"cell_type": "code",
"execution_count": 26,
"metadata": {},
"outputs": [
{
"name": "stdout",
"output_type": "stream",
"text": [
"Number: 14640 \t Size: 2.54372 MB \t Bytes per tweet: 173.7513661202186\n"
]
}
],
"source": [
"import csv\n",
"\n",
"def load_data(filename):\n",
" \n",
" rtn = []\n",
" #open the file with the csv reader\n",
" with open(filename, newline='') as csvfile:\n",
" tweets = csv.reader(csvfile, delimiter=',', quotechar='\"')\n",
" \n",
" next(tweets)#skip the header\n",
" \n",
" for row in tweets:\n",
" rtn.append(row[10])\n",
" \n",
" return rtn\n",
"\n",
"tweets = load_data('Tweets.csv')\n",
"\n",
"#figure out how much data we have\n",
"size = sum([i.__sizeof__() for i in tweets]) + tweets.__sizeof__()\n",
"\n",
"print('Number: ', len(tweets), '\\t Size:', size/1e6,'MB','\\t Bytes per tweet:', size/len(tweets))"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"This dataset contains a large list of tweets represented as string. We want to be able to search for phrases in these tweets. Of course, the first thing that we can do is the simple naive search routine where we scan through the entire dataset.\n",
"\n",
"## Naive Search\n",
"Suppose, we wanted to find a substring in this collection of tweets, we could write the following code that iterates through each tweet and searches for a substring:"
]
},
{
"cell_type": "code",
"execution_count": 11,
"metadata": {},
"outputs": [
{
"name": "stdout",
"output_type": "stream",
"text": [
"Find() elapsed time: 0.002858\n",
"Find() elapsed time: 0.0032\n"
]
},
{
"data": {
"text/plain": [
"[\"@VirginAmerica So excited for my first cross country flight LAX to MCO I've heard nothing but great things about Virgin America. #29DaysToGo\",\n",
" '@VirginAmerica LAX to EWR - Middle seat on a red eye. Such a noob maneuver. #sendambien #andchexmix',\n",
" '@VirginAmerica help, left expensive headphones on flight 89 IAD to LAX today. Seat 2A. No one answering L&F number at LAX!',\n",
" '@VirginAmerica plz help me win my bid upgrade for my flight 2/27 LAX--->SEA!!! 🍷👍💺✈️',\n",
" '@VirginAmerica just landed in LAX, an hour after I should of been here. Your no Late Flight bag check is not business travel friendly #nomorevirgin',\n",
" '@VirginAmerica trying to add my boy Prince to my ressie. SF this Thursday @VirginAmerica from LAX http://t.co/GsB2J3c4gM',\n",
" '@VirginAmerica Can you find us a flt out of LAX that is sooner than midnight on Monday? That would be great customer service 😃',\n",
" '@VirginAmerica congrats, you just got all my business from EWR to SFO/LAX. Fuck you @united fl1289 SFO/EWR was the clincher...',\n",
" '@VirginAmerica nervous about my flight from DC to LAX getting Cancelled Flightled tomorrow! Just sent you a DM to help me!',\n",
" '@VirginAmerica @VirginAtlantic I have just checked in flight to SFO from LAX & been told as Atlantic Flying Club Gold I get no benefits?!',\n",
" '@VirginAmerica I applied for a position in @flyLAXairport ,and I was wondering if you guys received my application.',\n",
" \"@VirginAmerica lost my luggage 4 days ago on flight VX 112 from LAX to IAD & I'm calling every day, no response.Please give me back my stuff\",\n",
" '@VirginAmerica Nice, Lofty View @flyLAXairport. #SilverStatus http://t.co/F4Tp0dAwbd',\n",
" '@VirginAmerica you should have 39 dollar LAX-Las fares!!!',\n",
" '@VirginAmerica Thanks for making my flight from LAX to JFK a nightmare by forcing me to check my carry on bag at the gate. (1)',\n",
" '@VirginAmerica Flying LAX to SFO and after looking at the awesome movie lineup I actually wish I was on a long haul.',\n",
" '@united our travel booked thru United group dept. Okc ticket agent less than willing to help with our connection in LAX.',\n",
" '@united I took the exact same aircraft in to LAX 3 days ago. It fit, no problem. The agent today told some nonsense about a policy change',\n",
" '@united the person is currently bettween gates 71A and 73 in LAX',\n",
" '@united I need the phone number to baggage claim in LAX, my mom left her phone and someone called saying they would put it there but on',\n",
" \"@united is doing musicians real dirty at LAX. I've never been blocked from getting on a flight with my bass.\",\n",
" '@united Hi, Im flying SFO-LAX-SAL-CLO. My connecting time in LAX is 1h45m. Is it enough time? Do I have to collect my bag and recheck on AV?',\n",
" '@united Maybe be hiring your own ground staff at LAX when multiple gate agents tell you your baggage is loaded you expect it to be. HOPELESS',\n",
" '@united SF crew lack a lot of customer service, LAX employees are a lot better. Wonder why...',\n",
" \"@united UA flight 1247. SFO to LAX took my carry on at gate. I'm Group 2, overhead bins are empty\",\n",
" '@united maybemange the airline alittlebetter. Arrived at LAX and no GATE! #howisthatpossible always the same thing w/u',\n",
" '@united #LAX #sunrise UAL212 LAX-JFK',\n",
" '@united Never had a flight delayed an hour due to an unbalanced load. And more delays at @flyLAXairport. Great job idiots.',\n",
" '@United is offering to reroute my SFO flight to LAX. Might be geography class time.',\n",
" \"@united I missed my connection already. Then I missed the next flight they put me on. Now I'm going to LAX instead of Hawaii. :(\",\n",
" '@united thanks for leaving our 3 year old in his own row flight 360 LAX-IAD',\n",
" \"@united - you sure missed the mark on tonight's redeye from LAX to Chicago. What a mess! You can do better!\",\n",
" '@united Flight has been delayed for another hour so only have 24 mins to transit at LAX... Extremely unlikely I will make it!',\n",
" '@united really fucked my day up Hilo to LAX 2hr30min delay because of software? missed connection, getting home 8hrs Late Flightr no upgrade nothin',\n",
" \"@united DEN-PHX flight tomorrow Cancelled Flighted. Asked for overnight 2nite in LAX/SNA. Told not without paying. That's wrong\",\n",
" \"@united Hi have a question re future Flight Booking Problems. DUB-JAC 29/9 JAC-LAX 8/10 LAX-DUB 13/10. I'm *G. What is checked bag allowance for JAC-LAX?\",\n",
" '@united now arrives LAX @ 8:03 am',\n",
" '@united I do I was on UA 495 LAX TO DEN - we are scheduled to land LAX @ 7:38 am - please rebook to Denver - best flight',\n",
" \"@united any chance you'll ever do CPUs on your JFK-LAX like @AmericanAir?\",\n",
" '@united Old school ride home to LAX from Houston #flyingRetro http://t.co/6asuwx3Kv0',\n",
" \"@united I think this is the best first class I have ever gotten!! Denver to LAX and it's wonderful!!!\",\n",
" '@united they did on a delta flight out of LAX which is why I should be compensated for my rental car there.',\n",
" '@united care less about the person - although he walked away while I was complaining. A man at 10p at LAX club. More.....',\n",
" '@united There is only one club at LAX - in terminal 7 across from gate 71',\n",
" '@united Joni did a great job on flight 5653 to LAX. Thanks for a great flight.',\n",
" '@united KOA-LAX should have fresh food service, right?',\n",
" \"@united I forgot that Intl flights out of LAX don't go from Intl Terminal! Easiest re-check in ever! woo!\",\n",
" '@united you should tell that to the staff at LAX then. I boarded group 5 at the very end of the queue as a Gold member. Thanks for nothing.',\n",
" '@united are my bags here yet? They were at Palm Springs airport. I was at LAX. How come I beat my bags here.',\n",
" \"@united Where are my bags!!! They weren't in LAX like your promised. 9 out of 10 things today were a mess today because of you.\",\n",
" '@united Ugh. My bags were sent to Palm Springs and not to LAX as promised. They better be at the hotel when I get there.',\n",
" '@united 12/13EWR-LAX UA1151 my seat/armrest broken discover after takeoff. flight full.FA filed report, who to chat with for partial refund?',\n",
" \"@united I'm on one of your 757-300 between JFK and LAX.When r u upgrading planes?Plane has no Screens,lousy seats And that's in UnitedFirst.\",\n",
" \"@United I'm hoping we don't miss our LAX - ITO connection. Not looking forward to being stuck at LAX overnight with our team....AGAIN!\",\n",
" '@united flight 86 LAX-IAD, back rows NOT CLEANED prior to boarding. How gross is that to find used tissues in your seat? Please.',\n",
" \"@united of course I need help. I've been DMing you ladies and gents all day. Your only solution is hope for the best and LAX.\",\n",
" '@SouthwestAir Flight 1700. (PHX TO LAX) Wheels stop. Glad to be home! Thanks to the professionals both up front and in the cabin!!!',\n",
" '@SouthwestAir another great trip! LAX 823 - LAS 3075- BNA. Thanks so much!!!',\n",
" '@SouthwestAir Seriously? FOUR DELAYS? Only takes 42 minutes to get to Vegas from @flyLAXairport & I have a connecting flight. #ridiculous',\n",
" '@SouthwestAir last week I flew from DAL to LAX. You got us in almost an hour early. Thank You.',\n",
" \"@SouthwestAir I start a new job tomorrow & you Cancelled Flight my flight (1629 BWI-LAX) and you really can't get me on another flight today ?!\",\n",
" '@SouthwestAir @ LAX is almost a mess. For some reason the express bag drop is slower than the full service line. http://t.co/ORY89eEGek',\n",
" '@SouthwestAir appreciate the reply, hopefully those LAX agents get the memo. Cheers!',\n",
" '@SouthwestAir any idea if there will be any \"spring sales\" soon for travel from Late Flight August to early September? Going from PA to LAX.',\n",
" '@SouthwestAir if you are giving tix to #DestinationDragons show would appreciate one or two for LA😄Flying from PHL to LAX on Friday',\n",
" '@southwestair Amazing view on the approach to LAX tonight. http://t.co/a68d5fULmH',\n",
" '@SouthwestAir So I am flying Chicago-LAX-PHX just to go spotting at LAX and PHX airports, then I am flying back to Chicago :)',\n",
" '@SouthwestAir think flight 1945 from BNA to LAX will get off the ground tomorrow??? #please #snowbama',\n",
" '@SouthwestAir any chance of adding LAX->JFK direct any time in the future?',\n",
" \"@JetBlue ....you haven't got me just yet\\n\\nCan a 1 way LAX-NYC cost me under 190?\",\n",
" '@JetBlue Or...\\n\\n....how about a 1way LAX-NYC(area) under 190?! Is this possible ?',\n",
" \"@JetBlue wondering if it's possible for my colleague and I to get on an earlier flight LAX>JFK tomorrow. Can you help?\",\n",
" \"@JetBlue I'm #MakingLoveOutofNothingAtAll on my #brandloveaffair to #LAX https://t.co/kdHRUF54sW\",\n",
" \"@JetBlue gr8 #Mint crew on #flight 123 to #LAX they're #Mintalicious #TrueBlueLove #ShelleyandMarcRock #travel #air\",\n",
" '@JetBlue service for baggage at JFK is incomprehensible. No employee knows where our luggage is frM flight 424 LAX to NYC ITS AN HR WAIT NOW',\n",
" '@jetblue rqstd upgrade to mint at LAX and was told no because i used points! Why turn down $1600 bcz I used points? #trueblue',\n",
" '@JetBlue received horrible customer service at LAX on 2/11. Reservation Cancelled Flighted without notification, despite having confirmation number.',\n",
" '@USAirways honest question - how is a 1-way ticket from Charlotte (your hub) to LAX (2nd biggest city in USA) almost $600?????',\n",
" '@USAirways Another great flight #FunFlightAttendants. Thanks for showing my dad wonderful customer service. #flt635 #LAX #PHX #SundayFunday',\n",
" '@usairways @AmericanAir LAX connect from term 6 to term 4. 55min layover due to delay. US755-AA2595. Is this realistic? Can 2595 be held?',\n",
" '@USAirways Everyone on Flight 669 from LAX to RDU enjoyed waiting an hour & a half in baggage claim for their bags just now',\n",
" '@USAirways your staff at LAX really messed up on this one. Failing to scan my suitcase tag.',\n",
" \"@USAirways doesn't seem likely bc your team failed to scan my bag in LAX and you recycle bagtag numbers that doesn't help #usairwaysfail\",\n",
" \"@USAirways what's happening with 1217 Phl to LAX? Now 3 hr delay. Poor communication!\",\n",
" \"@USAirways I have been doing that all day. Can't find my bag anywhere bc they're saying it was never scanned & technically never left LAX.\",\n",
" \"@USAirways Hey! I booked a flight (Isabelle Gramp, Boston to LAX), and it said that it charged my credit card but the transaction didn't go\",\n",
" '@USAirways I will be traveling from LAX to CLT to HTS, I have been rebooked for tomorrow due to the travel advisory.',\n",
" '@americanair thanks for no fresh food on my cross country flight and for making my connection so close No time to eat. TPA-DFW-LAX',\n",
" '@AmericanAir - Please find my bag!! In Singapore for three days already without my bag. Last known destination LAX Tag: 580815 Please help.',\n",
" '@AmericanAir None of the #LAX flights into #DFW have been Cancelled Flightled. Those landing before and after ours are fine. Completely arbitrary.',\n",
" '@americanair thanks for no fresh food on my cross country flight and for making my connection so close No time to eat. TPA-DFW-LAX',\n",
" '@AmericanAir - Please find my bag!! In Singapore for three days already without my bag. Last known destination LAX Tag: 580815 Please help.',\n",
" '@AmericanAir None of the #LAX flights into #DFW have been Cancelled Flightled. Those landing before and after ours are fine. Completely arbitrary.',\n",
" '@AmericanAir Yes I am. 2495/1170. RNO departure at 1229 on 2/25 w/connection at DFW to LGA. I can do the 1120am to LAX and then to JFK',\n",
" '@AmericanAir @USAirways #Boo! Wack ass terminal 6 @flyLAXairport. No food. No lounge. No Bueno!! Never again!!!',\n",
" \"@AmericanAir again, no special meal catered for me in F JFK-LAX. thankfully i'm on qantas the rest of way-i fear what youd NOT cater on that\",\n",
" '@AmericanAir LAX-OGG-LAX using hard earned miles and was given lousy service, faulty seats on both legs and damaged bag. Your customer care',\n",
" '@AmericanAir they were no where to be found at Midnight Last Night! I would think the agent in LAX could have relayed that info.Bag on flt',\n",
" \"@AmericanAir yes it is in Dulles and I need it delivered to the Embassy Suites in Herndon, VA. I'm still in Chicago from the fiasco in LAX\",\n",
" \"@AmericanAir another generic response guys? Cmon. You're terrible. How about an actual helpful person? Not all the rude employees at LAX\",\n",
" '@AmericanAir Cancelled Flightled flight from fresno then rebooked for LAX now flight Cancelled Flightled again and its midnight with no more hotel available???',\n",
" '@AmericanAir If SNA curfew causes diversion, do you provide transportation from LAX? On AA1237 now, pilot not sure if we have time.',\n",
" \"@AmericanAir - you broke my sick wife's luggage handle going from JFK to LAX...she had to drag her bag thru the airport! #customerservice\",\n",
" '@AmericanAir originating at SFO and going to LAX.',\n",
" \"@AmericanAir because your plane's toilet wasn't working and they needed gas. This is flight 1081 leaving Dulles going to LAX. Do some...\",\n",
" \"@AmericanAir it's always nice coming home but I wish you'd fly LAX-MAD and keep me away from Iberia 😜✈️ #GoingForGreat\",\n",
" '@AmericanAir SJC->LAX. After the fourth time, I gave up!',\n",
" '@AmericanAir the most stressful morning and still had to pay to check a bag. LAX is a madhouse with a lot of angry customers. Yikes',\n",
" '@AmericanAir thanks for the DM rescheduling. Unfortunately your operations process at LAX is chaos & the reps refused to print the ticket',\n",
" \"@AmericanAir at LAX & just got off the phone w/reservations. Every flight that'd get me to BOS before 11 am tmrw is apparently unavailable 😐\",\n",
" \"@AmericanAir at LAX and your service reps just hand out the 800 number to call. So that's not helpful.\",\n",
" '@AmericanAir Hi. I have KOA-LAX-PHL-ORD booked as a 1-way savr awrd. If I called to chnge it to KOA-LAX-PHX-ORD would I have to pay any fees',\n",
" \"@AmericanAir it was my friend. She shouldn't have been scheduled so close together at LAX then not reimbursed. It's costing her 12 hrs &$250\",\n",
" \"@AmericanAir when will tomorrow's flight Cancelled Flightlations at Dfw for AA flights be posted? We are on 2424 at 7am from LAX!\",\n",
" \"@AmericanAir usually raving about the service to LAX. Your nbr1 and helper can't figure out how to hang a coat and serve a drink. 6F\",\n",
" '@AmericanAir at JFK flight to Boston delayed 20 min waiting for catering they are boarding flight to LAX waiting for over and hour WTF?',\n",
" '@AmericanAir Flight AA1691 LAX to LAS closes too early and gate agents give us hassles #PatheticCX',\n",
" '@AmericanAir please help my flight appears Cancelled Flight 1503 from ft. Lauderdale to Dallas to LAX. Is there anything leaving from Mia or FLL? ThxU',\n",
" '@AmericanAir no one met flight 1081 to LAX to tell passengers where to go or what flights they were rebooked on #badmgmt #AmericanAirlines',\n",
" \"@AmericanAir flight 1081 from IAD to LAX sat for more than 3hrs because ground crew couldn't drive in snow #badmgmt #AmericanAirlines\",\n",
" '@AmericanAir @dotnetnate Two hour wait for EXPs as I sit on a JFK PHX flight because US computers are down. Any shot at an LAX flight?',\n",
" '@AmericanAir flight Cancelled Flighted out of LAX for tomorrow due to connection in DFW. Help please? we can go out of orange or burbank',\n",
" \"@AmericanAir I've been trying to change frm AA 2401 to LAX at 6:50am MONDAY morning then AA 2586 from LAX to FAT to flight AA 1359?#helpAA\",\n",
" '@AmericanAir a friend is having flight Cancelled Flightlations out of LAX to CMH on Feb 23. Anyway to help her? 800 number has been no help',\n",
" '@AmericanAir Love the new planes for the JFK-LAX run. Maybe one day I will be on one where the amenities all function. #NoCharge #Ever']"
]
},
"execution_count": 11,
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"import datetime\n",
"\n",
"def find(phrase, tweets):\n",
" #Naive full scan approach\n",
" \n",
" start = datetime.datetime.now()\n",
" \n",
" rtn = []\n",
" \n",
" for t in tweets:\n",
" if phrase in t:\n",
" rtn.append(t)\n",
" \n",
" \n",
" print('Find() elapsed time: ', (datetime.datetime.now()-start).total_seconds())\n",
" \n",
" return rtn\n",
"\n",
"\n",
"find('choppy landing', tweets)\n",
"\n",
"find('LAX', tweets)"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"That's pretty fast (about 3 ms!) But imagine if you had to run a million of such lookups, that would be 3000 seconds! At scale, small overheads add up. \n",
"\n",
"Now, we use our \"inverted indexing\" trick to make such searches faster.\n",
"\n",
"## Inverted Index\n",
"Next, we will try to do the same search with an inverted index. The indexing structure that we will use is a python dictionary."
]
},
{
"cell_type": "code",
"execution_count": 20,
"metadata": {},
"outputs": [
{
"name": "stdout",
"output_type": "stream",
"text": [
"build_index() elapsed time: 0.220968\n"
]
}
],
"source": [
"import string \n",
"\n",
"def build_index(tweets):\n",
" start = datetime.datetime.now()\n",
" \n",
" index = {}\n",
" \n",
" #some code to deal with punctuation\n",
" table = str.maketrans('', '', string.punctuation)\n",
"\n",
" for i,t in enumerate(tweets):\n",
" \n",
" words = t.translate(table).split() \n",
" \n",
" for w in words:\n",
" \n",
" if w not in index:\n",
" index[w] = set()\n",
" \n",
" index[w].add(i) #add a pointer to the relevant tweet\n",
" \n",
" print('build_index() elapsed time: ', (datetime.datetime.now()-start).total_seconds())\n",
" \n",
" return index\n",
"\n",
"index = build_index(tweets)"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"Notice that build_index is about a 100x slower than a single query. What does this mean? Basically, indexing is only valuable if you run a lot of queries! \n",
"\n",
"The next challenge is how to use an inverted index to answer general substring queries. In class, we showed how to do exact keyword lookup but the phrase 'choppy landing' is actually two words. This is actually not a problem, and we can use the inverted index to retrieve a set of candidates and then use the naive find method among just those candidates.\n",
"\n",
"So, let's write a new find function that can use this index:\n",
"* It splits the phrase into its constituent words\n",
"* Searches each word in the inverted index, finds a set of possibly relevant tweets (that match on a single word)\n",
"* Then double checks that set."
]
},
{
"cell_type": "code",
"execution_count": 24,
"metadata": {},
"outputs": [
{
"name": "stdout",
"output_type": "stream",
"text": [
"Find() elapsed time: 4e-06\n",
"Find() elapsed time: 2.9e-05\n"
]
},
{
"data": {
"text/plain": [
"[\"@united is doing musicians real dirty at LAX. I've never been blocked from getting on a flight with my bass.\",\n",
" \"@United I'm hoping we don't miss our LAX - ITO connection. Not looking forward to being stuck at LAX overnight with our team....AGAIN!\",\n",
" \"@united I forgot that Intl flights out of LAX don't go from Intl Terminal! Easiest re-check in ever! woo!\",\n",
" \"@united - you sure missed the mark on tonight's redeye from LAX to Chicago. What a mess! You can do better!\",\n",
" \"@VirginAmerica So excited for my first cross country flight LAX to MCO I've heard nothing but great things about Virgin America. #29DaysToGo\",\n",
" '@VirginAmerica LAX to EWR - Middle seat on a red eye. Such a noob maneuver. #sendambien #andchexmix',\n",
" '@VirginAmerica help, left expensive headphones on flight 89 IAD to LAX today. Seat 2A. No one answering L&F number at LAX!',\n",
" \"@USAirways I have been doing that all day. Can't find my bag anywhere bc they're saying it was never scanned & technically never left LAX.\",\n",
" '@JetBlue received horrible customer service at LAX on 2/11. Reservation Cancelled Flighted without notification, despite having confirmation number.',\n",
" '@united #LAX #sunrise UAL212 LAX-JFK']"
]
},
"execution_count": 24,
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"def find_index(phrase, tweets, index):\n",
" start = datetime.datetime.now()\n",
" \n",
" words = phrase.split()\n",
" \n",
" #find tweets that contain all words\n",
" candidates = None\n",
" for w in words: #for each words in the phrase\n",
" try:\n",
" \n",
" if candidates is None:\n",
" candidates = index[w] #return the set of tweets for w\n",
" else:\n",
" candidates = candidates.intersection(index[w])\n",
" \n",
" except KeyError:\n",
" return []\n",
" \n",
" candidate_tweets = [tweets[ref] for ref in candidates]\n",
" return find(phrase, candidate_tweets)\n",
"\n",
" print('find_index() elapsed time: ', (datetime.datetime.now()-start).total_seconds())\n",
"\n",
"find_index('choppy landing', tweets, index)\n",
"find_index('LAX', tweets, index)[:10]"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"In essence, you are paying a small upfront cost for greatly improved find performance (nearly a 1000x faster!). Speed is only aspect of search engine performance. We also like to support situations where a user mistypes a phrase. For example, if we mistype choppy landing:"
]
},
{
"cell_type": "code",
"execution_count": 25,
"metadata": {},
"outputs": [
{
"data": {
"text/plain": [
"[]"
]
},
"execution_count": 25,
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"find_index('chopy landing', tweets, index)"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"Our system returns nothing. Can we write a fast suggestion utility that can quickly identify typos."
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"## Did you mean? \n",
"So now we are going to write a utility that can identify mispelling and typos and suggest potential alternatives. So let's start off with a naive approach that simply finds the closest word in the index in terms of edit distance:"
]
},
{
"cell_type": "code",
"execution_count": 2,
"metadata": {},
"outputs": [
{
"data": {
"text/plain": [
"2"
]
},
"execution_count": 2,
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"import distance\n",
"\n",
"#distance.jaccard('a b', 'b c')\n",
"#distance.levenshtein('a b','b c') \"called edit distance\""
]
},
{
"cell_type": "code",
"execution_count": 29,
"metadata": {},
"outputs": [
{
"name": "stdout",
"output_type": "stream",
"text": [
"did_you_mean_naive() elapsed time: 0.992237\n"
]
},
{
"data": {
"text/plain": [
"'choppy'"
]
},
"execution_count": 29,
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"def did_you_mean_naive(word, index):\n",
" start = datetime.datetime.now()\n",
" \n",
" if word in index:\n",
" return word\n",
" \n",
" else:\n",
" \n",
" distances = [(distance.levenshtein(word, iw), iw) for iw in index]\n",
" distances.sort()\n",
" \n",
" print('did_you_mean_naive() elapsed time: ', (datetime.datetime.now()-start).total_seconds())\n",
" \n",
" return distances[0][1]\n",
" \n",
" \n",
"\n",
"did_you_mean_naive('chopy', index)"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"The suggestion utility runs much slower than the actual query!!! How do we fix this? We can use the same trick as before: a fast algorithm to find reasonable candidates and a slower algorithm to refine those candidates.\n",
"\n",
"In fact, we will use an inverted index again. Just this time over sub-sequences of letters and not words. The first thing that we are going to do is to calculate n-grams these are contiguous sub-sequences of letters."
]
},
{
"cell_type": "code",
"execution_count": 35,
"metadata": {},
"outputs": [
{
"data": {
"text/plain": [
"[('d', 'a'), ('a', 'v'), ('v', 'e')]"
]
},
"execution_count": 35,
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"#ngram\n",
"def find_ngrams(word, n):\n",
" return list(zip(*[word[i:] for i in range(n)]))\n",
"\n",
"find_ngrams('dave', 2)"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"Next, we are going to build a \"word\" index, an indexing structure that maps ngrams to words that contain them."
]
},
{
"cell_type": "code",
"execution_count": 40,
"metadata": {},
"outputs": [
{
"name": "stdout",
"output_type": "stream",
"text": [
"build_word_index() elapsed time: 0.195927\n"
]
}
],
"source": [
"def build_word_index(index, n):\n",
" start = datetime.datetime.now()\n",
" \n",
" word_index = {}\n",
" \n",
" for word in index:\n",
" ngrams = find_ngrams(word, n)\n",
" \n",
" for subseq in ngrams:\n",
" \n",
" if subseq not in word_index:\n",
" word_index[subseq] = set()\n",
" \n",
" word_index[subseq].add(word) #add a pointer to the relevant word\n",
" \n",
" print('build_word_index() elapsed time: ', (datetime.datetime.now()-start).total_seconds())\n",
" \n",
" return word_index\n",
"\n",
"word_index = build_word_index(index, 3)"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"We can use this word index to build a more sophisticated search:\n",
"* Only consider words that share a minimum number of ngrams with the lookup word"
]
},
{
"cell_type": "code",
"execution_count": 41,
"metadata": {},
"outputs": [
{
"name": "stdout",
"output_type": "stream",
"text": [
"did_you_mean_better() elapsed time: 0.003581\n"
]
},
{
"data": {
"text/plain": [
"'choppy'"
]
},
"execution_count": 41,
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"def did_you_mean_better(word, word_index, n, thresh=1):\n",
" start = datetime.datetime.now()\n",
" \n",
" candidate_words = {}\n",
" ngrams = find_ngrams(word, n)\n",
" \n",
" for ngram in ngrams:\n",
" candidates = word_index.get(ngram, set())\n",
" \n",
" for candidate in candidates:\n",
" candidate_words[candidate] = candidate_words.get(candidate,0) + 1\n",
" \n",
" \n",
" \n",
" distances = [(distance.levenshtein(word, iw), iw) for iw in candidate_words if candidate_words[iw] >= thresh]\n",
" distances.sort()\n",
" \n",
" print('did_you_mean_better() elapsed time: ', (datetime.datetime.now()-start).total_seconds())\n",
" \n",
" return distances[0][1]\n",
" \n",
"\n",
"did_you_mean_better('chopy', word_index, 3)"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"Notice how much faster this approach is!! 0.992237 secs v.s. 0.003581 seconds."
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"## Putting it all together\n",
"\n",
"Now, let's write the full program and try out some queries"
]
},
{
"cell_type": "code",
"execution_count": 44,
"metadata": {},
"outputs": [
{
"name": "stdout",
"output_type": "stream",
"text": [
"Searching for...choppy landing in 14640 tweets\n",
"Find() elapsed time: 8e-06\n",
"Found 1 matches\n",
"['@VirginAmerica pilot says we expect a choppy landing in NYC due to some gusty winds w/a temperature of about 5 degrees & w/the windchill -8']\n"
]
}
],
"source": [
"def find_final(phrase, \\\n",
" tweets, \\\n",
" index, \\\n",
" word_index, \\\n",
" n=3, \\\n",
" thresh=1):\n",
" print('Searching for...' + phrase + \" in \" + str(len(tweets)) + \" tweets\")\n",
" out = find_index(phrase, tweets, index)\n",
" print('Found ' + str(len(out)) + ' matches')\n",
" \n",
" if len(out) == 0:\n",
" for word in phrase.split():\n",
" if word not in index:\n",
" print('Did you mean: ' + did_you_mean_better(word, word_index, n, thresh) + ' instead of ' + word + '?')\n",
" else:\n",
" print(out)\n",
"\n",
"find_final('choppy landing', tweets, index, word_index)"
]
},
{
"cell_type": "code",
"execution_count": 45,
"metadata": {},
"outputs": [
{
"name": "stdout",
"output_type": "stream",
"text": [
"Searching for...chopy landing in 14640 tweets\n",
"Found 0 matches\n",
"did_you_mean_better() elapsed time: 0.002837\n",
"Did you mean: choppy instead of chopy?\n"
]
}
],
"source": [
"find_final('chopy landing', tweets, index, word_index)"
]
},
{
"cell_type": "code",
"execution_count": 46,
"metadata": {},
"outputs": [
{
"name": "stdout",
"output_type": "stream",
"text": [
"Searching for...choppy landig in 14640 tweets\n",
"Found 0 matches\n",
"did_you_mean_better() elapsed time: 0.032908\n",
"Did you mean: landing instead of landig?\n"
]
}
],
"source": [
"find_final('choppy landig', tweets, index, word_index)"
]
},
{
"cell_type": "code",
"execution_count": 47,
"metadata": {},
"outputs": [
{
"name": "stdout",
"output_type": "stream",
"text": [
"Searching for...LAX in 14640 tweets\n",
"Find() elapsed time: 2.8e-05\n",
"Found 95 matches\n",
"[\"@united is doing musicians real dirty at LAX. I've never been blocked from getting on a flight with my bass.\", \"@United I'm hoping we don't miss our LAX - ITO connection. Not looking forward to being stuck at LAX overnight with our team....AGAIN!\", \"@united I forgot that Intl flights out of LAX don't go from Intl Terminal! Easiest re-check in ever! woo!\", \"@united - you sure missed the mark on tonight's redeye from LAX to Chicago. What a mess! You can do better!\", \"@VirginAmerica So excited for my first cross country flight LAX to MCO I've heard nothing but great things about Virgin America. #29DaysToGo\", '@VirginAmerica LAX to EWR - Middle seat on a red eye. Such a noob maneuver. #sendambien #andchexmix', '@VirginAmerica help, left expensive headphones on flight 89 IAD to LAX today. Seat 2A. No one answering L&F number at LAX!', \"@USAirways I have been doing that all day. Can't find my bag anywhere bc they're saying it was never scanned & technically never left LAX.\", '@JetBlue received horrible customer service at LAX on 2/11. Reservation Cancelled Flighted without notification, despite having confirmation number.', '@united #LAX #sunrise UAL212 LAX-JFK', '@united our travel booked thru United group dept. Okc ticket agent less than willing to help with our connection in LAX.', '@USAirways Another great flight #FunFlightAttendants. Thanks for showing my dad wonderful customer service. #flt635 #LAX #PHX #SundayFunday', '@SouthwestAir another great trip! LAX 823 - LAS 3075- BNA. Thanks so much!!!', '@united Hi, Im flying SFO-LAX-SAL-CLO. My connecting time in LAX is 1h45m. Is it enough time? Do I have to collect my bag and recheck on AV?', \"@AmericanAir at LAX & just got off the phone w/reservations. Every flight that'd get me to BOS before 11 am tmrw is apparently unavailable 😐\", '@AmericanAir originating at SFO and going to LAX.', '@VirginAmerica just landed in LAX, an hour after I should of been here. Your no Late Flight bag check is not business travel friendly #nomorevirgin', '@USAirways your staff at LAX really messed up on this one. Failing to scan my suitcase tag.', '@VirginAmerica trying to add my boy Prince to my ressie. SF this Thursday @VirginAmerica from LAX http://t.co/GsB2J3c4gM', \"@AmericanAir at LAX and your service reps just hand out the 800 number to call. So that's not helpful.\", '@united they did on a delta flight out of LAX which is why I should be compensated for my rental car there.', '@USAirways honest question - how is a 1-way ticket from Charlotte (your hub) to LAX (2nd biggest city in USA) almost $600?????', '@VirginAmerica Can you find us a flt out of LAX that is sooner than midnight on Monday? That would be great customer service 😃', '@AmericanAir flight Cancelled Flighted out of LAX for tomorrow due to connection in DFW. Help please? we can go out of orange or burbank', '@united you should tell that to the staff at LAX then. I boarded group 5 at the very end of the queue as a Gold member. Thanks for nothing.', '@SouthwestAir appreciate the reply, hopefully those LAX agents get the memo. Cheers!', '@SouthwestAir @ LAX is almost a mess. For some reason the express bag drop is slower than the full service line. http://t.co/ORY89eEGek', \"@AmericanAir I've been trying to change frm AA 2401 to LAX at 6:50am MONDAY morning then AA 2586 from LAX to FAT to flight AA 1359?#helpAA\", '@AmericanAir - Please find my bag!! In Singapore for three days already without my bag. Last known destination LAX Tag: 580815 Please help.', '@AmericanAir they were no where to be found at Midnight Last Night! I would think the agent in LAX could have relayed that info.Bag on flt', \"@united of course I need help. I've been DMing you ladies and gents all day. Your only solution is hope for the best and LAX.\", '@usairways @AmericanAir LAX connect from term 6 to term 4. 55min layover due to delay. US755-AA2595. Is this realistic? Can 2595 be held?', \"@AmericanAir yes it is in Dulles and I need it delivered to the Embassy Suites in Herndon, VA. I'm still in Chicago from the fiasco in LAX\", \"@AmericanAir usually raving about the service to LAX. Your nbr1 and helper can't figure out how to hang a coat and serve a drink. 6F\", \"@USAirways doesn't seem likely bc your team failed to scan my bag in LAX and you recycle bagtag numbers that doesn't help #usairwaysfail\", '@USAirways Everyone on Flight 669 from LAX to RDU enjoyed waiting an hour & a half in baggage claim for their bags just now', '@VirginAmerica nervous about my flight from DC to LAX getting Cancelled Flightled tomorrow! Just sent you a DM to help me!', '@united Flight has been delayed for another hour so only have 24 mins to transit at LAX... Extremely unlikely I will make it!', '@United is offering to reroute my SFO flight to LAX. Might be geography class time.', '@AmericanAir None of the #LAX flights into #DFW have been Cancelled Flightled. Those landing before and after ours are fine. Completely arbitrary.', \"@AmericanAir another generic response guys? Cmon. You're terrible. How about an actual helpful person? Not all the rude employees at LAX\", '@jetblue rqstd upgrade to mint at LAX and was told no because i used points! Why turn down $1600 bcz I used points? #trueblue', '@united care less about the person - although he walked away while I was complaining. A man at 10p at LAX club. More.....', \"@USAirways Hey! I booked a flight (Isabelle Gramp, Boston to LAX), and it said that it charged my credit card but the transaction didn't go\", '@AmericanAir at JFK flight to Boston delayed 20 min waiting for catering they are boarding flight to LAX waiting for over and hour WTF?', \"@united I missed my connection already. Then I missed the next flight they put me on. Now I'm going to LAX instead of Hawaii. :(\", '@AmericanAir Cancelled Flightled flight from fresno then rebooked for LAX now flight Cancelled Flightled again and its midnight with no more hotel available???', '@SouthwestAir any idea if there will be any \"spring sales\" soon for travel from Late Flight August to early September? Going from PA to LAX.', '@AmericanAir If SNA curfew causes diversion, do you provide transportation from LAX? On AA1237 now, pilot not sure if we have time.', '@AmericanAir a friend is having flight Cancelled Flightlations out of LAX to CMH on Feb 23. Anyway to help her? 800 number has been no help', '@united Old school ride home to LAX from Houston #flyingRetro http://t.co/6asuwx3Kv0', '@VirginAmerica @VirginAtlantic I have just checked in flight to SFO from LAX & been told as Atlantic Flying Club Gold I get no benefits?!', \"@VirginAmerica lost my luggage 4 days ago on flight VX 112 from LAX to IAD & I'm calling every day, no response.Please give me back my stuff\", '@united Maybe be hiring your own ground staff at LAX when multiple gate agents tell you your baggage is loaded you expect it to be. HOPELESS', '@AmericanAir - Please find my bag!! In Singapore for three days already without my bag. Last known destination LAX Tag: 580815 Please help.', '@SouthwestAir think flight 1945 from BNA to LAX will get off the ground tomorrow??? #please #snowbama', '@united There is only one club at LAX - in terminal 7 across from gate 71', \"@united I think this is the best first class I have ever gotten!! Denver to LAX and it's wonderful!!!\", '@united Joni did a great job on flight 5653 to LAX. Thanks for a great flight.', '@united SF crew lack a lot of customer service, LAX employees are a lot better. Wonder why...', '@united are my bags here yet? They were at Palm Springs airport. I was at LAX. How come I beat my bags here.', '@AmericanAir Flight AA1691 LAX to LAS closes too early and gate agents give us hassles #PatheticCX', \"@united Where are my bags!!! They weren't in LAX like your promised. 9 out of 10 things today were a mess today because of you.\", \"@united UA flight 1247. SFO to LAX took my carry on at gate. I'm Group 2, overhead bins are empty\", '@AmericanAir None of the #LAX flights into #DFW have been Cancelled Flightled. Those landing before and after ours are fine. Completely arbitrary.', '@united maybemange the airline alittlebetter. Arrived at LAX and no GATE! #howisthatpossible always the same thing w/u', '@united Ugh. My bags were sent to Palm Springs and not to LAX as promised. They better be at the hotel when I get there.', \"@JetBlue I'm #MakingLoveOutofNothingAtAll on my #brandloveaffair to #LAX https://t.co/kdHRUF54sW\", '@united really fucked my day up Hilo to LAX 2hr30min delay because of software? missed connection, getting home 8hrs Late Flightr no upgrade nothin', \"@AmericanAir because your plane's toilet wasn't working and they needed gas. This is flight 1081 leaving Dulles going to LAX. Do some...\", '@SouthwestAir last week I flew from DAL to LAX. You got us in almost an hour early. Thank You.', \"@JetBlue gr8 #Mint crew on #flight 123 to #LAX they're #Mintalicious #TrueBlueLove #ShelleyandMarcRock #travel #air\", \"@AmericanAir it was my friend. She shouldn't have been scheduled so close together at LAX then not reimbursed. It's costing her 12 hrs &$250\", \"@AmericanAir when will tomorrow's flight Cancelled Flightlations at Dfw for AA flights be posted? We are on 2424 at 7am from LAX!\", '@united I took the exact same aircraft in to LAX 3 days ago. It fit, no problem. The agent today told some nonsense about a policy change', '@USAirways I will be traveling from LAX to CLT to HTS, I have been rebooked for tomorrow due to the travel advisory.', '@AmericanAir please help my flight appears Cancelled Flight 1503 from ft. Lauderdale to Dallas to LAX. Is there anything leaving from Mia or FLL? ThxU', \"@USAirways what's happening with 1217 Phl to LAX? Now 3 hr delay. Poor communication!\", '@united the person is currently bettween gates 71A and 73 in LAX', '@VirginAmerica Thanks for making my flight from LAX to JFK a nightmare by forcing me to check my carry on bag at the gate. (1)', '@united I need the phone number to baggage claim in LAX, my mom left her phone and someone called saying they would put it there but on', '@SouthwestAir if you are giving tix to #DestinationDragons show would appreciate one or two for LA😄Flying from PHL to LAX on Friday', '@AmericanAir no one met flight 1081 to LAX to tell passengers where to go or what flights they were rebooked on #badmgmt #AmericanAirlines', '@southwestair Amazing view on the approach to LAX tonight. http://t.co/a68d5fULmH', \"@AmericanAir flight 1081 from IAD to LAX sat for more than 3hrs because ground crew couldn't drive in snow #badmgmt #AmericanAirlines\", '@SouthwestAir So I am flying Chicago-LAX-PHX just to go spotting at LAX and PHX airports, then I am flying back to Chicago :)', '@SouthwestAir Flight 1700. (PHX TO LAX) Wheels stop. Glad to be home! Thanks to the professionals both up front and in the cabin!!!', '@AmericanAir @dotnetnate Two hour wait for EXPs as I sit on a JFK PHX flight because US computers are down. Any shot at an LAX flight?', '@AmericanAir Yes I am. 2495/1170. RNO departure at 1229 on 2/25 w/connection at DFW to LGA. I can do the 1120am to LAX and then to JFK', '@AmericanAir the most stressful morning and still had to pay to check a bag. LAX is a madhouse with a lot of angry customers. Yikes', '@JetBlue service for baggage at JFK is incomprehensible. No employee knows where our luggage is frM flight 424 LAX to NYC ITS AN HR WAIT NOW', '@VirginAmerica Flying LAX to SFO and after looking at the awesome movie lineup I actually wish I was on a long haul.', '@united now arrives LAX @ 8:03 am', '@united I do I was on UA 495 LAX TO DEN - we are scheduled to land LAX @ 7:38 am - please rebook to Denver - best flight', '@AmericanAir thanks for the DM rescheduling. Unfortunately your operations process at LAX is chaos & the reps refused to print the ticket']\n"
]
}
],
"source": [
"find_final('LAX', tweets, index, word_index)"
]
},
{
"cell_type": "code",
"execution_count": 48,
"metadata": {},
"outputs": [
{
"name": "stdout",
"output_type": "stream",
"text": [
"Searching for...LAXS in 14640 tweets\n",
"Found 0 matches\n",
"did_you_mean_better() elapsed time: 0.001225\n",
"Did you mean: LAX instead of LAXS?\n"
]
}
],
"source": [
"find_final('LAXS', tweets, index, word_index)"
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": []
}
],
"metadata": {
"kernelspec": {
"display_name": "Python 3",
"language": "python",
"name": "python3"
},
"language_info": {
"codemirror_mode": {
"name": "ipython",
"version": 3
},
"file_extension": ".py",
"mimetype": "text/x-python",
"name": "python",
"nbconvert_exporter": "python",
"pygments_lexer": "ipython3",
"version": "3.7.4"
}
},
"nbformat": 4,
"nbformat_minor": 2
}
This source diff could not be displayed because it is too large. You can view the blob instead.
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 sign in to comment