crbn4/Building
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

modified: Figure_1.png

Browse Source 
modified:   checkpoints/forward/best_tcn_model.pt
	modified:   checkpoints/forward/training_history.csv
	new file:   checkpoints_mlp/task1_feature_mlp/best_feature_mlp.pt
	new file:   checkpoints_mlp/task1_feature_mlp/training_history.csv
	new file:   checkpoints_rms/forward_rms/best_rms_model.pt
	new file:   checkpoints_rms/forward_rms/training_history.csv
	modified:   evaluation_outputs/forward/evaluation_forward_test_b0_s0.png
	new file:   evaluation_outputs/forward/evaluation_forward_val_b0_s0.png
	new file:   evaluation_outputs/forward/evaluation_forward_val_b3_s0.png
	new file:   evaluation_outputs/forward_rms/evaluation_test_all_samples.csv
	new file:   evaluation_outputs/forward_rms/evaluation_test_s0.png
	new file:   evaluation_outputs/forward_rms/evaluation_train_all_samples.csv
	new file:   evaluation_outputs/forward_rms/evaluation_val_all_samples.csv
	new file:   evaluation_outputs/forward_rms/evaluation_val_s0.png
	new file:   evaluation_outputs/forward_rms/evaluation_val_s0_waveform.png
	new file:   evaluation_outputs/task1_feature_mlp/evaluation_train_all_samples.csv
	new file:   evaluation_outputs/task1_feature_mlp/evaluation_train_curve.png
	new file:   evaluation_outputs/task1_feature_mlp/evaluation_val_all_samples.csv
	new file:   evaluation_outputs/task1_feature_mlp/evaluation_val_curve.png
	new file:   evaluation_outputs/task1_feature_mlp/evaluation_val_s0.png
	new file:   evaluation_outputs/task1_feature_mlp/harmonic_5mm_0.75Hz_prediction.png
	new file:   evaluation_outputs/task1_feature_mlp/harmonic_5mm_1.55Hz_prediction.png
	new file:   scripts/__pycache__/config.cpython-310.pyc
	new file:   scripts/__pycache__/dataset.cpython-310.pyc
	new file:   scripts/__pycache__/model.cpython-310.pyc
	new file:   scripts/config.py
	new file:   scripts/dataset.py
	new file:   scripts/evaluate.py
	new file:   scripts/model.py
	new file:   scripts/predict_single.py
	new file:   scripts/train.py
	modified:   src/__pycache__/config.cpython-310.pyc
	modified:   src/__pycache__/dataset.cpython-310.pyc
	modified:   src/__pycache__/model.cpython-310.pyc
	modified:   src/config.py
	modified:   src/dataset.py
	modified:   src/model.py
	new file:   src_new/__pycache__/config.cpython-310.pyc
	new file:   src_new/__pycache__/dataset.cpython-310.pyc
	new file:   src_new/__pycache__/evaluate.cpython-310.pyc
	new file:   src_new/__pycache__/model.cpython-310.pyc
	new file:   src_new/__pycache__/train.cpython-310.pyc
	new file:   src_new/config.py
	new file:   src_new/dataset.py
	new file:   src_new/evaluate.py
	new file:   src_new/model.py
	new file:   src_new/train.py
This commit is contained in:
CrbnsCat10n
2026-05-06 12:19:55 +08:00
parent dcc023cc04
commit 484643409d
48 changed files with 2706 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
checkpoints/forward/training_history.csv
View File

@@ -1,14 +1,11 @@
epoch,lr,train_loss,train_time_loss,train_weighted_time_loss,train_fft_loss,train_rms_loss,train_scale_loss,train_underestimate_loss,train_envelope_loss,train_rms_error,train_dominant_freq_error,val_loss,val_time_loss,val_weighted_time_loss,val_fft_loss,val_rms_loss,val_scale_loss,val_underestimate_loss,val_envelope_loss,val_rms_error,val_dominant_freq_error
1,0.001,32.23490004269582,0.9063257234838774,0.09063257336757093,2462.537229142099,0.5202015545570625,0.5072746667659508,0.23183272828189833,0.9878060196368199,0.5202015545570625,0.6667674588373308,16.848691019518622,0.5030307245665583,0.05030307315033058,712.1304384428879,0.21638367042459292,0.3054182426682834,0.16712227438030572,0.7986705179872184,0.21638367042459292,0.9571996228951049
2,0.001,21.43670712776904,0.9317702313639084,0.09317702508338217,1694.0047123747052,0.35049294160222105,0.3616391454102858,0.1121402242625097,0.672775794312639,0.35049294160222105,0.6002240721274537,18.15417513354071,0.5811605972462687,0.058116060392609956,948.0657793242356,0.23726162314414978,0.3922087324076685,0.15394838554142365,0.7320991713425209,0.23726162314414978,0.8361816405868994
3,0.001,18.081004331696708,0.949055330933265,0.09490553458344261,1372.9708672289578,0.28145609332143134,0.3131425976753235,0.09374795892750318,0.6080609613432074,0.28145609332143134,0.616063318727439,16.73266759412042,0.5420598767954727,0.05420598895128431,673.0913622625943,0.238979727286717,0.33492567107595245,0.1804515516449665,0.7293170896069757,0.238979727286717,0.7597824622933206
4,0.001,15.158426779621053,0.9718681652590914,0.09718681784030402,1005.0683164776497,0.2189214069325969,0.2836238658934269,0.08367646379695046,0.5845890998278024,0.2189214069325969,0.6179393338571753,17.278400026518725,0.5685602380283947,0.056856024560743366,892.716255977236,0.22618299260221678,0.36168746264844104,0.1538948353444194,0.6990852653980255,0.22618299260221678,0.7597824622933206
5,0.001,13.61913053044733,0.9698394334541177,0.09698394513776842,859.7200766509434,0.19106561557020782,0.2557676170232161,0.07612784807833861,0.5538543634257227,0.19106561557020782,0.6155450957215706,17.31632186626566,0.5785790409507423,0.057857905097048856,895.00206572434,0.21634554451909557,0.35666010605877846,0.1514351657302729,0.7220557270378902,0.21634554451909557,0.7402091190436203
6,0.001,12.678725962368947,0.9694307513956754,0.0969430767351164,788.4346546676924,0.1713324544845887,0.23463592369039105,0.07080332469195127,0.5323761788741598,0.1713324544845887,0.6164817462027068,18.423015594482422,0.5721222347226637,0.057212224808232535,888.629751271215,0.25913861959144985,0.4007708651238474,0.16395779438959113,0.7525067822686557,0.25913861959144985,0.7218985721265917
7,0.001,12.268796533908484,0.967802310889622,0.09678023287429,744.56112426182,0.1675024907684551,0.2299804154713199,0.06777841198029665,0.5231116253812358,0.1675024907684551,0.5961900787542253,17.48041952067408,0.5510032737049563,0.055100328578003524,704.0817605380354,0.2379121222886546,0.37086750232967836,0.19612120528673305,0.7370918125941835,0.2379121222886546,0.7088496766840734
8,0.001,11.889431161700555,0.967772865070487,0.09677728824317455,712.115890215028,0.1614244244289848,0.2226250537161557,0.06655820054968573,0.5118434342010966,0.1614244244289848,0.6022199021486582,18.410391971982758,0.5461424337378864,0.05461424415738418,738.1983037488214,0.2533088120920905,0.3795644687167529,0.21159803308546543,0.7770289305982918,0.2533088120920905,0.707797346401937
9,0.001,11.234980547203207,0.9654024828155086,0.09654024967326308,670.3140682004532,0.14827435323089924,0.20859187953876998,0.062428659900038874,0.4916352005499714,0.14827435323089924,0.5623616312479263,16.51186601046858,0.5469307437025267,0.05469307598882708,658.5074381335028,0.23887184570575581,0.3477302913008065,0.18191729235494958,0.6951427665250055,0.23887184570575581,0.7065345500594454
10,0.001,10.992490152143082,0.9626774315564137,0.09626774419591112,649.039663350807,0.14625250653557056,0.20260991113928128,0.061842711396374796,0.48383018318212256,0.14625250653557056,0.568747647623218,17.501161114922887,0.5506949491541961,0.05506949608439002,671.2075416301859,0.2541997463538729,0.373840541675173,0.19259302314884705,0.7480303932880533,0.2541997463538729,0.708428744563363
11,0.0005,10.806822551871246,0.9680146614335617,0.09680146787245318,637.6420590382702,0.14451587228280194,0.2035975716305229,0.0572100906404403,0.4776927213061531,0.14451587228280194,0.5574953460125556,17.13407661174906,0.5288547395632185,0.05288547495829648,618.1973874322299,0.2417878899081,0.34800343832065317,0.21005815672206468,0.7388678706925491,0.2417878899081,0.7122171335564683
12,0.0005,10.255595112746617,0.9616962041494981,0.09616962194723903,599.1373590433373,0.12593858316540718,0.18720266560338578,0.05598305191247249,0.4703599948365733,0.12593858316540718,0.5670544286673502,16.41022307297279,0.5580467443014013,0.055804675894564594,691.4893851444639,0.23402822017669678,0.3565609496215294,0.17255781141334567,0.6764166149599798,0.23402822017669678,0.6966426454535165
13,0.0005,9.979247138185322,0.9624017550135558,0.09624017732885648,580.7163422782467,0.12185739905063836,0.18553815514973873,0.05258095577657926,0.4598706980358879,0.12185739905063836,0.5524671217591512,16.273903156148975,0.5295915886245924,0.05295915985158805,541.1639246447332,0.23793373128463483,0.3361044392503541,0.20454522344315873,0.7062619534032099,0.23793373128463483,0.7056926858376642
1,0.001,25.599940884788083,1.7562742283884085,0.1756274257347269,2774.0617503040244,0.48780856886,0.4243854204157613,0.1559595560549565,0.8805500887474924,0.48780856886,0.5304876364710495,11.634879671294113,0.5565754904829222,0.05565755005026686,1030.2479747903758,0.19599251089424924,0.30123303316790484,0.07480601036262795,0.6266409343686598,0.19599251089424924,1.0759024783184585
2,0.001,18.8950198911271,1.807937761522689,0.18079377879511635,1975.3159283332104,0.3459213998801303,0.35067820830165214,0.1092575388907824,0.758364588865694,0.3459213998801303,0.4821035959354592,10.542049144876414,0.5095152341086289,0.05095152441283752,783.4861005585769,0.16758740950247336,0.2732687505154774,0.10411370940634916,0.6904992646184461,0.16758740950247336,0.9969777073497166
3,0.001,16.69826183678969,1.7823544706938401,0.17823545016207784,1672.4529568654186,0.30989001663226,0.3281119924108937,0.10324830568905147,0.7008864249823228,0.30989001663226,0.5562505104268722,11.536806665617844,0.5393576467859333,0.05393576609163449,807.3707762093379,0.21286767027501402,0.33425827504232014,0.10436723167718999,0.6995668801768072,0.21286767027501402,1.0535930765001758
4,0.001,14.293646308611024,1.7581631035174963,0.1758163128540201,1394.1521462494472,0.25787185048157313,0.2936001386282579,0.08524404234200153,0.6856944434485346,0.25787185048157313,0.4707189468993712,11.0088351677204,0.512058621850507,0.05120586244196727,716.754319289635,0.18768149871250678,0.30934940792363264,0.12457355485972145,0.7022169437901727,0.18768149871250678,0.9801404230908486
5,0.001,13.64289491581467,1.7736638503254585,0.1773663880127781,1282.9595152656987,0.2586157521549261,0.2966467724093851,0.0801986690856657,0.662726102291413,0.2586157521549261,0.4816929557024733,12.69674718791041,0.5321119869577473,0.0532112003400408,923.869591548525,0.2018765033832912,0.3501281591838804,0.1300795058286267,0.763439338782738,0.2018765033832912,0.8944807380729035
6,0.001,11.776648678869572,1.7715154886245728,0.17715155192703572,1060.149999654518,0.21744662122625225,0.2669158032480276,0.07291070573067046,0.6286382666736279,0.21744662122625225,0.49275163173347086,13.566723692006079,0.5599699513665561,0.055996995740409554,1081.8859031940328,0.23347977780062576,0.399347482827203,0.09571884981966738,0.753588840879243,0.23347977780062576,0.802507071705832
7,0.001,10.598831523139522,1.8021629632643934,0.180216298631902,959.8165948256006,0.1918528508746399,0.24521326632151064,0.06225078732197015,0.5634953882896675,0.1918528508746399,0.43498506576279067,12.335502788938324,0.5373578256574171,0.053735782938270735,897.6921434073613,0.22004679693230267,0.3851976631016567,0.09088767872288309,0.7485740184783936,0.22004679693230267,0.7111648032890617
8,0.0005,10.096835316352124,1.771101622086651,0.17711016507643573,904.1345197569649,0.18218446649470418,0.2265465495721349,0.06017633231427028,0.5831915904890816,0.18218446649470418,0.4688892364850107,12.481576097422632,0.55741053922423,0.05574105519416003,993.8351429906385,0.20113462980451255,0.37394588877414836,0.08785425563310754,0.7251736295634302,0.20113462980451255,0.7193729794563382
9,0.0005,9.301918011791301,1.8092092241881028,0.18092092534281173,872.4488234609928,0.15496070823579464,0.2050890047454609,0.05094481654078612,0.5316191723324218,0.15496070823579464,0.45952377956299667,12.33887189010094,0.529184412339638,0.052918442107480146,888.4320715542498,0.2059096264941939,0.35693827361382285,0.11748063429820768,0.7306599390917811,0.2059096264941939,0.7187415812949122
10,0.0005,9.509747361237148,1.775009681593697,0.17750097143481364,875.1525979671839,0.15211023376235422,0.21900151374767413,0.054768192924488826,0.5556492130711393,0.15211023376235422,0.48492007759302524,11.80303616359316,0.5109482181483301,0.05109482271404102,766.1427895118451,0.20308723608995305,0.33445684663180647,0.13568100256138835,0.7249893361124499,0.20308723608995305,0.7126380656820888
1 epoch lr train_loss train_time_loss train_weighted_time_loss train_fft_loss train_rms_loss train_scale_loss train_underestimate_loss train_envelope_loss train_rms_error train_dominant_freq_error val_loss val_time_loss val_weighted_time_loss val_fft_loss val_rms_loss val_scale_loss val_underestimate_loss val_envelope_loss val_rms_error val_dominant_freq_error
2 1 0.001 32.23490004269582 25.599940884788083 0.9063257234838774 1.7562742283884085 0.09063257336757093 0.1756274257347269 2462.537229142099 2774.0617503040244 0.5202015545570625 0.48780856886 0.5072746667659508 0.4243854204157613 0.23183272828189833 0.1559595560549565 0.9878060196368199 0.8805500887474924 0.5202015545570625 0.48780856886 0.6667674588373308 0.5304876364710495 16.848691019518622 11.634879671294113 0.5030307245665583 0.5565754904829222 0.05030307315033058 0.05565755005026686 712.1304384428879 1030.2479747903758 0.21638367042459292 0.19599251089424924 0.3054182426682834 0.30123303316790484 0.16712227438030572 0.07480601036262795 0.7986705179872184 0.6266409343686598 0.21638367042459292 0.19599251089424924 0.9571996228951049 1.0759024783184585
3 2 0.001 21.43670712776904 18.8950198911271 0.9317702313639084 1.807937761522689 0.09317702508338217 0.18079377879511635 1694.0047123747052 1975.3159283332104 0.35049294160222105 0.3459213998801303 0.3616391454102858 0.35067820830165214 0.1121402242625097 0.1092575388907824 0.672775794312639 0.758364588865694 0.35049294160222105 0.3459213998801303 0.6002240721274537 0.4821035959354592 18.15417513354071 10.542049144876414 0.5811605972462687 0.5095152341086289 0.058116060392609956 0.05095152441283752 948.0657793242356 783.4861005585769 0.23726162314414978 0.16758740950247336 0.3922087324076685 0.2732687505154774 0.15394838554142365 0.10411370940634916 0.7320991713425209 0.6904992646184461 0.23726162314414978 0.16758740950247336 0.8361816405868994 0.9969777073497166
4 3 0.001 18.081004331696708 16.69826183678969 0.949055330933265 1.7823544706938401 0.09490553458344261 0.17823545016207784 1372.9708672289578 1672.4529568654186 0.28145609332143134 0.30989001663226 0.3131425976753235 0.3281119924108937 0.09374795892750318 0.10324830568905147 0.6080609613432074 0.7008864249823228 0.28145609332143134 0.30989001663226 0.616063318727439 0.5562505104268722 16.73266759412042 11.536806665617844 0.5420598767954727 0.5393576467859333 0.05420598895128431 0.05393576609163449 673.0913622625943 807.3707762093379 0.238979727286717 0.21286767027501402 0.33492567107595245 0.33425827504232014 0.1804515516449665 0.10436723167718999 0.7293170896069757 0.6995668801768072 0.238979727286717 0.21286767027501402 0.7597824622933206 1.0535930765001758
5 4 0.001 15.158426779621053 14.293646308611024 0.9718681652590914 1.7581631035174963 0.09718681784030402 0.1758163128540201 1005.0683164776497 1394.1521462494472 0.2189214069325969 0.25787185048157313 0.2836238658934269 0.2936001386282579 0.08367646379695046 0.08524404234200153 0.5845890998278024 0.6856944434485346 0.2189214069325969 0.25787185048157313 0.6179393338571753 0.4707189468993712 17.278400026518725 11.0088351677204 0.5685602380283947 0.512058621850507 0.056856024560743366 0.05120586244196727 892.716255977236 716.754319289635 0.22618299260221678 0.18768149871250678 0.36168746264844104 0.30934940792363264 0.1538948353444194 0.12457355485972145 0.6990852653980255 0.7022169437901727 0.22618299260221678 0.18768149871250678 0.7597824622933206 0.9801404230908486
6 5 0.001 13.61913053044733 13.64289491581467 0.9698394334541177 1.7736638503254585 0.09698394513776842 0.1773663880127781 859.7200766509434 1282.9595152656987 0.19106561557020782 0.2586157521549261 0.2557676170232161 0.2966467724093851 0.07612784807833861 0.0801986690856657 0.5538543634257227 0.662726102291413 0.19106561557020782 0.2586157521549261 0.6155450957215706 0.4816929557024733 17.31632186626566 12.69674718791041 0.5785790409507423 0.5321119869577473 0.057857905097048856 0.0532112003400408 895.00206572434 923.869591548525 0.21634554451909557 0.2018765033832912 0.35666010605877846 0.3501281591838804 0.1514351657302729 0.1300795058286267 0.7220557270378902 0.763439338782738 0.21634554451909557 0.2018765033832912 0.7402091190436203 0.8944807380729035
7 6 0.001 12.678725962368947 11.776648678869572 0.9694307513956754 1.7715154886245728 0.0969430767351164 0.17715155192703572 788.4346546676924 1060.149999654518 0.1713324544845887 0.21744662122625225 0.23463592369039105 0.2669158032480276 0.07080332469195127 0.07291070573067046 0.5323761788741598 0.6286382666736279 0.1713324544845887 0.21744662122625225 0.6164817462027068 0.49275163173347086 18.423015594482422 13.566723692006079 0.5721222347226637 0.5599699513665561 0.057212224808232535 0.055996995740409554 888.629751271215 1081.8859031940328 0.25913861959144985 0.23347977780062576 0.4007708651238474 0.399347482827203 0.16395779438959113 0.09571884981966738 0.7525067822686557 0.753588840879243 0.25913861959144985 0.23347977780062576 0.7218985721265917 0.802507071705832
8 7 0.001 12.268796533908484 10.598831523139522 0.967802310889622 1.8021629632643934 0.09678023287429 0.180216298631902 744.56112426182 959.8165948256006 0.1675024907684551 0.1918528508746399 0.2299804154713199 0.24521326632151064 0.06777841198029665 0.06225078732197015 0.5231116253812358 0.5634953882896675 0.1675024907684551 0.1918528508746399 0.5961900787542253 0.43498506576279067 17.48041952067408 12.335502788938324 0.5510032737049563 0.5373578256574171 0.055100328578003524 0.053735782938270735 704.0817605380354 897.6921434073613 0.2379121222886546 0.22004679693230267 0.37086750232967836 0.3851976631016567 0.19612120528673305 0.09088767872288309 0.7370918125941835 0.7485740184783936 0.2379121222886546 0.22004679693230267 0.7088496766840734 0.7111648032890617
9 8 0.001 0.0005 11.889431161700555 10.096835316352124 0.967772865070487 1.771101622086651 0.09677728824317455 0.17711016507643573 712.115890215028 904.1345197569649 0.1614244244289848 0.18218446649470418 0.2226250537161557 0.2265465495721349 0.06655820054968573 0.06017633231427028 0.5118434342010966 0.5831915904890816 0.1614244244289848 0.18218446649470418 0.6022199021486582 0.4688892364850107 18.410391971982758 12.481576097422632 0.5461424337378864 0.55741053922423 0.05461424415738418 0.05574105519416003 738.1983037488214 993.8351429906385 0.2533088120920905 0.20113462980451255 0.3795644687167529 0.37394588877414836 0.21159803308546543 0.08785425563310754 0.7770289305982918 0.7251736295634302 0.2533088120920905 0.20113462980451255 0.707797346401937 0.7193729794563382
10 9 0.001 0.0005 11.234980547203207 9.301918011791301 0.9654024828155086 1.8092092241881028 0.09654024967326308 0.18092092534281173 670.3140682004532 872.4488234609928 0.14827435323089924 0.15496070823579464 0.20859187953876998 0.2050890047454609 0.062428659900038874 0.05094481654078612 0.4916352005499714 0.5316191723324218 0.14827435323089924 0.15496070823579464 0.5623616312479263 0.45952377956299667 16.51186601046858 12.33887189010094 0.5469307437025267 0.529184412339638 0.05469307598882708 0.052918442107480146 658.5074381335028 888.4320715542498 0.23887184570575581 0.2059096264941939 0.3477302913008065 0.35693827361382285 0.18191729235494958 0.11748063429820768 0.6951427665250055 0.7306599390917811 0.23887184570575581 0.2059096264941939 0.7065345500594454 0.7187415812949122
11 10 0.001 0.0005 10.992490152143082 9.509747361237148 0.9626774315564137 1.775009681593697 0.09626774419591112 0.17750097143481364 649.039663350807 875.1525979671839 0.14625250653557056 0.15211023376235422 0.20260991113928128 0.21900151374767413 0.061842711396374796 0.054768192924488826 0.48383018318212256 0.5556492130711393 0.14625250653557056 0.15211023376235422 0.568747647623218 0.48492007759302524 17.501161114922887 11.80303616359316 0.5506949491541961 0.5109482181483301 0.05506949608439002 0.05109482271404102 671.2075416301859 766.1427895118451 0.2541997463538729 0.20308723608995305 0.373840541675173 0.33445684663180647 0.19259302314884705 0.13568100256138835 0.7480303932880533 0.7249893361124499 0.2541997463538729 0.20308723608995305 0.708428744563363 0.7126380656820888
11 0.0005 10.806822551871246 0.9680146614335617 0.09680146787245318 637.6420590382702 0.14451587228280194 0.2035975716305229 0.0572100906404403 0.4776927213061531 0.14451587228280194 0.5574953460125556 17.13407661174906 0.5288547395632185 0.05288547495829648 618.1973874322299 0.2417878899081 0.34800343832065317 0.21005815672206468 0.7388678706925491 0.2417878899081 0.7122171335564683
12 0.0005 10.255595112746617 0.9616962041494981 0.09616962194723903 599.1373590433373 0.12593858316540718 0.18720266560338578 0.05598305191247249 0.4703599948365733 0.12593858316540718 0.5670544286673502 16.41022307297279 0.5580467443014013 0.055804675894564594 691.4893851444639 0.23402822017669678 0.3565609496215294 0.17255781141334567 0.6764166149599798 0.23402822017669678 0.6966426454535165
13 0.0005 9.979247138185322 0.9624017550135558 0.09624017732885648 580.7163422782467 0.12185739905063836 0.18553815514973873 0.05258095577657926 0.4598706980358879 0.12185739905063836 0.5524671217591512 16.273903156148975 0.5295915886245924 0.05295915985158805 541.1639246447332 0.23793373128463483 0.3361044392503541 0.20454522344315873 0.7062619534032099 0.23793373128463483 0.7056926858376642