Korzystam z korpusu nltk
biblioteki, movie_reviews
który zawiera dużą liczbę dokumentów. Moim zadaniem jest uzyskanie predykcyjnej wydajności tych przeglądów z wstępnym przetwarzaniem danych i bez wstępnego przetwarzania. Ale nie ma problemu, w listach documents
i documents2
mam te same dokumenty i muszę przetasować je w celu utrzymania tej samej kolejności w obu listach. Nie mogę tasować ich osobno, ponieważ za każdym razem, gdy tasuję listę, otrzymuję inne wyniki. Dlatego muszę od razu przetasować je w tej samej kolejności, ponieważ na koniec muszę je porównać (zależy to od kolejności). Używam Pythona 2.7
Przykład (w rzeczywistości łańcuchy są tokenizowane, ale nie są względne):
documents = [(['plot : two teen couples go to a church party , '], 'neg'),
(['drink and then drive . '], 'pos'),
(['they get into an accident . '], 'neg'),
(['one of the guys dies'], 'neg')]
documents2 = [(['plot two teen couples church party'], 'neg'),
(['drink then drive . '], 'pos'),
(['they get accident . '], 'neg'),
(['one guys dies'], 'neg')]
Potrzebuję tego wyniku po przetasowaniu obu list:
documents = [(['one of the guys dies'], 'neg'),
(['they get into an accident . '], 'neg'),
(['drink and then drive . '], 'pos'),
(['plot : two teen couples go to a church party , '], 'neg')]
documents2 = [(['one guys dies'], 'neg'),
(['they get accident . '], 'neg'),
(['drink then drive . '], 'pos'),
(['plot two teen couples church party'], 'neg')]
Mam ten kod:
def cleanDoc(doc):
stopset = set(stopwords.words('english'))
stemmer = nltk.PorterStemmer()
clean = [token.lower() for token in doc if token.lower() not in stopset and len(token) > 2]
final = [stemmer.stem(word) for word in clean]
return final
documents = [(list(movie_reviews.words(fileid)), category)
for category in movie_reviews.categories()
for fileid in movie_reviews.fileids(category)]
documents2 = [(list(cleanDoc(movie_reviews.words(fileid))), category)
for category in movie_reviews.categories()
for fileid in movie_reviews.fileids(category)]
random.shuffle( and here shuffle documents and documents2 with same order) # or somehow