config.py

import numpy as np

LANDMARK_IDS = np.array((61, 292, 0, 17, 50, 280, 48, 4, 289, 206, 426, 133, 130, 159, 
                145, 362, 359, 386, 374, 122, 351, 46, 105, 107, 276, 334, 336))

EDGE_IDS = ((0, 2), (0, 3), (1, 2), (1, 3), (7, 6), (7, 8), (6, 4), (8, 5),
            (6, 9), (9, 0), (4, 0), (8, 10), (10, 1), (7, 19), (7, 20), (7, 0),
            (7, 1), (19, 23), (19, 14), (23, 22), (22, 21), (21, 12), (12, 13),
            (12, 14), (11, 13), (11, 14), (14, 4), (20, 26), (26, 25), (25, 24),
            (24, 16), (16, 17), (16, 18), (15, 17), (15, 18), (18, 20), (18, 5))

ANGLE_IDS = ((2, 0, 3), (0, 2, 1), (6, 7, 8), (9, 7, 10), (0, 7, 1), (1, 5, 8),
             (1, 10, 8), (13, 12, 14), (21, 22, 23), (6, 19, 23))

FEATURE_IDS = [LANDMARK_IDS[np.array(triplet)] for triplet in ANGLE_IDS]

KNN_CONFIG = {
    "n_neighbors": 1,
}

SVM_CONFIG = {
    "C": 275,
    "gamma": "scale",
    "kernel" : "rbf",
}