peet-droid

from sklearn.metrics import confusion_matrix
import matplotlib.pyplot as plt
import seaborn as sn

cfm = confusion_matrix(y_true, y_pred)

classes = ['door_close', 'door_open', 'door_stop', 'unknown']

df_cfm = pd.DataFrame(cfm, index = classes, columns = classes)
plt.figure(figsize = (10,7))