classification with centroids.py

import pickle
import numpy as np
import pandas as pd
import seaborn as sns
import matplotlib.pyplot as plt
from utils import load_classes_from_folder
from process_signal import autocorrelation, calculate_lsp, get_energy, \
    get_edges, run_whole_signal, in_region,\
    euclidian_distance, dtw, get_new_matrix, \
    get_global_distance

pf = 146
ws = 80
wa = ws
p = 16

k1 = .0001
k2 = .0003

gender = "male"
train_folder = "./corpus_digitos/training-examples/" + gender
test_folder = "./corpus_digitos/test-examples/" + gender

classes = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11]

centroids = pd.read_pickle(r'centroids/centroids.pickle')

tests = load_classes_from_folder(test_folder, extension=".wav"); tests = np.array(tests)

confusion_matrix = np.zeros((len(classes), len(classes)))

test_c = 0
for test_class in tests:

    for test in test_class:
        test_signal = test[0]
        # Calculate the lsfs
        lsfs_test, _, _ = run_whole_signal(test_signal, ws, wa, pf, k1, k2, p, to_plot=False)

        # Go through each centroid and calculate the distance
        distances = []
        for key in centroids:
            if key == 10: continue

            # Get the lsf of the centroid
            lsf_centroid = centroids[key]

            # Calculate the distance between the two signals
            dtw_matrix = dtw(lsf_centroid, lsfs_test, p, to_plot=False)
            min_matrix = get_new_matrix(dtw_matrix, to_plot=False)
            distance, new_matrix = get_global_distance(min_matrix, to_plot=False)
            distances = np.append(distances, distance)

        # Get the index of the minimum distance
        index = np.argmin(distances)
        predicted_class = classes[index]

        confusion_matrix[classes[test_c] - 1, predicted_class - 1] += 1

    test_c += 1

print(confusion_matrix)
# Get accuracy from confusion matrix
accuracy = np.trace(confusion_matrix)/np.sum(confusion_matrix)
print("Accuracy:", accuracy)

ax = sns.heatmap(confusion_matrix, annot=True, cmap='Blues')

ax.set_title('Confusion Matrix\n\n')
ax.set_xlabel('Predicted')
ax.set_ylabel('Actual')

## Ticket labels - List must be in alphabetical order
ax.xaxis.set_ticklabels(['1','2', '3', '4', '5', '6', '7', '8', '9', '0', 'Z'])
ax.yaxis.set_ticklabels(['1','2', '3', '4', '5', '6', '7', '8', '9', '0', 'Z'])

## Display the visualization of the Confusion Matrix.
plt.show()