Skip to content

Instantly share code, notes, and snippets.

@MarynaLongnickel

MarynaLongnickel/data_prep.py

Created June 16, 2018 21:06
Show Gist options
  • Save MarynaLongnickel/f0f9f891f2adf24274a61f151fe2eed5 to your computer and use it in GitHub Desktop.
Save MarynaLongnickel/f0f9f891f2adf24274a61f151fe2eed5 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
data_prep.py
data = pd.DataFrame(columns=['text', 'sentiment'])
for id in movie_reviews.fileids():
text = ' '.join(movie_reviews.words(id))
sentiment = 1 if movie_reviews.categories(id) == 'pos' else 0
data = data.append(pd.DataFrame({'text': text,'sentiment': sentiment}, index=[0]))
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment