diff --git a/mitdeeplearning/lab1.py b/mitdeeplearning/lab1.py index 19d448a3..1b826dfe 100644 --- a/mitdeeplearning/lab1.py +++ b/mitdeeplearning/lab1.py @@ -7,7 +7,7 @@ from IPython.display import Audio - +#current working directory cwd = os.path.dirname(__file__) def load_training_data(): @@ -17,7 +17,7 @@ def load_training_data(): return songs def extract_song_snippet(text): - pattern = '(^|\n\n)(.*?)\n\n' + pattern = '(^|\n\n\n)(.*?)\n\n\n' search_results = re.findall(pattern, text, overlapped=True, flags=re.DOTALL) songs = [song[1] for song in search_results] print("Found {} songs in text".format(len(songs)))