Skip to content
Snippets Groups Projects
Commit 204ad6c5 authored by ale's avatar ale
Browse files

generate full sequences; fix padding error in tuple generation

parent 40df4c92
No related branches found
No related tags found
No related merge requests found
......@@ -79,6 +79,17 @@ class MarkovModel(object):
if off > r:
return self._i2hash[value]
def generate_sequence(self, prev, n, count):
if len(prev) < n:
prev = ([None] * (n - len(prev))) + prev
out = []
for i in xrange(count):
song = self.suggest(prev)
prev.pop(0)
prev.append(song)
out.append(song)
return out
def main():
parser = optparse.OptionParser()
......
......@@ -58,7 +58,7 @@ class PlayLog(Base):
if plog.prev:
hashes = plog.prev.split(',')[:n]
if len(hashes) < n:
hashes.extend([None] * (n - len(hashes)))
hashes = ([None] * (n - len(hashes))) + hashes
else:
hashes = [None] * n
yield (plog.sha1, hashes)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment