repairList = [None]*10
for i in range(len(matrix)):

    tal = np.argmax(matrix[i])
    repairList[tal] = i
    for j in range(len(matrix)):
        matrix[j][tal] = 0

print(repairList)
for i in range(len(y_pred)):
    y_pred[i] = repairList[y_pred[i]]