Skip to content

Instantly share code, notes, and snippets.

What would you like to do?
Markov model based on youtube comments
# for more info check out
# made to be run in the ipython console
import urllib, urllib2, time, random
import simplejson as json
def fetch_url(url, get=None, post=None):
user_agent = 'Andrei Olariu\'s Web Mining for Dummies'
headers = {'User-Agent': user_agent}
if get:
data = urllib.urlencode(get)
url = "%s?%s" % (url, data)
req = urllib2.Request(url, post, headers)
response = urllib2.urlopen(req).read()
response = json.loads(response)
except Exception, e:
print 'error in reading %s: %s' % (url, e)
return None
return response
# fetch comments for a youtube video (given a video id) by doing repeated
# api calls (one call returnes up to 50 comments)
def fetch_comments(yid, maxcount=1000):
url = '' % yid
COUNT = 50
values = {
'alt': 'json',
'max-results': COUNT,
results = []
for i in range(1, maxcount, COUNT):
values['start-index'] = i
data = fetch_url(url, get=values)
if data and 'feed' in data and 'entry' in data['feed'] and \
len(data['feed']['entry']) > 0:
results.extend([c['content']['$t'] for c in data['feed']['entry']])
return results
# builds the markov model
# every state is defined as a pair of words
# state (word[k], word[k+1]) depends on state (word[k-1], word[k])
def add_to_markov(markov, words):
if len(words) < 3:
if words[0] not in markov:
markov[words[0]] = {}
if words[1] not in markov[words[0]]:
markov[words[0]][words[1]] = {}
if words[2] not in markov[words[0]][words[1]]:
markov[words[0]][words[1]][words[2]] = 0
markov[words[0]][words[1]][words[2]] += 1
add_to_markov(markov, words[1:])
# given a state (aka a pair of words (word1, word2)),
# find the next state (aka another pair of words (word2, word3))
def get_next(markov, word1, word2):
if word1 not in markov or word2 not in markov[word1]:
return None
total = sum([c for c in markov[word1][word2].itervalues()])
choose = random.randint(1, total)
for w, c in markov[word1][word2].iteritems():
choose -= c
if choose <= 0:
return w
# given a starting state, find future states in a recursive way
def get_phrase(markov, word1, word2, limit=50):
if limit == 0:
return ''
word3 = get_next(markov, word1, word2)
if not word3:
return ''
return '%s %s' % (word3, get_phrase(markov, word2, word3, limit - 1))
# given a sentence beginning, add words using the markov model
# the starting state is given by the last 2 words in the sentence,
# all other words are not used
def talk(markov, start):
words = re.findall(r'\w+', start.lower())
if len(words) < 2:
return None
return '%s %s' % (start, get_phrase(markov, words[-2], words[-1]))
# get comments for a video
yid = 'UzxYlbK2c7E' # machine learning
# use 'qrO4YZeyl0I' for lady gaga
comments = fetch_comments(yid)
# split comments in phrases
texts = []
r = re.compile("[.!?;]")
for c in comments:
for line in c.splitlines():
# split phrases into words and build the markov model
markov = {}
for t in texts:
remove_first = t.startswith('@') # remove usernames
t = t.lower()
words = re.findall(r'\w+', t)
if remove_first:
words = words[1:]
add_to_markov(markov, words)
# have fun
print talk(markov, 'i like')
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
You can’t perform that action at this time.