m53297601
/
Paddle
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Fix sentiment dataset

Browse Source
add-doc-readme
yuyang 7 years ago
parent 3fbfcd9c4f
commit 480c7c4ee3
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File

2
python/paddle/dataset/sentiment.py
Unescape View File

@ -67,7 +67,7 @@ def get_word_dict():
for field in movie_reviews.fileids(category): for field in movie_reviews.fileids(category):
for words in movie_reviews.words(field): for words in movie_reviews.words(field):
word_freq_dict[words] += 1 word_freq_dict[words] += 1
words_sort_list = six.iteritems(word_freq_dict) words_sort_list = list(six.iteritems(word_freq_dict))
words_sort_list.sort(cmp=lambda a, b: b[1] - a[1]) words_sort_list.sort(cmp=lambda a, b: b[1] - a[1])
for index, word in enumerate(words_sort_list): for index, word in enumerate(words_sort_list):
words_freq_sorted.append((word[0], index)) words_freq_sorted.append((word[0], index))

Write Preview
Loading…
Cancel
Save