EjBHRdZddlmZddlZddlZddlZddlmZddlm Z ddl Z e jdddl m ZddlZddlZddlmZddlmcmZdd lmZdd lmZmZmZmZdd lmZed z Z d Z!ejDfddZ#ddZ$ddZ%ddZ&ddZ'ddZ(ddZ)e*dk(re)yy)a*Diagnostic weak-label GNN training for CAD risk and deformation fields. This command deliberately requires an opt-in flag. The current two runs have unvalidated registration, point-dilated defect areas, and one defect class. The result is a pipeline/training diagnostic, never a production model. ) annotationsN)Path)AnyAgg)Tensor)PhysicsConstrainedRiskGNN)ROOTfixed_physical_normalize graph_inputssplit_embedded_units)seed_everythingzresults/models/weak_gnng>@c0tj||S)N)dtype)torch as_tensor)valuers models/train_weak_gnn.py_tensorrs ??5 ..ctj|dtjj}tj|dt}tj|dj t }t|\}}tjt|t}|j}d|d|dfd|d|d ffD]M\}}} td d D]8} || d } | |vs|j| } | | d z s)|| d z || <d || <:O||z} tj| |dj tj}t|dtj||d}tj||j tj|j tjgS)zFUse interpolated heater values while retaining observed/imputed flags. process_xprocess_feature_validprocess_feature_namesu 상부온도 heater_top_cheater_top_imputedu 하부온도heater_bottom_cheater_bottom_imputedr02dTrN)npasarrayfloat32copyboolastypestrr zeroslentolistrangeindexwherer concatenate)datarawobservednames base_namesunitsimputedprefixfilled filled_maskzonekeyr,usablevalues normalizeds r_process_vectorr>#s **T+& 3 8 8 :Czz$67>H JJt34 5 < >:xrzz'BGNNSUS]S]D^_ ``rc j ttd}ttdz j d}|s t dg}|D]}t j|d5}|jDcic]}||| }}ddddjd |d k7rt |jd |j||ft j|d |d jt jgd}t j|Dcgc]}|c}d } |d } t jtt!|D cgc] } |d| | zzc} d} t j|Dcgc]}|d c}d jt j} | ddd fxxdzcc<t j"t j$t!|| }t j&|Dcgc]\}}t)|c}}}t j|Dcgc]\}}|ddddfc}}jt j}t j|Dcgc]\}}|ddddfc}}jt j}t j|Dcgc] \}}|d c}}jt j}t j|Dcgc] \}}|d c}}jt j}g}t+||Dcgc]!\}}t-|dj/#c}}D]\\}}}t j0|dt j}t j0|dt j}|j|t3t!|t3|t4k\j7t3t j8|t4k\j7t;|jt;t j8|jdt j<|t4 t4t4z }t j<|d t4t4z }|Dcgc]!\}}t-|dj/#}}}tt?|}t j0|Dcgc]}|jA|c}t jB}id| d| d| d|d|d|d|d |d!|d"|d#|d$|d%|Dcgc]!\}}t-|dj/#c}}d|d ddd&| d'|Scc}w#1swYexYwcc}wcc} wcc}wcc}}wcc}}wcc}}wcc}}wcc}}wcc}}wcc}}wcc}wcc}}w)(Nc |dS)Nnodes)items rz$load_training_data..=sgr)r:zdata_processed/pairedz*_cad_ply_targets.npzzIAt least one paired target file is required for the diagnostic experimentF) allow_pickle cad_pos_mmrrAz0: target nodes do not match the finest CAD graphxmaskr)axis edge_index edge_attrg@@weak_defect_probabilityweak_defect_maskcad_to_scan_normal_residual_mmcad_to_scan_distance_mmrun_id)runrAdistance_nodes_at_or_above_capsigned_nodes_outside_capraw_max_distance_mmraw_max_abs_signed_mm defect_classnodeedgebatchprocessrisk risk_binarysigned magnitude class_labels class_namespathsruns nodes_per_rundeformation_clip_by_run)"maxr sortedr glob RuntimeErrorr!loadfilesshapenameappendr.r&r#r+r)repeatarangestackr>zipr'rCr"intDEFORMATION_SCALE_MMsumabsfloatclipsetr,int64)graphrarbpathr0r:rCnode_one_rWcountr,rXrKrYrZr[r\r]r^rdrQdistancenormalclassesr`rlabelss rload_training_datar<s  $> ?E D22889PQ RE fgg D WWT .#-0YY7YcCSMYD7/   # #A &%. 8$)),\]^ ^ T4L! ~~uSz5=+?+? +KLSTUH >>T2T8T2 ;D 'NE >>ERUVZR[L\]L\55.>L\]de fDDADqk 2DAJQQRTR\R\]I adOvO IIbiiD *E 2EhhTBT'!T-TBCG >>PTUPTWQ4 9:1a4@PTU V ] ]^`^h^h iD..PT!UPTWQ$'9":1a4"@PT!UV]]^`^h^hiK ^^SWXSW4T"BCSWX Y ` `acakak lFtTtGAt%> ?tTU\\]_]g]ghI dt$TtGAtSh)<)<)>%?t$TU D3::d#<=rzzJD!ABBJJO&&S]!3.18?S3S2X2X2Z.[(+RVVF^?S-S,X,X,Z([#(#8%*266&>+=+=+?%@ (  VWWV224H IL` `F 1&:;>RRI?CDtGAts4',,./tGDW&K ZZwGwe**51wG RF  d $/ d?|d@|dAd>dB|dC*),tbdD|dEdFdG+dHdIgdJdKgte|5jgthdLdMg|dN }6tZdOz jktmjn|6d*d$PdQRdSjqdT|,D}7tZdUz jkdVt|d&dW|6d?dXdY|*dXdZ|)dXd[tsjt|d\|7d] dQR|6Scc}%}$w#1swYxYw)^NrWrXrKrYrZr[r\r]r^r_rr`gh㈵>)lr weight_decay)reinfrtrained_multiclass!unavailable_single_observed_classT) set_to_none risk_logit) pos_weightsigned_normal_deformationg333333?)deltadeformation_magnitudedefect_type_logitsg?g{Gz?g{Gzt?)epochtotalbce dice_loss deformationrisk_graph_smoothnessdeformation_graph_smoothness type_losszNo training epoch completedg?rrcrb)rQtarget_positive_nodespredicted_positive_nodes training_iou training_dicemax_fitted_risk_probability)parentsexist_okzdiagnostic_weak_gnn.ptF)r[signed_deformationr^zK unvalidated NG run(s); weak point-dilated labels; training diagnostic only) state_dictproduction_validr`output_channelswarningstatus(diagnostic_weak_label_training_completedrgeneralization_measuredinputz`STEP-derived CAD graph + XLS process values/masks; interpolated heaters include imputation flags target_sourcezNregistered fused all_merge_pointcloud_full PLY + defect CSV weak geodesic areaindependent_samplesdefect_classesdefect_type_trainingdefect_type_reasonzHThe supplied run-level target has only one class and no reviewed OK run.epochs learning_rate loss_initialr loss_finalloss_last_epochselected_epochtraining_threshold_iouzFbound the influence of unvalidated nearest-scan registration residualsrd)diagnostic_clip_mmreason counts_by_runz?predicted magnitude >= abs(predicted signed normal deformation)zrisk graph smoothnessz#signed-deformation graph smoothnesszmheat-equation residual: missing time histories, material conductivity/capacity and physical heater footprintszrthermo-mechanical residual: missing temperature-dependent material law, thickness, contact and boundary conditionsz$results/models/weak_gnn/training.pngz'results/models/weak_gnn/predictions.png) training_threshold_dicetraining_metrics_by_run deformation_target_preprocessinghard_constrainthard_constraint_max_violationsoft_constraintsphysics_constraints_unavailable checkpointfigureshistoryz report.json ensure_asciiindentzutf-8)encodingc 3K|]:}dtj|dd|dd|dd|ddd|ddd <y w) zrQz rrr.4frz N)htmlescape).0rows r ztrain..s C {{3u:&'(*+,--./0>"3'(?#C()  sAAz index.htmla Diagnostic weak-label GNN

CAD risk-area + deformation GNN

Diagnostic training-set fit only. ah unvalidated NG run(s), weak point-dilated masks, one CAD and one run-level class. These values do not measure generalization or production accuracy.

Input: STEP-derived graph + XLS process values/masks. Outputs: CAD-node risk probability, signed normal deformation, deformation magnitude, and a graph-level defect-type head.

Final loss: rz(; pooled training IoU: z); pooled training Dice: z%. Defect-type status: z.

uK
RunTarget nodesPredicted nodesTraining IoUTraining Dice

Training trace

Diagnostic GNN training curve

Fitted fields

Weak target, fitted risk and fitted deformation fields

Full JSON report · Diagnostic checkpoint

);rrrrlongr rkrer)optimAdamW parametersrt clamp_minclamprvr+train zero_gradF binary_cross_entropy_with_logitsr huber_lossrsquaremean cross_entropybackwardstepdetachritemsclonermrhload_state_dictevalno_gradr%rCreluru enumerateslicerrOUTmkdirsave plot_resultsrsr' relative_tor write_textjsondumpsjoinrr)8rrr/rWrXrKrYrZr[r\ signed_targetmagnitude_target class_targetmodel optimizerpositivepositive_weightsrcdst undirectedr selected_lossrselected_state type_statusroutputrdicerr risk_smoothdeformation_smoothrloss loss_valuer:r prediction intersectionunion dice_value iou_valueconstraint_violationper_runr~r,rQslrun_prediction run_targetrun_intersection run_unionrresult metrics_rowss8 rrrts`   D 4< Dfuzz)J$[)*IGDM5::4VEd9o&GwtF|/D$}-.KX@W tK01'$~J^`e`j`jBk< % 1 yq)7==+;SC]H[D\=] _E !!%"2"2"4UY!ZI **1-HK(83x?FF3FOOHCsJG%LMN/3N*-d=.A*BQ*F&LoKv y**t*<tT9gu=00 1EtXghf\2D9ll6*E#F ]`a !ALL8O1PRbjm$nn mmF<$89 "3z?3k#j/6RRZZ\aac . /J @ . /J @ ABH&(446  tM" #a '/C(DlSI3488:Q>ITzC+--k0AADK]D]]`ii )4;;=)  %&M"QYNLQL\L\L^LdLdLfgLfjc5c5<<>#7#7#99LfNgQYE#**,Y%*;+=+=+?%@,12D2K2K2M,Ny//12   />899 .) JJL tT9gu=mmF<$89  B& "[%5%5%77<<>CCE k..00557<<>,&#a1A1F1F1H;??K\KaKaKc1c*dd  3q%=0 $zz . / 3 3 5?V8W W YY\Y\Y^_c_c_e _%#DL1JE3uu}uqyE&9:B'^N$R--/J . ;@@BGGI '*499;@@BI NN),Z^^-=)>,/0B0B0D,E 03q)3D D!"%5!5A~?Q?Q?S?X?X?Z]g]k]k]m]r]r]t?t8u!u/4[_5H5H5J/K   2 6IIdTI*//J JJ&&(eM*?j$v,'((st  w'<E# !)IS#*"6^!";<- ])=46[\ { A, *0067:E/6s M= N c tjt}|jdd|jdtd|jdt d |j }|js td t|j|j}ttjd Dcic]}||| c}d d ycc}w)N) descriptionz--allow-diagnostic-weak-labels store_true)actionz--epochs<)typedefaultz--learning-rategMb`?zwRefusing weak/unvalidated targets. Pass --allow-diagnostic-weak-labels for the requested diagnostic overfit experiment.)rrrrrrFrr)argparseArgumentParser__doc__ add_argumentrrrv parse_argsallow_diagnostic_weak_labels SystemExitrrrprintrr)parserargsrr:s rmainrb2s  $ $ 9F 8N  b9 )tD    D  , ,ST T 4;; 2 2 3F $**3;<3;3c6#;&3;rrrrrb__name__rBrrrus #  u 4TT% &&5:MM/a20f_ Up#BL \ zFr