from lsa import LSA
if __name__ == '__main__':
documents = [line.rstrip() for line in open('temp/all_book_titles.txt')]
stopwords = set(line.rstrip() for line in open('temp/stopwords.txt')).union({
'introduction', 'edition', 'series', 'application',
'approach', 'card', 'access', 'package', 'plus', 'etext',
'brief', 'vol', 'fundamental', 'guide', 'essential', 'printed',
'third', 'second', 'fourth'
})
model = LSA(stopwords)
model.fit(documents)
model.transform_plot()
