Skip to content

Instantly share code, notes, and snippets.

#import data
train_pos = pd.read_csv("/Users/nicolas/work/Dataset/imdb_sentiment/imdb_sentiment_text/train/pos/train_pos.txt", engine='python', sep='delimiter', header=None)
train_neg = pd.read_csv("/Users/nicolas/work/Dataset/imdb_sentiment/imdb_sentiment_text/train/neg/train_neg.txt", engine='python', sep='delimiter', header=None)
test_pos = pd.read_csv("/Users/nicolas/work/Dataset/imdb_sentiment/imdb_sentiment_text/test/pos/test_pos.txt", engine='python', sep='delimiter', header=None)
test_neg = pd.read_csv("/Users/nicolas/work/Dataset/imdb_sentiment/imdb_sentiment_text/test/neg/test_neg.txt", engine='python', sep='delimiter', header=None)
#transform data
train_pos[1] = 1
train_neg[1] = 0
test_pos[1] = 1