CustomCorpus.py

__author__ = 'Darko'

import json
from gensim import corpora, models, similarities
from functools import partial
import codecs

pp = partial(json.dumps, indent=1)

documents = ["Human machine interface for lab abc computer applications",
              "A survey of user opinion of computer system response time",
              "The EPS user interface management system",
              "System and human system engineering testing of EPS",
              "Relation of user perceived response time to error measurement",
              "The generation of random binary unordered trees",
              "The intersection graph of paths in trees",
              "Graph minors IV Widths of trees and well quasi ordering",
              "Graph minors A survey"]

def createCorpus():
    stoplist = set('for a of the and to in'.split())
    texts = [[word for word in document.lower().split() if word not in stoplist]
                for document in documents]
    all_tokens = sum(texts, [])
    tokens_once = set(word for word in set(all_tokens) if all_tokens.count(word) == 1)
    texts = [[word for word in text if word not in tokens_once]
          for text in texts]
    dictionary = corpora.Dictionary(texts)
    dictionary.save('resources/deerwester.dict')
    # print(dictionary.token2id)
    new_doc = "Human computer interface"
    new_vec = dictionary.doc2bow(new_doc.lower().split())
    print(new_vec)

def file_to_doc(filename="Temp"):
    name = 'resources/{0}.txt'.format(filename)
    with codecs.open(name, "r", encoding="utf-8") as f:
        text = f.read()
        text_arr = text.split()
        list_arr = []
        list_arr.append(text_arr)
        return list_arr

def create_document_from_file(filename="Temp"):
    docs = file_to_doc(filename)[0]
    return docs
    # return dictionary



def add_to_dictionary(dictionary=None, filename="Temp"):
    docs = file_to_doc(filename)
    dictionary.add_documents(docs)
    return dictionary

def dictionary_to_vector(dictionary=None):
    return dictionary.token2id

def main():
    print("Corpus")
    # createCorpus()
    # create_dictionary_from_file("Temp")

if __name__ == "__main__":
    main()