Merge pull request #13423 from reyoung/feature/fix_sentimental_dataset

Fix sentiment dataset
6 years ago · d9297c1231
parent 61490528da 480c7c4ee3
commit d9297c1231
1 changed files with 1 additions and 1 deletions
--- a/python/paddle/dataset/sentiment.py
+++ b/python/paddle/dataset/sentiment.py
@ -67,7 +67,7 @@ def get_word_dict():
        for field in movie_reviews.fileids(category):
            for words in movie_reviews.words(field):
                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])
    for index, word in enumerate(words_sort_list):
        words_freq_sorted.append((word[0], index))