-
Notifications
You must be signed in to change notification settings - Fork 92
/
example.py
13 lines (11 loc) · 679 Bytes
/
example.py
1
2
3
4
5
6
7
8
9
10
11
12
13
from representations.sequentialembedding import SequentialEmbedding
"""
Example showing how to load a series of historical embeddings and compute similarities over time.
Warning that loading all the embeddings into main memory can take a lot of RAM
"""
if __name__ == "__main__":
fiction_embeddings = SequentialEmbedding.load("embeddings/eng-fiction-all_sgns", range(1950, 2000, 10))
time_sims = fiction_embeddings.get_time_sims("lesbian", "gay")
print "Similarity between gay and lesbian drastically increases from 1950s to the 1990s:"
for year, sim in time_sims.iteritems():
print "{year:d}, cosine similarity={sim:0.2f}".format(year=year,sim=sim)