def get_one_hot_matrix(vocab):
    one_hot_matrix = np.zeros((len(vocab), len(vocab)))
    np.fill_diagonal(one_hot_matrix, 1)
    return one_hot_matrix
def get_one_hot_matrix(vocab):
    one_hot_matrix = np.zeros((len(vocab), len(vocab)))
    np.fill_diagonal(one_hot_matrix, 1)
    return one_hot_matrix