def get_dist()

in svhn_based_exp/utils.py [0:0]

• 6 lines of code
• 3 McCabe index (conditional complexity)

def get_dist(feat_set_1, feat_set_2, dist_type="cos"):
    if dist_type == "l2":
        dist = euclidean_distances(feat_set_1, feat_set_2)
    elif dist_type == "cos":
        dist = cosine_distances(feat_set_1, feat_set_2)
    return dist