U Ùfü ã@s‚ddlZddlZddlZzddlmZddlZWnek rNe  d¡YnXddl m Z ddl m Z ddl mZd d d „ZdS)éNzmatplotlib could not be loaded!é)Úlabels)Úcolorsé)Ú convert_nameÚpcaçð?Tc Csl|dkr dd„t|jdƒDƒ}t|||ƒ}|dkrD| d¡}d}n|dd…|f}||}t|ƒtkrˆ|dkrˆtj d¡}|  |¡} n(t |d ƒr¦|jddkr¦|} n t d |ƒt j | dd…d f| dd…df|tj|d d t  d ¡t  ¡} | jd|dd| j d¡t  ¡ dd¡| j ¡ t  ¡j ¡¡} | j | jdd¡|  d¡|rht  ¡dS)a¡ Use the SHAP values as an embedding which we project to 2D for visualization. Parameters ---------- ind : int or string If this is an int it is the index of the feature to use to color the embedding. If this is a string it is either the name of the feature, or it can have the form "rank(int)" to specify the feature with that rank (ordered by mean absolute SHAP value over all the samples), or "sum()" to mean the sum of all the SHAP values, which is the model's output (minus it's expected value). shap_values : numpy.array Matrix of SHAP values (# samples x # features). feature_names : None or list The names of the features in the shap_values array. method : "pca" or numpy.array How to reduce the dimensions of the shap_values to 2D. If "pca" then the 2D PCA projection of shap_values is used. If a numpy array then is should be (# samples x 2) and represent the embedding of that values. alpha : float The transparency of the data points (between 0 and 1). This can be useful to the show density of the data points when using a large dataset. NcSsg|]}tdt|ƒ‘qS)ZFEATURE)rÚstr)Ú.0Úi©r ú;/tmp/pip-target-lpfmz8o1/lib/python/shap/plots/embedding.pyÚ +sz"embedding_plot..rzsum()zsum(SHAP values)rrÚshapezUnsupported embedding method:r)ÚcZcmapÚalphaZ linewidthÚoffzSHAP value for é )ÚsizeFg@égffffffæ?é ) ÚrangerrÚsumÚtyper ÚsklearnÚ decompositionZPCAZ fit_transformÚhasattrÚprintÚplZscatterrZred_blueZaxisZcolorbarZ set_labelÚoutlineZ set_visibleZgcfZset_size_inchesZaxZget_window_extentZ transformedZdpi_scale_transÚinvertedZ set_aspectÚheightZ set_alphaÚshow) ÚindZ shap_valuesZ feature_namesÚmethodrr"ZcvalsÚfnamerZembedding_valuesÚcbZbboxr r r Úembedding_plots@     þ   r')NrrT)ÚnumpyÚnprÚwarningsZmatplotlib.pyplotZpyplotrZ matplotlibÚ ImportErrorÚwarnÚrrÚcommonrr'r r r r Ús