o 5tiM$@sFddlZddlmZddlZddlZGdddZgddfddZ dS)Nc@s2eZdZdZd ddZddZd d Zd d ZdS) ComplexRadaraCreate a complex radar chart with different scales for each variable Args: fig (`matplotlib.figure`) : A matplotlib figure object to add the axes on. variables (`list`) : a list of variables to. plot ranges (`list` of `tuples`): A list of ranges (min, max) for each variable n_ring_levels (`int): Number of ordinate or ring levels to draw. Default: 5. show_scales (`bool`): Indicates if we the ranges for each variable are plotted. Default: True. format_cfg (`dict`): A dictionary with formatting configurations. Default: None. Returns: `matplotlib.figure.Figure`: a radar plot. TNcs|_tdddt|}fddtt|dD}|D]} | d| d| d q t|D]c\} } | dksB| dkrDdn| d} | j || tj || |jd d } d d| D} d | d<| j | f| || djd\}}| j || | j d d| jdd|dur| gq6|ddD]} | j d| j dqttj||df_|_|d_|d_d_jjjd#ijdjj djd#ijdjjjd#ijdjdjdjj||d\}}ddjD}fdd|D}jj|fijdt j|D]3\}}|dkrC|!dq3|dkrT|dkrT|!dq3|dkr`|!dq3|!dq3jj"d jd!d"dS)$Nrihgv@cs4g|]}jgdfdd|djdqS))皙?r?rTzaxes{})polarlabel axes_args)add_axesformat format_cfg).0ifigselfJ/home/ubuntu/.local/lib/python3.10/site-packages/evaluate/visualization.py s&z)ComplexRadar.__init__..NT incl_endpoint)numendpointcSsg|] }dt|dqS)z{})r roundr xrrrr2s)labelsanglergrid_tick_lbls_argsrF)visible rad_ln_args outer_ring angle_ln_argsoff )rcSsg|]}|qSr)get_text)r trrrr[sc s.g|]}dtj|jdjddqS) theta_tick_lbls_txt_wraptheta_tick_lbls_brk_lng_wrds)break_long_words)jointextwrapwrapr )r r)rrrr\stheta_tick_lblscenterleftrightboththeta_tick_lbls_pad)axispadr)#r nparangelenrangeset_theta_zero_locationset_theta_direction set_axisbelow enumerateset_ylimlinspace set_rgridsspines set_visiblegridset_yticklabelspatchxaxisdeg2radr_r rangesaxax1 plot_counteryaxissetr8 set_zorderset_thetagridsget_xticklabelsset_xticklabelszipset_ha tick_params)rr variablesrM n_ring_levels show_scalesr anglesaxesrNr jrG gridlabellinesrltextr)arrr__init__sp                zComplexRadar.__init__c Cst|dd|ddD]\}\}}||kr|ks,n||kr)|ks,JJq |d\}}|d}|g}t|dd|ddD]\}\}}||||||||qG|S)zScales data[1:] to ranges[0]rNr)rWappend) rdatarMdy1y2x1x2sdatarrr _scale_datats&2 &$zComplexRadar._scale_datacOsJ|||j}|jj|jtj||dfg|Ri||jd|_dS)z Plots a linerrN)rnrMrOplotr r:rLrP)rrgargskwargsrmrrrros,zComplexRadar.plotcOs|jj|i|dS)zShows a legendN)rOlegend)rrprqrrr use_legendszComplexRadar.use_legend)rTN)__name__ __module__ __qualname____doc__rernrorsrrrrrs  \ rc svt|}||_|tfddDdurtd|jddg}|dd|d|d|d<|dd|d|d|d<t|j ddd }fd d t |D}id d id d id d iddiddidd ddddddd}|dur| ||durt }t||dd |d}t |jD]"} |j|j| j| |d|dd|jdi|d|ddq|S)aCreate a complex radar chart with different scales for each variable Source: https://towardsdatascience.com/how-to-create-and-visualize-complex-radar-charts-f7764d0f3652 Args: data (`List[dict]`): the results (list of metric + value pairs). E.g. data = [{"accuracy": 0.9, "precision":0.8},{"accuracy": 0.7, "precision":0.6}] names (`List[dict]`): model names. E.g. names = ["model1", "model 2", ...] invert_range (`List[dict]`, optional): the metrics to invert (in cases when smaller is better, e.g. speed) E.g. invert_range=["latency_in_seconds"] config (`dict`, optional) : a specification of the formatting configurations, namely: - rad_ln_args (`dict`, default `{"visible": True}`): The visibility of the radial (circle) lines. - outer_ring (`dict`, default `{"visible": True}`): The visibility of the outer ring. - angle_ln_args (`dict`, default `{"visible": True}`): The visibility of the angle lines. - rgrid_tick_lbls_args (`dict`, default `{"fontsize": 12}`): The font size of the tick labels on the scales. - theta_tick_lbls (`dict`, default `{"fontsize": 12}`): The font size of the variable labels on the plot. - theta_tick_lbls_pad (`int`, default `3`): The padding of the variable labels on the plot. - theta_tick_lbls_brk_lng_wrds (`bool`, default `True` ): Whether long words in the label are broken up or not. - theta_tick_lbls_txt_wrap (`int`, default `15`): Text wrap for tick labels - incl_endpoint (`bool`, default `False`): Include value endpoints on calse - marker (`str`, default `"o"`): the shape of the marker used in the radar plot. - markersize (`int`, default `3`): the shape of the marker used in the radar plot. - legend_loc (`str`, default `"upper right"`): the location of the legend in the radar plot. Must be one of: 'upper left', 'upper right', 'lower left', 'lower right'. - bbox_to_anchor (`tuple`, default `(2, 1)`: anchor for the legend. fig (`matplotlib.figure.Figure`, optional): figure used to plot the radar plot. Returns: `matplotlib.figure.Figure` c3s|]}|vVqdS)Nrr)rZrr szradar_plot..FzDAll of the metrics in `invert_range` should be in the data provided.minmaxrN)indexnamecs,g|]\}\}}|vr||fn||fqSrr)r var min_value max_value) invert_rangerrrs zradar_plot..r"Tfontsize oz upper right)rr)rr#r$r%r!r1r7r,r+rmarker markersize legend_locbbox_to_anchor)r[r\r rr)rrrrr)locrr)pd DataFramer{keysall ValueErrordescribeTlist itertuplesrWupdatepltfigurerrorvaluesrs) rg model_namesrconfigrmin_max_per_variablerMr radargr)rrZr radar_plots^ +    " r) r/matplotlib.pyplotpyplotrnumpyr:pandasrrrrrrrs