#! /usr/bin/python import pickle from nltk import * from nltk.corpus import PlaintextCorpusReader training_corpus_root = "../data/training_set" dev_corupus_root = "../data/development_set" test_corupus_root = "../data/test_set" convote_training = PlaintextCorpusReader(training_corpus_root, '.*') convote_dev = PlaintextCorpusReader(dev_corupus_root, '.*') convote_test = PlaintextCorpusReader(test_corupus_root, '.*') bigram = bigrams(convote_training.tokenized()) outfile = open('../data/training_bigrams', 'w') pos_tokens = pos_tag(convote_training.tokenized()) prep_tokens = [] for (word, pos) in pos_tokens: if(pos == 'IN'): prep_tokens.append(word + '|' + pos) else: prep_tokens.append(pos) trigram = nltk.trigrams(prep_tokens) trigram_file = open('../data/pos_trigrams', 'w') pickle.dump(bigram, outfile) pickle.dump(trigram, trigram_file)