`j#kdZddlmZddlZddlZddlZddlmZddlm Z m Z ddl Z e jdddl mZddl mZddlZddlZddlmZmZd d lmZmZmZmZmZmZmZmZmZm Z d d l!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+d d l,m-Z-e"d z Z.dZ/dZ0dZ1edZ2e2jgr?ejhjke6e2dejnd<dejnd<ejpfd0dZ9d1dZ:d2dZ;d3dZd5d7dZ?d8dZ@d9dZAd:dZBdddd d! d;d"ZCdd%ZFd?d&ZGd@d'ZHd?d(ZId?d)ZJd?d*ZKd?d+ZLdAd,ZMdBd-ZNdCd.ZOePd/k(reOyy)DaGenerate local, sample-level input-response diagnostics for prepared models. The artifacts in this module are deliberately *not* feature importance claims. They are absolute input-times-gradient responses of an unvalidated model's mean raw output at one prepared sample (run ``18h01`` where a run is applicable). They answer which inputs the executed network is locally sensitive to, not what caused a forming defect or what is predictive in a new production run. ) annotationsN)Path)AnyCallableAgg) font_manager)Tensornn) CNN2DUNetDGCNNSegFNO2dMeshGraphProcessorPhysicsConstrainedRiskGNN PointNetSegSparseUNetOptionalTabularSurrogateUNet3DUVNetFaceModel) ROOT brep_input graph_inputs image_input point_inputsprocess_conditionssparse_batch_collatesparse_voxel_input tabular_input voxel_input)load_training_datazresults/models/attributions18h01z6absolute input-times-gradient of mean raw model outputzLocal response diagnostic only. production_valid=false; no causal, validated-prediction, or cross-sample feature-importance claim.z2/System/Library/Fonts/Supplemental/AppleGothic.ttf AppleGothicz font.familyFzaxes.unicode_minusc0tj||S)Ndtype)torch as_tensor)valuer%s models/attributions.py_tensorr*0s ??5 ..c|yt|z }|jsd|Stj|dd}|j |d|S)zFLoad only an existing diagnostic checkpoint, retaining its provenance.random_initializationzmissing_checkpoint:cpuF) map_location weights_only state_dict)rexistsr&loadload_state_dict)model checkpointpathpayloads r)_loadr94sV& * D ;;=$ZL11jjEFG ',/0 r+cDd|d}t|z jr|S|S)zKUse the exact state behind the displayed 18h01 CAD diagnostic when present.z1results/models/cad_predictions/checkpoints/18h01-z.pt)rr2)slugfallback candidates r)_preferred_checkpointr>@s+CD6MI )1139AAr+c|j|Dcgc]/}|jjjd1}}|j d||}|j dk(s#t j|js td|jjg}|D]}|j-t j|jjs td|j||jzjjjj!j#t$j&|Scc}w)NT) set_to_nonerz)attribution output is empty or non-finitez-attribution gradient is missing or non-finite)evaldetachclonerequires_grad_ zero_gradnumelr&isfiniteall RuntimeErrormeanbackwardgradappendabsr.numpyastypenpfloat32)r5forwardinputsr(leavesresultleafs r) _grad_x_inputrXFs# JJLGM Nveelln""$33D9vF N OOO% V E {{}!6!:!:!<FGG JJL F 99 ENN499$=$A$A$CNO O tdii',,.557;;=CCELLRZZXY MOs4Fcr|Dcgc] }t|}}||Dcgc]}d|z c}zScc}wcc}w)Nz valid_mask: )str)namesnames r)_names_with_masksr]VsA#( )54SY5E ) eESX YSXac!f 5A3H ISX YY Ys-Bc zt|||}|Dcgc]}|d c}ddd}|Dcgc]}|d c}ddd}tjdtddt |zdzfd \} } | j t jt ||d | jt jt ||d | jd| j|| jdd| j|dtj| ycc}wcc}w)Nr`r_ra g @g(\?g333333?Tfigsizeconstrained_layoutz#276fbf)color)fontsizeuabsolute input × gradientxg?)axisalphadpi)rmpltsubplotsmaxrgbarhrQarange set_yticks set_xlabel set_titlegridsavefigclose) r7rir[titlerjpicksrowlabelsscorefigaxs r) _plot_barras   &E(- .c)n .tt 4F(- .S_ .tt 4EllAs3c&k0AC0G'H#I^bcGCGGBIIc&k "EG;MM"))CK(&1M=MM./LLGGCG KK#KIIcN/ .s D3 D8ctjdd\}}|j|d}|j||j |j ||d|j |d tj|y) N)g@Trpmagma)cmapmean absolute input × gradient)rlabelryrz)r|r}imshowr set_axis_offcolorbarrr)r7spatialrrrimages r) _plot_maproskll8EGC IIgGI ,ELL*LL2%FLGKK#KIIcNr+ctdDcgc]}|j|}}tjdddd\}}t ||dD]O\}}} |j |dd } |j | |j|j| |d Q|j||j|d tj|ycc}w)Nrwr ) g @Trp)z max over Xz max over Yz max over Zrauto)raspectg?)rfractionryrz) ranger~r|r}ziprrrrsuptitlerr) r7rirrl projectionsraxesrwr(rrs r) _plot_volumerxs/4Qx8x!6::1:%xK8 Q9NIC!$ 5_`eU E ? ut002CLL4Z]L4^aLLKK#KIIcN9sCc tdt|dz}tjdd}|j d}|j |dd|df|dd|df|dd|d f|dd|d d }|j ||j|j||d d|j|dtj|y)Nr i.)rTrp3d) projectionrg?r)csrg?r)rshrinkrryrz) r~rgr|figure add_subplotscatterrrrrr)r7posr(rsteprrpointss r) _plot_surfacers q#c(e# $D **V =C D )B ZZFdFAIFdFAIFdFAI%PVRVPV-G-FLL*CLLBs@a-9-cKK#KIIcNr+ spatial_npzmap_fileprocessnotecd|v} id|dddtd|d| rdnd |vrd nd d | d |dtd|dddddtdt||d|d|d|d| S)Nz"cad_predictions/checkpoints/18h01-r5status completedmethodr6checkpoint_scope18h01 weak-label diagnostic fitweak_gnnzweak-label diagnostic6two-step synthetic zero-output learnability diagnostic"matches_displayed_18h01_prediction input_sourcesampleoutput_selectorproduction_validFgeneralization_measuredwarningfeature_response_toprrprocess_responser)METHODRUNWARNINGrm) r\r6rr feature_namesfeature_valuesrrrr displayeds r)_recordrs5 BI   .6 j "4\e7X,6*,D(I  -i    '/  6G  E $=e FOPW  ^] C  { %/ ;Mg    r+cNtt||tt|dS)N)rr feature_count)rrmrhrg)rar[s r)_process_summaryrs$d8U6K X/ 11r+ctt}t\}}}}|jt}t j |d|dj tjgd}t j ||j tjgd||dz}t|}t|jt|d} tt jt|dz dd } t jt|z d} tt| d j tt| d j tz} g} i}d t d dfdt"ddffD]\}}}}||j$d| j$dt't)||}t+fd|| \}}|j-d}|j-d}|j/j1ddj1dd}t3t4|dz || |dt7t4|dz ||dt3t4|dz |j-d| |d|||<t j8t4|dz |||j-dt j:| t j:|  | j=t?|||dd!| ||d|dtA|j-d| " | |fS)#Nrvmaskr rrsourceF allow_picklerprocess_feature_namesgeometry_feature_namesz 2D CNN/U-Netz 2d-cnn-u-netz3results/models/checkpoints/synthetic_2d_cnn_unet.ptrfno2dz-results/models/checkpoints/synthetic_fno2d.pt global_dimc<tj||SNr&sigmoidabr5s r)zimage_models..su}}UZ[\^_U`Gar+rrrrr /- z -channels.pngz: CAD image channel response-map.pngz: six-view mean responsez -process.pngz: XLS/CAD-summary response.npzchannel_responsespatial_responserrrz8mean weak-fit predicted probability across six CAD viewsrrr)!rrrkrrQ concatenaterPrRr*repeatrgr]r3rlistrZr rshaper9r>rXrJlowerreplacerOUTrsavez_compressedrbrMrr)data process_x process_maskrun_idsprocess_sourcerunx_npp_nprvprtabular process_namesoutarraysrclsr;ckptloadedsaliencyprocess_saliencychannelrstemr5s @r) image_modelsrs/ =DM_MaJ)\7N -- C >>49d6l&9&9"**&EF JD >>9l&9&9"**&EF J3sUVw WD A74;;s4yq;#ABq%bggdT(^.CRW&XYh&ijMggd^+%@G%d73J+K+R+RSV+W&X[_`giAaBaIaIJMaN\O'OPM "CBF N4ij %"QR#sD$AGGAJ1771:6uNcdhjnNoAp%25:acdfg%h""--Y-/8==f=;U{{}$$S#.66sC@#4& ..5'QmHno#4&))7ug=U4VW#4& --/?/D/DQ/G[`Zaa{X|}t  CTF$-/'\c-=-B-B1-EUWU_U_`mUn24**]2K M 75&$x.:t('$t}Z^Y__gWh#34D4I4I!4Lm#\^ _#$ ;r+cDt}t\}}}}|jt}t j |d|dj tjgdd}t j ||j tjgd||dz}t|t|} }t|jd| jdttdd} tfd|| \} } | jd } | jd }tt j t"|d z d d}t j t"|z d }tt%|dj t&t%|dj t&z}t)t*dz | |dt-t*dz |dt)t*dz | jd|dt j.t*dz | || jdt j0|t j0|t3d| |d d|| ddt5| jd|d |fS)Nrvrrr rzdense-3d-u-netz/results/models/checkpoints/synthetic_3d_unet.ptc<tj||Srrrs r)rzvoxel_model..s5==QVWXZ[Q\C]r+)rrrrrrFrrrrzdense-3d-u-net-channels.pngz&Dense 3D U-Net: voxel-channel responsezdense-3d-u-net-map.pngz&Dense 3D U-Net: surface-voxel responsezdense-3d-u-net-process.pngz(Dense 3D U-Net: XLS/CAD-summary responsezdense-3d-u-net.npzrzDense 3D U-Netz)mean weak-fit predicted voxel probabilityzPSurface-only voxel representation; the map is not a solid material-volume field.r)rrrkrrQrrPrRr*rrr9r>rXrJr]r3rrrZrrrrrbrr)rrrrrrrrrvrr r rrrr[rawrr5s @r) voxel_modelrs =DM_MaJ)\7Nipivivwzi{cf >>49d6l&9&9"**&EF J4 PD >>9l&9&9"**&EF J3sUVw WD 4='$-qA 1771:!''!* 5EeMbcsvgNhAiv!.u6]_`bc!dHmmm.G(--V-:T bggdT(^&;%PQ`a bE ''$'e )>q)AQSQ[Q[\aQbz|{E{EFS{TU #VT(^=hjoqx3>V+,<,A,A!,DmTj lnu uur+c  ddlt}t |d |d \}}t |t jt |}t|jd td d }t|z jrt|}d}n1t jdt|jd d}d}tfd|d}|j!d|j!d } } t#j$t|dz d} t'| d} t#j(| dt"j*t-| d}} | t#j(|d t"j*dz|zz}t/t0dz | | dt3t0dz || dt#j4t0dz | | |t#j(|d t"jt#j(| t7d||dd| | ddd ! }d"|vrd#nd$|d%<d"|vrd&n||d'<d"|vrd(|d)<|S|dk7rd*|d)<|S#t$r}ddddtd|dcYd}~Sd}~wwxYw)+zGRun a true sparse attribution only when the optional backend is usable.rNz Sparse U-Netunavailable_optional_backendMinkowskiEngineFzbackend import unavailable: )r5rbackendrrreasoncoordsfeaturesr zsparse-3d-u-netz3results/models/checkpoints/synthetic_sparse_unet.ptr*random_initialization_seed17_no_checkpointz9fixed-seed random initialization; no checkpoint availablec~tjj|jdddfS)N)r coordinatesr)r&r SparseTensorF)rMErr5s r)rzsparse_model..s4emmE"//[\jp/Bqrr2r9 manual_seedrXrJrQr3r]rbrRrcrrrrr)excr coords_np features_nprr6r scoper r`rrr[originpitchrrr$rr5s @@@r) sparse_modelr6so$  D1$x.$zBRSI{ Y ,F9Mh x~~a0 1E&'8:opJ z!!#uj)H ""8>>!#45=KU${~FGHIJH}}Q'q)9WG ''$h'e $I93I>9I>ctd}tj|d|djtjgdd}t |}tj t|dz d}t|d}tj|d tj}g}d td d t|fd tdddffD]9\}}} } } |ddd| f} || jdfi|turddinitt| | } t!fd| d}|j#d}|j#d}|j%}t't(|dz |||dt+t(|dz |d| ||dtj,t(|dz |||d| tj||j/t1|| |dd|||d|d| dk(rdnd ! <|S)"Nrrvrr rFrrpos_mmPointNetpointnetzCresults/models/checkpoints/synthetic_pointnet_cad_surface_points.ptDGCNNdgcnnz@results/models/checkpoints/synthetic_dgcnn_cad_surface_points.ptir_k c:tj|Srrrr5s r)rzpoint_models..s%--a2Ir+rr)rrz -features.pngz: point-feature responserz: CAD point responser)r'point_response point_pos_mmrz)mean weak-fit predicted point probabilityz7DGCNN uses the executed deterministic 512-point prefix.rr+)rrQrrPrRr*r3rr]rbrrgr rr9r>rXrJrrrrrrMr)itemrrvrr[rrowsrr r;r capzr r r`rrr5s @r) point_modelsrG s: >! D >>49d6l&9&9"**&EF J4 PD A ''$h'e ;fhmov+/& 4&@Q_bfi_i"[oqt u( Kr+cVttd}t\}}}}|jt}t j |d|djt jgd}t j|dt jj}|dddfxxdzcc<t j ||jt jgd||dz}t|t|t|} } } t|d tjtjt| tj t!| j"d| j"d| j"dt%t'd d } t)fd | | | \} }}| j+d}| j+d}t-t j.t0|dz dd}t j.t0|z d}t-t3|djt4t3|djt4z}t7t8dz ||dt;t8dz t j.t0|dz dd|dt7t8dz |j+d|dt j<t8dz |||j+d|j+dt j.t0|dz ddt j|t j|t?d| |dd||ddtA|j+d| S) Nc |dSNnodesrs r)rzmesh_model..%G r+keyrvrr edge_attrrg@@ edge_indexr$zmesh-gnnz4results/models/checkpoints/synthetic_meshgraphnet.ptc Btj|||Srr)rrrbatchrRr5s r)rzmesh_model..-s0UZUbUbchijlvxy{|DdEVFr+rFrrrrzmesh-gnn-features.pngzMesh GNN: node-feature responsezmesh-gnn-map.pngr8zMesh GNN: CAD-node responsezmesh-gnn-process.pngz"Mesh GNN: XLS/CAD-summary responsez mesh-gnn.npznode_feature_response node_responseedge_feature_responserr8rrzMesh GNNz,mean weak-fit predicted CAD-node probabilityr)!r~rrrkrrQrrPrRrbcopyr*r&longzerosrgrrr9r>rXrJr]r3rrrZrrrrrr)graphrrrrrredge_np process_nprvedgerr r  edge_saliencyrr`rr[rrrTrRr5s @@@r) mesh_modelra$sr  $< =E7I7K4I|WnSZS`S`adSeS >>5:uV}';';BJJ'GH! LDjj{+RZZ8==?GARXAXL,?,? ,K LaPQTUX[\U\]J 0'*2EQtA|,ejj9J5;;sSTv]b]g]gCh5 qwwqz4::a=!''!* EEPUV[]rs}@v^wQxv0=eFFHIKOQR1S-Hm-mmAG(--*:DUVXV]V]^bejkset^tDIWJKZW[E\E ''$'e .Ds6UZUbUbchijlprsuvx}c~@LdMVNr+rc |dSrJrLrMs r)rzrisk_gnn..FrNr+rOrFrrz/data_processed/paired/18h01_cad_ply_targets.npzrzobserved_mask: zimputed_mask: zrisk-gnn-features.pngz'Constrained risk GNN: CAD-node responsezrisk-gnn-map.png cad_pos_mmz)Constrained risk GNN: risk-logit responsezrisk-gnn-process.pngz"Constrained risk GNN: XLS responsez risk-gnn.npzrUzConstrained risk GNNzrXrJrr]rQr3rrZrrrrrbrr)rrnslrvrQrr r r`rr`rr\r[ raw_namesr\rrTr_r5s @@@r)risk_gnnro=s  DtF|'9'9#'>DDY`efilmfmpsvwpw{|o|`}[]V R !A'$v,qBm4Itu c**,<,A,A!,DmUyzn,G[b.;.@.@.CVfVkVklmVn#L1EARjljtjtvCkDE )63dQSXZa-8J+,<,A,A!,DmTr  ss68mqes0L)L.%L3c  t}t|dt|dtjc}t |j d t tdd}t  fd|d}|jd |jd }}gd }tjt|d z d djd }ttdz ||dttdz ||dtj tdz |||tj"|t%d||d d||ddS)Ngrids adjacencyr zuv-netz-results/models/checkpoints/synthetic_uvnet.ptc<tj|Srr)rrrr5s r)rzuv_model..YsemmE!YrXrJrQr3rrrrrrbr) rrvr r r`rr[rrrr5s @@r)uv_modelrVs^ [HY[`[e[e@fI 1771: &Eu>ST\_N?O2PU$PRSTUVWH}})}4hmmm6SWG rE ''$h'e .jsemmE!H.Er+rztabular-mlp-features.pngzTabular MLP: run-level responseztabular-mlp.npz)r'rz Tabular MLPz(weak-fit predicted run-level probabilityr)rrkrr*rQrrPrRr3rr]rrZrrr9r>rXrJrrrrbr)rrrvrr[r r r5s @r) tabular_modelrdsg ?D$v,"4"4S"9CS 4<+>+>rzz+JKQOPSTWZ[T[\]A ''$h'e #?#F#Fs#KLtTWXpTqTxTxy|T}O~~ E QWWQZ (E5@UVcfaAb4c&U$EqI!LQQRSTH c..%Abc//(Z\ZdZdejZkl =&$x.:dfkmu0;U WWr+cg}|D] }djd|jdgddD}|jdxsi}djd|jdgddD}|jdr5dtj|dd tj|d d nd}|j d tj|d d tj|dd|dtj|dd|d|xsdd #t dz j dtjtdtjtddj|dy)Nrc3bK|]'}dtj|dd|ddd)ywz
  • r`z: raz.3gz
    • Nhtmlescape.0rCs r) z_page..tsENfM]aT[[i9:"T*=Mc.vsFZnYeiT[[i%A$B"T*EUVYDZZ_`nYrrz r5���z local response mapz

    z

    rz

    z

    machine-readable NPZ

    Geometry/topology channels

      z$

    XLS/process branch

      z"
    1. Not wired for this model.
      2. z
    z index.htmluLocal model response diagnostics

    Model dashboard · machine-readable report

    Local input-response diagnostics

    z Values use z for one prepared 18h01 sample and the named checkpoint. Where available, each checkpoint is the exact weak-label diagnostic state used for the displayed CAD prediction; a synthetic fallback is named explicitly in the JSON.

    utf-8encoding) joingetrrrMr write_textrr)rDcardsrtopr process_topmap_tags r)_pagerqs^ EggNehelelnDFHfIJLKLfMNN'',-3ggZmtmxmxzPRTnUVXWXnYZZ x{xxAKyLJt{{3z?;$?~dkkZ]^oZpNqMrr}F~GGTUYU`U`aderasUtTuu{|{@@depeXtXdYYhi j   <##)kk'*+< F8K7LMpqsqxqxy~qp@%CNU$Vr+c Ntjddtjt dt dtj g}t\}}|j|t\}}|j||jt|jt|jt|jt|jt|jt!t"t$dt&|d}tdz j)t+j,|dd d t/|Dcgc]}|d j1d s|c}tdz Scc}w)NT)parentsexist_okr rF)rrrrmodelsz report.jsonr) ensure_asciiindentrrrr)rmkdirr&set_num_threadsr~rfget_num_threadsrextendrrMr6rGrarorrrrrrjsondumpsr startswith)rD image_rows_ voxel_rowreportrs r)buildrs\IIdTI* #aQ(=(=(?!@AB!#D NMJDKK $;=LIq$++i"8KK \^!WY]YdYdemeoYprvr}r}GIsJLPLWLWXeXgLh#5U\hl mF=$$TZZUST%U`g$h $ H$3#h-":":;"G3$ HI   Is 4F"F"ctjt}|jddd|j }|j s|j dtt y)N) descriptionz--build store_truez1generate prepared-sample local response artifacts)actionhelpzchoose --build)argparseArgumentParser__doc__ add_argument parse_argsrerrorprint)parserargss r)mainrsS  $ $ 9F  ,=pq    D :: %& %'Nr+__main__)r( np.ndarrayr%z torch.dtypereturnr )r5 nn.Moduler6 str | NonerrZ)r;rZr<rZrrZ)r5rrSzCallable[..., Tensor]rTr rzlist[np.ndarray])r[zlist[str] | np.ndarrayr list[str]))rirr[rrjrhrlist[dict[str, Any]]) r7rrirr[rrrZrjrhrNone)r7rrrrrZrr)r7rrirrrZrr) r7rrrr(rrrZrr)r\rZr6rZrrZrrZrrrrrrrrrzdict[str, Any] | NonerrZrdict[str, Any])rarr[rrr)rz2tuple[list[dict[str, Any]], dict[str, np.ndarray]])rz!tuple[dict[str, Any], np.ndarray])rr)rr)rDrrr)rr)rr)Qr __future__rrrrpathlibrtypingrr matplotlibusematplotlib.pyplotpyplotr|rrOrQr&r r architecturesr r rrrrrrrrrrrrrrrrrrrtrain_weak_gnnr rrrr _KOREAN_FONTr2 fontManageraddfontrZrcParamsrRr*r9r>rXr]rmrrrrrrrrr6rGrarorrrrr__name__rLr+r)rs#   u# ZZZ//// **  A M HI $$S%67"/CLL).CLL%&5:MM/ B  = Z   `d#'$$6@R\ 2G .&1 >u,)X4V2s2 H W V  zFr+