jxdZddlmZddlZddlZddlZddlmZddlZ ddl Z ee jjjZej j#deeddlmZmZmZddlmZmZddlmZddlmZmZmZddd Z dd Z!dd Z"d d ddZ#ddZ$e%dk(r e&e$y)aResearch-only 4H SOPR entry-inhibition ablation. This never changes MLP weights, promoted parameters, Pine code, or exits. It compares the promoted Python replay with the same replay in which an entry is permitted only when SOPR is above a discovery-calibrated low-tail cutoff. ) annotationsN)Path) OOS_START SCORE_START WFO_FOLDS)calculate_metricsgenerate_signals) load_params) DISCOVERY_ENDVALIDATION_STARTparse_chart_timec :||n+||dtj|kj}t||d}|j Dcic]<\}}|t |t tjfr t |n t|>c}}Scc}}w)Ntimer) score_start min_trades) pd Timestampcopyritems isinstancefloatnpfloatingint)signalsstartendwindowresultkeyvalues $tools/run_mlp_sopr_entry_ablation.py_metricsr#s WBLLQTDU1U)V)[)[)]F v5Q GF!,,. (JC Zr{{/CDU5\#e*T(  sABc,|dtj|k\}|||dtj|kz}|d|z}||z}|djdj|dj ddfd }||||dS) Nr execute_entrycloseg?c*|j}t|jtt|t|rt |j dzndt|rt |dkDj dSddS)Ndr)entriesnext_bar_observationsnext_bar_mean_return_pctnext_bar_positive_rate)dropnarsumlenrmean)rowsvalues next_returns r"describez$_entry_diagnostics..describe+sT"))+488:%(V%5FI&kfkkmc.A(BW[DGKeVaZ,=,=,?&@   VZ   )baseline_entriesblocked_baseline_entries)r2 pd.Seriesreturndict[str, float | int | None])rrshiftdivsub) rblockedrrmask candidatesselectedr5r4s @r"_entry_diagnosticsrC#s 6?bll51 1D  2<<#444)D0JG#H'"((,001ABFFsKK !) 4RZ[cRd eer6c||dtjtk\|dtj|kz}|jr t dt |dj |S)zGReturn a low-SOPR cutoff using only rows available through history_end.rz0no SOPR history available for cutoff calibration sopr_norm)rrrempty ValueErrorrquantile)data history_end feature_tailhistorys r"calibrate_sopr_cutoffrM7sgDLBLL$==$v,RTR^R^_jRkBklmG}}KLL %..|< ==r6g?)rKcd|cxkrdkstdtdtj|}|jjj jj |_d|jvr tdt|d|d<|jdjd}tt |}|jd std |t|t|}|d|kDjd jt }t#|j%fi|}t#|j%fd |i|}t#|j%fd t'j(t+|t i|} dD]'} || j-| | rt/d| tj0||j2} t4t tft t6t tj8t:tj<dz ft:dfd} | j?D cic]1\} \}}| tA|||tA|||tC|| ||d3}}} }i}tDD]\}}}t|||}|djG|jd}t#|j%fd |jt i|}|tA|||tA|||tC||||d||<t |t |d||gddddd||dScc}}} w)Nrg?z&feature_tail must be between 0 and 0.5rEzdata has no sopr_norm columnrT)dropmlp_weights_filez winner has no mlp_weights_file: F)dtype entry_allowed)positionr% execute_exitz%all-true entry mask changed baseline )index) nanoseconds) discovery validationoos)baselinesopr_entry_inhibitedentry_diagnostics)cutoffr[r\r]zewhen flat and the unchanged MLP entry condition fires, inhibit entry if sopr_norm <= discovery cutoff)z MLP weightszall promoted parametersz exit logicz Pine codezYstrategy metrics include the repository's 0.5% entry and 0.5% exit commission assumptionszgeach fold recalibrates its fixed 10th-percentile cutoff using only history before that fold's OOS startzNthis is a one-policy ablation, not a threshold sweep or a promotable candidate)policyrK sopr_cutoff unchangedcostswfo_calibrationwarning) data_path winner_path definitionwindows wfo_folds)$rGrread_csvcolumnsstrlowerstripr sort_values reset_indexr getrMr fillnato_numpyboolr rronesr0equalsAssertionErrorSeriesrUrr rr Timedeltarr#rCrle)rerfrKrIparamsr^rRr[gated all_allowedcolumnr?rhnamerr summariesfoldsis_end fold_cutoff fold_blocked fold_gateds r"runr?s^ | !c !ABB "ABB ;;y !D<<##))+//557DL$,,&788#DL1DL   F # / /T / :D [) *F ::( );K=IJJ "4 EF+&/77>GGdGSM 6v6H TYY[ P P PE"499;gbggc$iW[>\g`fgK?&&{6':; #H!QR R@yydjj99G!3}#56+,c",,y2IBLLefLg2g.hi4 2G#*--/  #2 D,5#  5#6$,UE3$?!3Hguc!R  #2 +-E's,D&,G K(++K8??E % IIK )5 '?'?d'?'K OU " 5#6$,Z$D!3HlESV!W  c (^;'}(!^p Ig  /s.6M*c tjt}|jdtd|jdtd|jdt|j }t |j|j}tj|d }t||jrE|jjjdd |jj|d zy ) N) descriptionz--dataT)typerequiredz--winnerz--output)r)indent)parentsexist_ok r)argparseArgumentParser__doc__ add_argumentr parse_argsrrIwinnerjsondumpsprintoutputparentmkdir write_text)parserargsrtexts r"mainrs  $ $ 9F td;  =  .    D DKK (F ::fQ 'D $K {{    = td{+ r6__main__)N)r pd.DataFramerrlr str | Noner:zdict[str, float | int]) rrr?r9rrlrrr:r;)rIrrJzstr | pd.TimestamprKrr:r)rerrfrrKrr:zdict[str, object])r:r)'r __future__rrrsyspathlibrnumpyrpandasr__file__resolverREPOpathinsertrlconfigrrrstrategies.strategy_mlp_scoresrr tools.check_mlp_parityr %tools.run_mlp_extreme_bar_event_studyr r r r#rCrMrr__name__ SystemExitr6r"rs #  H~&&--3t944N.ccf(>FJDN   z TV r6