Obscure analogy generator from word embeddings

300 IQ Analogies.py

def random_analogies():
	while(1):
		r = lambda: random.randint(0,len(embeddings)-1)
		x = list(embeddings.keys())[r()]
		y = list(embeddings.keys())[r()]
		z = list(embeddings.keys())[r()]
		restart = False
		for w in [x, y, z]:
			if not w.isalpha():
				restart = True
		if restart:
			continue
		vx = embeddings[x]
		vy = embeddings[y]
		vz = embeddings[z]
		vector = vx - vy + vz  # SOooo cool! :)
		word, cosine_sim = get_nearest(vector, embeddings)[0]
		if not word.isalpha():
			continue
		if cosine_sim > 1e-4:
			continue
		if word not in {x, y, z}:
		     break
	print(x, "is to", y, "as", z, "is to", word, 'cosine sim:', cosine_sim)
	return word


def x_is_to_y_as_z_is_to(x, y, z, embeddings):
    vx = embeddings[x.lower()]
    vy = embeddings[y.lower()]
    vz = embeddings[z.lower()]
    vector = vy - vx + vz
    word, cosine_sim = get_nearest(vector, embeddings)[0]
    print(x, "is to", y, "as", z, "is to", word, 'cosine sim:', cosine_sim)
    return word