if __name__ == '__main__':
ad1 = AnalizedDocument.from_file("./data/test_doc")
ad2 = AnalizedDocument.from_file("./data/test_doc2")
print "Text Similarity for ad1 and ad2"
comparsion = ad1.compare(ad2)
print "Result: ", comparsion.similarity_result
print "Important categories:"
for cat in comparsion.important_categories:
print cat
print_sorted_dict(ad1.categories_membership, "test_doc")
print_sorted_dict(ad2.categories_membership, "test_doc2")
print "Two images:"
im1 = Image.open("./data/red.jpg")
im2 = Image.open("./data/pom.png")
print compare(im1, im2)
print "Many images:"
first_group = glob("./data/test_a/*")
sec_group = glob("./data/test_b/*")
creator = lambda x: (x, Image.open(x))
comparsion = compare_many(map(creator, first_group), map(creator, sec_group))
print "Result: ", comparsion.similarity_result
print "Pictures with each other:"
for pair, result in comparsion.image_pairs_results.items():
first, sec = map(lambda x: path.split(x)[-1], pair)
print "{} : {} is {}".format(first, sec, result)
def get_image_similatiry(self, web_page):
return compare_many(self.pictures, web_page.pictures)