pred = proba.argmax(0) # max colonne par colonne