ȗi"dZddlmZddlZddlZddlmZmZm Z ddl Z e je Z d1deedeedeeded ed ed eeed efd Z d2de eeefde eeefdeedee eeefd ef dZ d3dededeededeed ef dZe dk(r2ddlZej,dZej1deddej1dedd ej1d!edd"ej1d#ed$d%ej1d&edd'ej1d(d)d*+ej1d,d)d-+ej3Ze j6e j8d./ej:xr ej< Zeej@ejBejDeejFejH0yy)4z*Plotting utilities for experiment results.N)ListOptionalDictXbaselineours output_pathtitlexlabelylabelpreference_weighteddpicttdt|dz}tjdtj ||ddddtj ||d d dd|2t|t|k(rtj ||d d ddtj |dtj|dtj|ddtjdtjddtjddgtjdt|gtj|rFtdt|} tj || dt"j%d|ntj&tj(y )aPlot learning curve comparing baseline, contrastive, and optionally preference-weighted methods. Args: baseline: List of accuracy scores for baseline method ours: List of accuracy scores for contrastive method output_path: Path to save plot (optional) title: Plot title xlabel: X-axis label ylabel: Y-axis label preference_weighted: Optional list of accuracy scores for preference-weighted method (2025-style) ) figsizeb-oBaselinerlabel linewidth markersizer-s ContrastiveNg-^Preference-Weighted (2025) fontsizeboldr" fontweight T333333?alphar-tightr bbox_incheszPlot saved to )listrangelenpltfigureplotr r r legendgridylimxlim tight_layoutmaxintsavefigloggerinfoshowclose) rrr r r r r rroundssafe_dpis M/home/ubuntu/codebase/yexijia/保研/colocation_mvp/src/experiment/plotter.pyplot_learning_curverD sI*%3x=1,- .FJJwHHVXuJ!PQRHHVT5 qQ&3/B+Cs8}+T ,e;WcdqrsJJv#JJv#IIebV4JJHHTHHaVHHaX  sCH% KX7C n[M23  IIKbaseline_metricscontrastive_metricspreference_weighted_metricsc ttdt|ddz}tjddd\}}|j ddd dd |d fd d |dfdd|dfdd|dfg}|D]j\} } } || } || } | j || dddd| j || dddd|7| |vr3|| }t|t| k(r| j ||dddd| jdd | j| d | j| d!d | jd" | jd#d$%| jd&dg| jdt| gt| d&kDs| j| d'd(t| | d'fd)d*d+d,-| j| d'd(t| | d'fd)d.d+d,-mtj|rFt!d/t#|}tj$||d01t&j)d2|ntj*tj,y)3aPlot all metrics (accuracy, precision, recall, F1) in subplots. Args: baseline_metrics: Dictionary with 'accuracy', 'precision', 'recall', 'f1' lists contrastive_metrics: Dictionary with 'accuracy', 'precision', 'recall', 'f1' lists output_path: Path to save plot (optional) preference_weighted_metrics: Optional third method metrics (2025-style) raccuracyr)r#rrzEvaluation Metrics Comparisonr$r%Accuracy)rr precision Precision)rrrecallRecall)rrf1zF1 Score)rrrrrrrNrrInteraction Roundrr!r Tr(r)rz.3fz offset points)rRrleft) textcoordsxytexthar")rRir+r,r-zAll metrics plot saved to )r/r0r1r2subplotssuptitler4 set_xlabel set_ylabel set_titler5r6set_ylimset_xlimannotater9r:r;r<r=r>r?r@)rFrGr rHrrAfigaxesmetrics metric_key metric_nameaxr contrastivepwrBs rCplot_all_metrics_funcrkEst%3/ ;?  IIKrE baseline_pathcontrastive_pathplot_all_metricspreference_weighted_pathcft|d5}tj|}dddt|d5}tj|}dddttr d|i} di} n|} } d} |r_t j j|r@t|d5}tj|} dddt tr| nd| i} |rd| vrddl m } | | | || |y| r| dnd}t| d| d|||y#1swYxYw#1swYxYw#1swYqxYw) aLoad results from JSON files and plot. Args: baseline_path: Path to baseline metrics JSON file contrastive_path: Path to contrastive metrics JSON file output_path: Path to save plot (optional) plot_all_metrics: If True, plot all metrics in subplots; if False, plot only accuracy preference_weighted_path: Optional path to preference-weighted metrics JSON (2025-style third method) rNrJrMr)rk)rHr)r r r) openjsonload isinstancer/ospathisfiledictexperiment.plotterrkrD)rlrmr rnrorf baseline_datacontrastive_datarFrG pw_metricspw_datarkpw_accs rCplot_from_jsonrs0" mS !Q !  "  $99Q< %-&& 6)+;<(.JBGGNN3K$L *C 0AiilG1 *7D 9W G?T K+;;<   (2  ,6J'4 Z (  +# &  ? " ! % $1 0s"DD!D'DD$'D0__main__zPlot learning curves) descriptionz --baselinezresults/metrics_baseline.jsonzPath to baseline metrics JSON)typedefaulthelpz --contrastivez results/metrics_contrastive.jsonz Path to contrastive metrics JSONz--preference-weightedz@Path to preference-weighted metrics JSON (optional third method)z--outputzresults/learning_curve.tiffzOutput path for plotz--dpiz5Output resolution in DPI (must be >300; default: 600)z--accuracy-only store_truezHOnly plot accuracy (single figure); default when not using --all-metrics)actionrz --all-metricszBPlot all metrics (accuracy, precision, recall, F1) in 2x2 subplotsz4%(asctime)s - %(name)s - %(levelname)s - %(message)s)levelformat)rnror)NzLearning Curve ComparisonrSrLNr)NNr)NFNr)%__doc__matplotlib.pyplotpyplotr2rsrvtypingrrrlogging getLogger__name__r=floatstrr;rDrkboolrargparseArgumentParserparser add_argument parse_argsargs basicConfigINFO all_metrics accuracy_onlyplot_allrrioutputr rrErCrs0 ''   8 $ "&,%1565k6 u+6#6  6  6  6"$u+.6 6x"&DH >3U +,>c4;./>#>"*$sDK/?*@!A > >H"&".2 6 6 6 #6  6 'sm 6 6 r z $X $ $1G HF  38W;= c;]>@ /c4^`  6S24 c3SU ),fh  `b    DGllE :(:(:$:H t''!!%!9!9 HH 9rE