# pour un modèle, des données, et un jeu d'étiquette Y def introspectionModel(classif, data, Y, filename=None): # introspection: qu'est ce qui est associé à chaque classe de données plt.figure() plt.imshow(classif.coef_, interpolation='nearest') localLabs = [data['labMeaning'][y-1] for y in np.unique(Y)] plt.yticks(range(len(localLabs)),localLabs) rgb = [color+' '+strength for color in ['red', 'green', 'blue'] for strength in ['(low)', '(med)', '(high)']] plt.xticks(np.arange(len(rgb))-0.5,rgb, rotation=45) plt.vlines(np.array([2.5,5.5]), -1, len(localLabs),linestyles='dashed') if filename != None: plt.savefig(filename)