ZurichNLP/unsupervised-semantic-diff

1"""2Get similarities of similar and dissimilar pairs.3The values are used for normalizing the colors in the visualization.4"""5from app import load_pipeline6from recognizers import DiffAlign, DiffDel7 8similar_pair = ("Hello!", "Hi!")9dissimilar_pair = ("Hello!", "asdf")10 11pipeline = load_pipeline()12diff_align = DiffAlign(pipeline=pipeline)13diff_del = DiffDel(pipeline=pipeline)14 15print("Similar pair:")16print(diff_align.predict(*similar_pair).min)17print(diff_del.predict(*similar_pair).min)18 19print("Dissimilar pair:")20print(diff_align.predict(*dissimilar_pair).max)21print(diff_del.predict(*dissimilar_pair).max)22