Skip to content

Instantly share code, notes, and snippets.

@vivekpadia70

vivekpadia70/imdb_lstm.py

Last active April 8, 2020 11:19
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save vivekpadia70/98f207ff1e2ee3ffacca66281ecf8b17 to your computer and use it in GitHub Desktop.
Save vivekpadia70/98f207ff1e2ee3ffacca66281ecf8b17 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
imdb_lstm.py
df = pd.read_csv("/home/aubergine/Downloads/imdb-dataset-of-50k-movie-reviews/IMDB Dataset.csv")
train_set = 45000
max_len_text = 2000
df['sentiment'] = df['sentiment'].replace('positive', 1)
df['sentiment'] = df['sentiment'].replace('negative', 0)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment