Vector Space Model Python