Add files using upload-large-folder tool

.gitattributes

@@ -33,3 +33,16 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+figures/confusion_matrix.png filter=lfs diff=lfs merge=lfs -text
+figures/confusion_matrix_row_normalized.png filter=lfs diff=lfs merge=lfs -text
+figures/epoch_vs_val_macro_f1.png filter=lfs diff=lfs merge=lfs -text
+figures/learning_curve_f1_all.png filter=lfs diff=lfs merge=lfs -text
+figures/learning_curve_accuracy.png filter=lfs diff=lfs merge=lfs -text
+figures/learning_curve_loss.png filter=lfs diff=lfs merge=lfs -text
+figures/per_class_f1.png filter=lfs diff=lfs merge=lfs -text
+figures/roc_macro.png filter=lfs diff=lfs merge=lfs -text
+figures/per_class_precision.png filter=lfs diff=lfs merge=lfs -text
+figures/per_class_recall.png filter=lfs diff=lfs merge=lfs -text
+figures/summary_metrics_bar.png filter=lfs diff=lfs merge=lfs -text
+figures/roc_ovr_per_class.png filter=lfs diff=lfs merge=lfs -text
+figures/summary_metrics_radar.png filter=lfs diff=lfs merge=lfs -text

config/class_to_idx.json

@@ -0,0 +1,9 @@
+{
+  "Brown Blight": 0,
+  "Gray Blight": 1,
+  "Green mirid bug": 2,
+  "Healthy leaf": 3,
+  "Helopeltis": 4,
+  "Red spider": 5,
+  "Tea algal leaf spot": 6
+}

config/classes.json

@@ -0,0 +1,9 @@
+[
+  "Brown Blight",
+  "Gray Blight",
+  "Green mirid bug",
+  "Healthy leaf",
+  "Helopeltis",
+  "Red spider",
+  "Tea algal leaf spot"
+]

config/config.json

@@ -0,0 +1,60 @@
+{
+  "model_name": "swin_small_patch4_window7_224",
+  "num_classes": 7,
+  "pretrained": true,
+  "drop_path_rate": 0.2,
+  "data_root": "/kaggle/input/tea-leaf701515/tea_leaf_processed_dataset",
+  "input_size": 224,
+  "batch_size": 64,
+  "num_workers": 0,
+  "img_mean": [
+    0.485,
+    0.456,
+    0.406
+  ],
+  "img_std": [
+    0.229,
+    0.224,
+    0.225
+  ],
+  "hsv_use_sincos": false,
+  "gate_warmup_epochs": 5,
+  "epochs": 100,
+  "warmup_epochs": 5,
+  "lr": 0.0005,
+  "warmup_lr_init": 1e-06,
+  "weight_decay": 0.05,
+  "min_lr": 1e-06,
+  "label_smoothing": 0.1,
+  "grad_clip_norm": 1.0,
+  "gradient_accumulation_steps": 1,
+  "color_jitter": 0.2,
+  "hue_jitter": 0.1,
+  "use_gaussian_blur": false,
+  "gaussian_blur_prob": 0.1,
+  "rrc_scale_min": 0.85,
+  "use_amp": true,
+  "use_ema": true,
+  "ema_decay": 0.9998,
+  "use_class_weights": false,
+  "use_hsv_branch": true,
+  "hsv_embed_dim": 128,
+  "hsv_dropout": 0.1,
+  "fuse_dropout": 0.2,
+  "gate_hidden": 256,
+  "gate_vector": false,
+  "compute_val_auc": false,
+  "early_stopping_patience": 25,
+  "run_dir": "/kaggle/working/experiments_swin",
+  "experiment_name": "swin_small_hsv_raw",
+  "log_interval": 50,
+  "save_cm_png": true,
+  "ckpt_temp_dir": "/kaggle/working/ckpt_cache",
+  "save_epoch_checkpoints": false,
+  "save_epoch_every": 5,
+  "keep_last_n_checkpoints": 3,
+  "device": "cuda",
+  "seed": 42,
+  "deterministic": true,
+  "use_data_parallel": true
+}

config/env.json

@@ -0,0 +1,10 @@
+{
+  "python": "Python 3.12.12",
+  "torch": "2.8.0+cu126",
+  "timm": "1.0.20",
+  "cuda_available": true,
+  "cuda_version": "12.6",
+  "cudnn_version": 91002,
+  "gpu0": "Tesla T4",
+  "num_gpus": 2
+}

logs/gate_stats_log.csv

@@ -0,0 +1,101 @@
+epoch,gate_global_mean,Brown Blight_mean,Brown Blight_std,Brown Blight_n,Gray Blight_mean,Gray Blight_std,Gray Blight_n,Green mirid bug_mean,Green mirid bug_std,Green mirid bug_n,Healthy leaf_mean,Healthy leaf_std,Healthy leaf_n,Helopeltis_mean,Helopeltis_std,Helopeltis_n,Red spider_mean,Red spider_std,Red spider_n,Tea algal leaf spot_mean,Tea algal leaf spot_std,Tea algal leaf spot_n
+1,0.1382163017988205,0.1410542130470276,0.00867896992713213,93,0.14350716769695282,0.008395800366997719,167,0.13349564373493195,0.008458040654659271,192,0.1354224532842636,0.008439570665359497,151,0.1366288810968399,0.006426404695957899,91,0.13967974483966827,0.007629912346601486,78,0.14088767766952515,0.007955428212881088,79
+2,0.14228865504264832,0.14588911831378937,0.009250386618077755,93,0.14812611043453217,0.008851083926856518,167,0.13798649609088898,0.008835618384182453,192,0.13786791265010834,0.00877518393099308,151,0.14064176380634308,0.0067110322415828705,91,0.1435007005929947,0.007997498847544193,78,0.1453162133693695,0.008243312127888203,79
+3,0.1456960141658783,0.15044312179088593,0.009975088760256767,93,0.15186643600463867,0.009372085332870483,167,0.14206600189208984,0.009146024473011494,192,0.13906192779541016,0.009088476188480854,151,0.14375589787960052,0.0071413652040064335,91,0.1466064453125,0.00842051487416029,78,0.14990234375,0.00866785179823637,79
+4,0.14981620013713837,0.156005859375,0.01092145498842001,93,0.15622952580451965,0.010134470649063587,167,0.1474262923002243,0.009625573642551899,192,0.1401561200618744,0.00958365947008133,151,0.14725033938884735,0.00793854333460331,91,0.1504751294851303,0.009133405983448029,78,0.1555500328540802,0.009275893680751324,79
+5,0.15626448392868042,0.16424494981765747,0.012538047507405281,93,0.16217078268527985,0.011649545282125473,167,0.1562703400850296,0.010489840060472488,192,0.14261166751384735,0.010341460816562176,151,0.1526985615491867,0.00916422437876463,91,0.15634076297283173,0.010533244349062443,78,0.1644982397556305,0.010484952479600906,79
+6,0.16254860162734985,0.1727793663740158,0.0146824074909091,93,0.16737228631973267,0.014068752527236938,167,0.1664155274629593,0.01218014769256115,192,0.14320141077041626,0.011131586506962776,151,0.15772020816802979,0.011160950176417828,91,0.16136130690574646,0.01249663345515728,78,0.17462390661239624,0.012314105406403542,79
+7,0.17011359333992004,0.1832682341337204,0.01801832765340805,93,0.17380912601947784,0.01778285950422287,167,0.17791683971881866,0.01517103798687458,192,0.14417150616645813,0.012258559465408325,151,0.16266405582427979,0.013995962217450142,91,0.1677418202161789,0.015335802920162678,78,0.1883591264486313,0.015204573050141335,79
+8,0.17948782444000244,0.19622737169265747,0.02250082604587078,93,0.1815730184316635,0.022995304316282272,167,0.1926593780517578,0.019654981791973114,192,0.14495687186717987,0.013670764863491058,151,0.16864751279354095,0.017655648291110992,91,0.17538844048976898,0.01924682967364788,78,0.20589862763881683,0.018964217975735664,79
+9,0.1898115873336792,0.21186943352222443,0.028323393315076828,93,0.1905883103609085,0.029816173017024994,167,0.2090180665254593,0.025453142821788788,192,0.14446577429771423,0.015062456019222736,151,0.17415006458759308,0.022215792909264565,91,0.18297244608402252,0.02437014877796173,78,0.22699052095413208,0.02366919443011284,79
+10,0.19942344725131989,0.22617003321647644,0.035222068428993225,93,0.19732628762722015,0.037340618669986725,167,0.22609710693359375,0.03227730095386505,192,0.14271919429302216,0.016094328835606575,151,0.17830581963062286,0.027191828936338425,91,0.19002200663089752,0.030173489823937416,78,0.24953489005565643,0.02891855686903,79
+11,0.21003882586956024,0.24354340136051178,0.04406864196062088,93,0.202839195728302,0.04705251008272171,167,0.24603207409381866,0.04118303582072258,192,0.13933800160884857,0.01718825101852417,151,0.18244816362857819,0.033438701182603836,91,0.19591033458709717,0.03790561854839325,78,0.2792072594165802,0.035334616899490356,79
+12,0.2219642847776413,0.26311665773391724,0.05509238690137863,93,0.2075367122888565,0.05834495648741722,167,0.26975885033607483,0.052080247551202774,192,0.1348412185907364,0.01817084290087223,151,0.18745170533657074,0.04088093712925911,91,0.20034711062908173,0.047373563051223755,78,0.31548377871513367,0.04236360266804695,79
+13,0.2360948920249939,0.28620368242263794,0.06840476393699646,93,0.21333175897598267,0.07104714959859848,167,0.29640260338783264,0.06467005610466003,192,0.1306128054857254,0.019311295822262764,151,0.19339291751384735,0.04963360354304314,91,0.20524635910987854,0.05887629836797714,78,0.3599189221858978,0.04982099309563637,79
+14,0.25121352076530457,0.31037622690200806,0.08373699337244034,93,0.21923023462295532,0.08453457057476044,167,0.325061172246933,0.07893320173025131,192,0.12705376744270325,0.020493023097515106,151,0.1992824673652649,0.05952998623251915,91,0.20775271952152252,0.07253441214561462,78,0.40974676609039307,0.05652161315083504,79
+15,0.26236122846603394,0.32856470346450806,0.09876101464033127,93,0.221846342086792,0.09657986462116241,167,0.3494046628475189,0.09276726841926575,192,0.12122592329978943,0.02107929065823555,151,0.20085328817367554,0.06843526661396027,91,0.20561805367469788,0.08512869477272034,78,0.4551631212234497,0.06118977442383766,79
+16,0.27411791682243347,0.3493402898311615,0.1157117411494255,93,0.22342339158058167,0.10905927419662476,167,0.3752473294734955,0.10791672766208649,192,0.11399628967046738,0.02160514146089554,151,0.20187948644161224,0.07842867821455002,91,0.20331592857837677,0.0988435223698616,78,0.5061189532279968,0.06465886533260345,79
+17,0.2863384485244751,0.3722015619277954,0.13398094475269318,93,0.22441640496253967,0.12125145643949509,167,0.40203413367271423,0.12409550696611404,192,0.10640731453895569,0.0220659039914608,151,0.20338121056556702,0.08913661539554596,91,0.1973603069782257,0.11294372379779816,78,0.5623022317886353,0.0668594166636467,79
+18,0.29785749316215515,0.3939025104045868,0.15309497714042664,93,0.22548872232437134,0.13143473863601685,167,0.4251619875431061,0.1402982622385025,192,0.09980723261833191,0.022379690781235695,151,0.20429673790931702,0.09960824996232986,91,0.19169577956199646,0.12605947256088257,78,0.6195176243782043,0.06755118817090988,79
+19,0.30571117997169495,0.4119558036327362,0.17174287140369415,93,0.22312480211257935,0.13862678408622742,167,0.4404700696468353,0.15516014397144318,192,0.09318815171718597,0.022416863590478897,151,0.2022383064031601,0.1083831638097763,91,0.18464386463165283,0.13651084899902344,78,0.6726444959640503,0.06714701652526855,79
+20,0.3118903636932373,0.43169575929641724,0.19039148092269897,93,0.22086942195892334,0.14451676607131958,167,0.4487762451171875,0.1695740967988968,192,0.08633624762296677,0.022066663950681686,151,0.19994781911373138,0.11673875153064728,91,0.17617523670196533,0.1456279754638672,78,0.7246464490890503,0.065232053399086,79
+21,0.31569111347198486,0.4467504322528839,0.20771852135658264,93,0.21689391136169434,0.14832830429077148,167,0.45346799492836,0.1827118843793869,192,0.08032691478729248,0.021740121766924858,151,0.19713886082172394,0.1240735799074173,91,0.1651885211467743,0.15223939716815948,78,0.770436704158783,0.061718400567770004,79
+22,0.31790709495544434,0.4575785994529724,0.22456373274326324,93,0.2109895795583725,0.148758202791214,167,0.4568605422973633,0.1938333362340927,192,0.07583396136760712,0.021461280062794685,151,0.1894749253988266,0.1301707923412323,91,0.15649257600307465,0.15838567912578583,78,0.8117984533309937,0.05835099145770073,79
+23,0.31786802411079407,0.4627436101436615,0.2406141757965088,93,0.2019326239824295,0.14743265509605408,167,0.4607032239437103,0.2026752084493637,192,0.07101905345916748,0.020982781425118446,151,0.18025459349155426,0.13441196084022522,91,0.14548511803150177,0.1616971641778946,78,0.8457958102226257,0.05513099953532219,79
+24,0.31493332982063293,0.4595310688018799,0.2555944621562958,93,0.18889155983924866,0.1422659009695053,167,0.4615701138973236,0.21044886112213135,192,0.06545515358448029,0.020063932985067368,151,0.17468160390853882,0.13831506669521332,91,0.13168060779571533,0.1615692377090454,78,0.8741099834442139,0.052874501794576645,79
+25,0.31246694922447205,0.45772430300712585,0.2701643407344818,93,0.178728848695755,0.1382339894771576,167,0.4602985382080078,0.2177983820438385,192,0.06059497594833374,0.019147370010614395,151,0.16928502917289734,0.142006978392601,91,0.12116437405347824,0.16275149583816528,78,0.9001310467720032,0.049747563898563385,79
+26,0.30819734930992126,0.4558282792568207,0.2833600342273712,93,0.16828013956546783,0.13367144763469696,167,0.45141348242759705,0.2240493893623352,192,0.05599995702505112,0.01812892220914364,151,0.16291122138500214,0.14507293701171875,91,0.11609453707933426,0.16617797315120697,78,0.9211827516555786,0.04571248218417168,79
+27,0.3016328811645508,0.4510268270969391,0.2938721179962158,93,0.15668949484825134,0.12779805064201355,167,0.4366172254085541,0.2278604507446289,192,0.05246921628713608,0.01719842664897442,151,0.15566480159759521,0.14609917998313904,91,0.11150810122489929,0.16852925717830658,78,0.936208188533783,0.04268023744225502,79
+28,0.2927577495574951,0.4400808811187744,0.30269697308540344,93,0.14416320621967316,0.11996407806873322,167,0.4199329912662506,0.2285679131746292,192,0.04880953207612038,0.015891199931502342,151,0.14480826258659363,0.1441352516412735,91,0.10557731986045837,0.16859038174152374,78,0.9458749294281006,0.04139940068125725,79
+29,0.2841106951236725,0.42667412757873535,0.309760719537735,93,0.13159380853176117,0.111671581864357,167,0.40665945410728455,0.22921264171600342,192,0.04497240111231804,0.014430874958634377,151,0.13402850925922394,0.141646146774292,91,0.09932317584753036,0.16660605370998383,78,0.9532671570777893,0.04048902168869972,79
+30,0.2761184871196747,0.4159562289714813,0.3160567581653595,93,0.12065864354372025,0.10411450266838074,167,0.39237213134765625,0.22977636754512787,192,0.04180695861577988,0.013008400797843933,151,0.12548811733722687,0.13965466618537903,91,0.09327423572540283,0.1640194058418274,78,0.9594911932945251,0.03990757837891579,79
+31,0.26876914501190186,0.40484651923179626,0.32196617126464844,93,0.11124534159898758,0.09778957813978195,167,0.37802696228027344,0.22951515018939972,192,0.03949621319770813,0.011998520232737064,151,0.11913760751485825,0.1376214176416397,91,0.0883636474609375,0.16170324385166168,78,0.9647448658943176,0.039874281734228134,79
+32,0.2617676258087158,0.3934660851955414,0.32735687494277954,93,0.10393519699573517,0.09255935251712799,167,0.36361709237098694,0.2284703403711319,192,0.03671557828783989,0.01107505988329649,151,0.11284109205007553,0.13475631177425385,91,0.08445583283901215,0.16008450090885162,78,0.969621479511261,0.03948076069355011,79
+33,0.25493571162223816,0.38388749957084656,0.33137810230255127,93,0.09747862815856934,0.08800920099020004,167,0.3496362268924713,0.22809401154518127,192,0.03378690034151077,0.010022709146142006,151,0.10527558624744415,0.13061663508415222,91,0.07982204854488373,0.1568363904953003,78,0.9738182425498962,0.03805107995867729,79
+34,0.2467111498117447,0.3715387284755707,0.3342493176460266,93,0.09083475172519684,0.08235965669155121,167,0.3315216600894928,0.22591887414455414,192,0.03129789978265762,0.009079452604055405,151,0.09699609875679016,0.12522515654563904,91,0.07545784115791321,0.15291917324066162,78,0.9764326810836792,0.03896477445960045,79
+35,0.2388567477464676,0.35867342352867126,0.3364614248275757,93,0.08478258550167084,0.07655325531959534,167,0.3133762776851654,0.2224610447883606,192,0.029333151876926422,0.008413898758590221,151,0.0909445658326149,0.12029307335615158,91,0.07161262631416321,0.15073461830615997,78,0.9783858060836792,0.04039757698774338,79
+36,0.22993408143520355,0.34359118342399597,0.3367360830307007,93,0.07991146296262741,0.0700964480638504,167,0.29120779037475586,0.2175084352493286,192,0.027446720749139786,0.007823113352060318,151,0.08484356105327606,0.11400937288999557,91,0.06723804771900177,0.1469503492116928,78,0.979152262210846,0.043235406279563904,79
+37,0.22201590240001678,0.33166536688804626,0.33729302883148193,93,0.07428453862667084,0.062214817851781845,167,0.2719860076904297,0.21235014498233795,192,0.025659523904323578,0.0072021642699837685,151,0.07958850264549255,0.10864346474409103,91,0.06351412087678909,0.14424079656600952,78,0.9796528816223145,0.04626334458589554,79
+38,0.2146243005990982,0.3199305236339569,0.33799177408218384,93,0.06885569542646408,0.0549592450261116,167,0.25471043586730957,0.20609882473945618,192,0.02413354255259037,0.006615066900849342,151,0.07424960285425186,0.10184471309185028,91,0.05974128842353821,0.14133265614509583,78,0.9800978899002075,0.04918605461716652,79
+39,0.20899856090545654,0.3098571002483368,0.3382943570613861,93,0.0642067939043045,0.048680149018764496,167,0.2436557561159134,0.20129168033599854,192,0.0225243978202343,0.006046498194336891,151,0.0685163214802742,0.09269697964191437,91,0.05666644871234894,0.1389903426170349,78,0.9807654023170471,0.05147730931639671,79
+40,0.20374590158462524,0.29979583621025085,0.33887767791748047,93,0.061046302318573,0.0438813753426075,167,0.2314293384552002,0.1969377100467682,192,0.021365588530898094,0.0056811124086380005,151,0.06450627744197845,0.08571420609951019,91,0.05452552065253258,0.1371164619922638,78,0.9813711643218994,0.0532693937420845,79
+41,0.19626393914222717,0.29199761152267456,0.3386501669883728,93,0.057773612439632416,0.040124282240867615,167,0.21022558212280273,0.19219763576984406,192,0.0197320394217968,0.00528682442381978,151,0.057236891239881516,0.07601705938577652,91,0.05229245871305466,0.13561300933361053,78,0.982106626033783,0.05362428352236748,79
+42,0.18987086415290833,0.28462547063827515,0.33800598978996277,93,0.05511748790740967,0.03739665076136589,167,0.19250257313251495,0.18799039721488953,192,0.01835576817393303,0.004812329076230526,151,0.051124151796102524,0.06727457046508789,91,0.0500590018928051,0.13318155705928802,78,0.9824836850166321,0.05511930212378502,79
+43,0.18478913605213165,0.2758818566799164,0.33877140283584595,93,0.05255958437919617,0.03515307977795601,167,0.18019135296344757,0.18409600853919983,192,0.017623648047447205,0.00460773054510355,151,0.045816611498594284,0.057561278343200684,91,0.04852549359202385,0.13227710127830505,78,0.9823909997940063,0.057894714176654816,79
+44,0.17954014241695404,0.2689560055732727,0.3401031494140625,93,0.05043039098381996,0.03371357545256615,167,0.16602277755737305,0.1797264665365219,192,0.016750816255807877,0.004383142571896315,151,0.04054352641105652,0.047505225986242294,91,0.046955011785030365,0.13182257115840912,78,0.9822302460670471,0.06148459389805794,79
+45,0.175197571516037,0.2595749795436859,0.34147679805755615,93,0.04702046141028404,0.031130310148000717,167,0.15790796279907227,0.17651017010211945,192,0.01615375280380249,0.004230291582643986,151,0.036486197263002396,0.038950365036726,91,0.04501587152481079,0.1299402415752411,78,0.9811547994613647,0.06819700449705124,79
+46,0.17310455441474915,0.25096672773361206,0.3434067964553833,93,0.04392831772565842,0.028947170823812485,167,0.15761296451091766,0.1752299666404724,192,0.015958521515130997,0.004074422176927328,151,0.03410263732075691,0.03290349617600441,91,0.043644342571496964,0.12950024008750916,78,0.98046875,0.07309229671955109,79
+47,0.16952265799045563,0.24543172121047974,0.3446449935436249,93,0.04249965772032738,0.02829546295106411,167,0.14879751205444336,0.17356029152870178,192,0.015242014080286026,0.003844745457172394,151,0.030390560626983643,0.02749553695321083,91,0.042410336434841156,0.13008449971675873,78,0.9797084927558899,0.07851114124059677,79
+48,0.16739462316036224,0.23924468457698822,0.34553587436676025,93,0.0400397926568985,0.02660844847559929,167,0.1470678597688675,0.17267701029777527,192,0.014888182282447815,0.0037041958421468735,151,0.028454288840293884,0.02426089346408844,91,0.04089091345667839,0.12939664721488953,78,0.977878987789154,0.08781227469444275,79
+49,0.16403499245643616,0.23610778152942657,0.3464118540287018,93,0.038379691541194916,0.025681328028440475,167,0.13821101188659668,0.16994337737560272,192,0.014075828716158867,0.003456117119640112,151,0.025737889111042023,0.02110331691801548,91,0.039514102041721344,0.1286863088607788,78,0.976457417011261,0.09562251716852188,79
+50,0.16126936674118042,0.23163579404354095,0.34720781445503235,93,0.03625940531492233,0.024358374997973442,167,0.13223011791706085,0.1663511097431183,192,0.013586082495748997,0.0033242497593164444,151,0.024003248661756516,0.018841516226530075,91,0.038026466965675354,0.1269134283065796,78,0.975351095199585,0.10221146047115326,79
+51,0.1583871990442276,0.2282349020242691,0.34824177622795105,93,0.03444740176200867,0.023270580917596817,167,0.1248699426651001,0.1624268740415573,192,0.013026559725403786,0.003127847332507372,151,0.022284790873527527,0.016969390213489532,91,0.03633792698383331,0.12368154525756836,78,0.9747422337532043,0.10677875578403473,79
+52,0.15487226843833923,0.2255253940820694,0.3493185043334961,93,0.03269191086292267,0.02247174084186554,167,0.11446849256753922,0.15809366106987,192,0.012328141368925571,0.002923692809417844,151,0.02013506181538105,0.014910401776432991,91,0.034976959228515625,0.12148068845272064,78,0.9742137789726257,0.11074857413768768,79
+53,0.15255261957645416,0.22202424705028534,0.35024017095565796,93,0.03067414090037346,0.021327396854758263,167,0.10944338887929916,0.15496055781841278,192,0.011990427039563656,0.002809605561196804,151,0.018834544345736504,0.013753142207860947,91,0.033502284437417984,0.11863149702548981,78,0.9734257459640503,0.11547864228487015,79
+54,0.1514977067708969,0.21750928461551666,0.3514144718647003,93,0.028687117621302605,0.020275646820664406,167,0.1100444421172142,0.15459774434566498,192,0.011963598430156708,0.0028206550050526857,151,0.018265776336193085,0.013316025026142597,91,0.03198061138391495,0.11624419689178467,78,0.9723255634307861,0.12130915373563766,79
+55,0.15029790997505188,0.2137995958328247,0.35218214988708496,93,0.02700764499604702,0.019358839839696884,167,0.10924474149942398,0.15337766706943512,192,0.012037668377161026,0.0027955377008765936,151,0.017679905518889427,0.012855077162384987,91,0.030302291736006737,0.11414733529090881,78,0.971452534198761,0.12674234807491302,79
+56,0.14947760105133057,0.2105484902858734,0.3525206744670868,93,0.025551939383149147,0.01872500404715538,167,0.10968562215566635,0.1526261270046234,192,0.012120581232011318,0.002800719579681754,151,0.017103509977459908,0.012435435317456722,91,0.028802089393138885,0.11281384527683258,78,0.9704357981681824,0.13233743607997894,79
+57,0.1469230353832245,0.20766285061836243,0.35255497694015503,93,0.024512238800525665,0.018311940133571625,167,0.10264226049184799,0.14854760468006134,192,0.011573968455195427,0.002653503092005849,151,0.015697773545980453,0.01130917388945818,91,0.027785129845142365,0.1121520921587944,78,0.9692985415458679,0.13839785754680634,79
+58,0.1452769637107849,0.20516188442707062,0.35245633125305176,93,0.023566285148262978,0.018070010468363762,167,0.09895078092813492,0.14595544338226318,192,0.011283747851848602,0.002584562636911869,151,0.014707963913679123,0.010509244166314602,91,0.02664913237094879,0.11138663440942764,78,0.9682987928390503,0.14390213787555695,79
+59,0.14340779185295105,0.2032783329486847,0.3527628183364868,93,0.02312231995165348,0.017970232293009758,167,0.09341367334127426,0.1429583579301834,192,0.010877218097448349,0.0024885081220418215,151,0.013666258193552494,0.009706706739962101,91,0.02582012116909027,0.11112921684980392,78,0.9675725698471069,0.1480710506439209,79
+60,0.14178405702114105,0.2012239694595337,0.352646142244339,93,0.022525856271386147,0.01780971884727478,167,0.08906390517950058,0.14073294401168823,192,0.010517701506614685,0.002400677651166916,151,0.012890742160379887,0.00926937721669674,91,0.025086721405386925,0.11092796176671982,78,0.9666369557380676,0.15268240869045258,79
+61,0.14071416854858398,0.1987266093492508,0.352120578289032,93,0.021522156894207,0.01721419394016266,167,0.08758284896612167,0.13989956676959991,192,0.010233215987682343,0.002329040551558137,151,0.012411683797836304,0.009144725278019905,91,0.024313902482390404,0.11081777513027191,78,0.9656317830085754,0.15681101381778717,79
+62,0.13936010003089905,0.19633238017559052,0.350849986076355,93,0.020807208493351936,0.01684260554611683,167,0.0843343734741211,0.1373903900384903,192,0.009976115077733994,0.0022547647822648287,151,0.011871589347720146,0.008857186883687973,91,0.023776372894644737,0.11077634245157242,78,0.964914858341217,0.1599254012107849,79
+63,0.13819602131843567,0.1942201852798462,0.3495364189147949,93,0.02014036849141121,0.01648634858429432,167,0.08166050910949707,0.13551941514015198,192,0.009682156145572662,0.002158380812034011,151,0.011370145715773106,0.008587238378822803,91,0.023270435631275177,0.11076577752828598,78,0.9644087553024292,0.1623648852109909,79
+64,0.13725410401821136,0.19263802468776703,0.3482154905796051,93,0.019725456833839417,0.016487719491124153,167,0.07913371175527573,0.13368986546993256,192,0.009445746429264545,0.0020991575438529253,151,0.010946756228804588,0.008419759571552277,91,0.022884221747517586,0.11077247560024261,78,0.964464008808136,0.16357208788394928,79
+65,0.1359560191631317,0.19094930589199066,0.3465971052646637,93,0.01936182752251625,0.016375768929719925,167,0.07531791925430298,0.1307927817106247,192,0.009144485928118229,0.0020066755823791027,151,0.010323450900614262,0.007952678017318249,91,0.022462062537670135,0.1107819452881813,78,0.9642218351364136,0.16523241996765137,79
+66,0.13505986332893372,0.1890851855278015,0.34492120146751404,93,0.018929829820990562,0.016227107495069504,167,0.07320491224527359,0.12868016958236694,192,0.008960244245827198,0.0019599259831011295,151,0.009925590828061104,0.007761155255138874,91,0.022093553096055984,0.1107984110713005,78,0.963985800743103,0.16668426990509033,79
+67,0.13395564258098602,0.1875254362821579,0.3434065580368042,93,0.01875382848083973,0.01643364317715168,167,0.06975112110376358,0.12648899853229523,192,0.008755488321185112,0.0019160077208653092,151,0.009420961141586304,0.007578838150948286,91,0.02173379808664322,0.11076847463846207,78,0.9640209078788757,0.16746100783348083,79
+68,0.13318076729774475,0.18614868819713593,0.34245043992996216,93,0.01841649040579796,0.016531167551875114,167,0.06772776693105698,0.12495788931846619,192,0.008595877327024937,0.00189407158177346,151,0.009003272280097008,0.007218674756586552,91,0.02130620926618576,0.11068834364414215,78,0.9641337394714355,0.1678631603717804,79
+69,0.1325986683368683,0.1843327283859253,0.34127694368362427,93,0.01797841675579548,0.01661684550344944,167,0.06680818647146225,0.12375596910715103,192,0.00845400057733059,0.0018751169554889202,151,0.008737333118915558,0.007039591670036316,91,0.02089708298444748,0.11060752719640732,78,0.9641433954238892,0.16807231307029724,79
+70,0.1317313015460968,0.18239757418632507,0.33985617756843567,93,0.01769697479903698,0.016884533688426018,167,0.0645633116364479,0.12154125422239304,192,0.008313564583659172,0.0018722570966929197,151,0.008455549366772175,0.006893009878695011,91,0.02055131457746029,0.11051982641220093,78,0.9640634059906006,0.1683942675590515,79
+71,0.13057154417037964,0.1804398149251938,0.3386974036693573,93,0.01749623380601406,0.017338387668132782,167,0.061162035912275314,0.11932965368032455,192,0.008093372918665409,0.001847102539613843,151,0.008056975901126862,0.006532127037644386,91,0.020223861560225487,0.11037671566009521,78,0.9637690782546997,0.16900095343589783,79
+72,0.1297169327735901,0.17873772978782654,0.3376661241054535,93,0.017218653112649918,0.017799194902181625,167,0.05886119604110718,0.11740538477897644,192,0.007939724251627922,0.0018379437969997525,151,0.007758172228932381,0.0063146017491817474,91,0.019911399111151695,0.11017701774835587,78,0.9636918306350708,0.1691044121980667,79
+73,0.12969650328159332,0.17590390145778656,0.3357871174812317,93,0.016600465402007103,0.01829429902136326,167,0.060999851673841476,0.11659683287143707,192,0.007981836795806885,0.0018902559531852603,151,0.00773993693292141,0.006406460888683796,91,0.01948791928589344,0.1099870428442955,78,0.9632753729820251,0.169765904545784,79
+74,0.12862172722816467,0.17361249029636383,0.33342939615249634,93,0.01630881242454052,0.018461724743247032,167,0.05812537670135498,0.11437316238880157,192,0.0077460459433496,0.001843046979047358,151,0.007360479328781366,0.006079100538045168,91,0.019171176478266716,0.10967890918254852,78,0.9631984829902649,0.16917596757411957,79
+75,0.12804833054542542,0.17167240381240845,0.331370085477829,93,0.01592930778861046,0.018511313945055008,167,0.05708356574177742,0.11255180090665817,192,0.007592523470520973,0.0018163120839744806,151,0.007139101158827543,0.005964352283626795,91,0.018845118582248688,0.10931854695081711,78,0.963510274887085,0.1679055094718933,79
+76,0.12729601562023163,0.1698349118232727,0.3291342556476593,93,0.015695560723543167,0.018839769065380096,167,0.05507137253880501,0.11061275005340576,192,0.007365813944488764,0.001776444842107594,151,0.006868530064821243,0.005745772738009691,91,0.018560457974672318,0.10895604640245438,78,0.9639797806739807,0.16646428406238556,79
+77,0.1268409639596939,0.16784881055355072,0.32717663049697876,93,0.015382504090666771,0.019386617466807365,167,0.054351408034563065,0.10951795428991318,192,0.007250097580254078,0.0017646399792283773,151,0.006708585191518068,0.005675754975527525,91,0.018253521993756294,0.10848500579595566,78,0.9645360708236694,0.16553156077861786,79
+78,0.126071035861969,0.16583724319934845,0.32455283403396606,93,0.015079429373145103,0.0194917730987072,167,0.052430372685194016,0.10714148730039597,192,0.007025775499641895,0.0017226471100002527,151,0.006483895238488913,0.005475425627082586,91,0.017966417595744133,0.10784560441970825,78,0.9648908972740173,0.16458113491535187,79
+79,0.12516917288303375,0.16411073505878448,0.3222690522670746,93,0.014980453066527843,0.020048892125487328,167,0.049492280930280685,0.10470800846815109,192,0.006746228784322739,0.0016653576167300344,151,0.006202928256243467,0.0051862141117453575,91,0.017726212739944458,0.10714755207300186,78,0.9656534194946289,0.16305489838123322,79
+80,0.12470585107803345,0.1621374785900116,0.3205712139606476,93,0.014760708436369896,0.02092759683728218,167,0.04856489226222038,0.10285817831754684,192,0.006614963058382273,0.0016496715834364295,151,0.006084546912461519,0.005143462214618921,91,0.017447642982006073,0.10645337402820587,78,0.966366171836853,0.161773219704628,79
+81,0.12407362461090088,0.1606443226337433,0.3191568851470947,93,0.014649922028183937,0.021905507892370224,167,0.046699535101652145,0.10114360600709915,192,0.006397057790309191,0.001616648631170392,151,0.00587976910173893,0.004970450885593891,91,0.01723487488925457,0.10603125393390656,78,0.9669436812400818,0.16063806414604187,79
+82,0.12352821975946426,0.15832096338272095,0.3173077702522278,93,0.014459735713899136,0.023083167150616646,167,0.04559698700904846,0.09927467256784439,192,0.006269050762057304,0.0016032509738579392,151,0.0057472605258226395,0.004864664748311043,91,0.016946278512477875,0.10500514507293701,78,0.9675673246383667,0.1593705713748932,79
+83,0.12286943197250366,0.15564818680286407,0.3148995637893677,93,0.014136319980025291,0.02349960245192051,167,0.044497013092041016,0.09732721745967865,192,0.006086109671741724,0.0015707805287092924,151,0.00560796121135354,0.004754871129989624,91,0.016562022268772125,0.1032659187912941,78,0.9678640365600586,0.15860824286937714,79
+84,0.12229049950838089,0.15395954251289368,0.31384778022766113,93,0.013944031670689583,0.024619076400995255,167,0.04294956848025322,0.09561968594789505,192,0.005929707083851099,0.0015387823805212975,151,0.005448309704661369,0.004582930356264114,91,0.016267605125904083,0.10185224562883377,78,0.96855628490448,0.15711437165737152,79
+85,0.12183791399002075,0.15341296792030334,0.31405946612358093,93,0.013890477828681469,0.026495547965168953,167,0.04119637608528137,0.09383770823478699,192,0.0057213385589420795,0.0014974446967244148,151,0.0052681597881019115,0.004413191694766283,91,0.016013365238904953,0.1007675975561142,78,0.9695554375648499,0.15519411861896515,79
+86,0.12154067307710648,0.1523897796869278,0.3139520585536957,93,0.013779828324913979,0.028412260115146637,167,0.04037933424115181,0.09226380288600922,192,0.005597461946308613,0.0014735291479155421,151,0.0051574078388512135,0.004343552049249411,91,0.015711955726146698,0.09935444593429565,78,0.9704396724700928,0.1536901891231537,79
+87,0.12106702476739883,0.15120245516300201,0.3132609724998474,93,0.013807673938572407,0.030634118244051933,167,0.03889249637722969,0.09036197513341904,192,0.00542758172377944,0.0014393178280442953,151,0.005012952256947756,0.004187608137726784,91,0.01540868729352951,0.09766358137130737,78,0.9710803627967834,0.15267197787761688,79
+88,0.12073215842247009,0.1501169502735138,0.31304529309272766,93,0.013780320063233376,0.033390406519174576,167,0.037966538220644,0.08843118697404861,192,0.00530420383438468,0.0014138096012175083,151,0.004901592619717121,0.00410974258556962,91,0.015108353458344936,0.09619484096765518,78,0.9717198610305786,0.15191563963890076,79
+89,0.12033922225236893,0.14937226474285126,0.31293490529060364,93,0.013824816793203354,0.03625663369894028,167,0.03653693199157715,0.08693194389343262,192,0.005161784123629332,0.0013899659970775247,151,0.004758918657898903,0.003968823701143265,91,0.014835162088274956,0.09472352266311646,78,0.9724505543708801,0.1509229689836502,79
+90,0.12013569474220276,0.14842954277992249,0.31262290477752686,93,0.013802065514028072,0.03893967345356941,167,0.03614126518368721,0.08580540120601654,192,0.005057587753981352,0.0013771768426522613,151,0.004691951908171177,0.003967313561588526,91,0.014503014273941517,0.09292451292276382,78,0.9729816913604736,0.14995642006397247,79
+91,0.11980102211236954,0.1475868672132492,0.31224575638771057,93,0.013871472328901291,0.04216957092285156,167,0.03510016202926636,0.08478496968746185,192,0.004924142733216286,0.0013506632531061769,151,0.004575079772621393,0.0038580610416829586,91,0.014173703268170357,0.0909588411450386,78,0.9734671115875244,0.14888671040534973,79
+92,0.11958242952823639,0.14661754667758942,0.3115168809890747,93,0.014028195291757584,0.046071913093328476,167,0.034535735845565796,0.08412256091833115,192,0.004805381875485182,0.0013320258585736156,151,0.004492707084864378,0.003815439296886325,91,0.013881707563996315,0.08926667273044586,78,0.9739041924476624,0.147905632853508,79
+93,0.11919249594211578,0.14607585966587067,0.3111006021499634,93,0.014216725714504719,0.04934651404619217,167,0.03305874392390251,0.08334904909133911,192,0.004625566769391298,0.0012945481576025486,151,0.0043468475341796875,0.003652335610240698,91,0.01364382728934288,0.08779148012399673,78,0.9742791056632996,0.14704366028308868,79
+94,0.11891059577465057,0.14516617357730865,0.3102452754974365,93,0.014301745221018791,0.05226233974099159,167,0.032291799783706665,0.08276300877332687,192,0.00451558455824852,0.0012744141276925802,151,0.0042563010938465595,0.003583137411624193,91,0.013335692696273327,0.08582339435815811,78,0.9746163487434387,0.14590740203857422,79
+95,0.11867201328277588,0.14430460333824158,0.30940160155296326,93,0.014398232102394104,0.0557277649641037,167,0.03165706992149353,0.08218210935592651,192,0.004407510161399841,0.0012560590403154492,151,0.004172461573034525,0.0035293286200612783,91,0.013085218146443367,0.08446043729782104,78,0.9749495387077332,0.1447126269340515,79
+96,0.11844654381275177,0.14338389039039612,0.30836576223373413,93,0.014482029713690281,0.058465808629989624,167,0.031098833307623863,0.08191431313753128,192,0.00430662278085947,0.001237343531101942,151,0.0040946584194898605,0.003477780381217599,91,0.012807197868824005,0.08271142095327377,78,0.9753412008285522,0.143198162317276,79
+97,0.11813247203826904,0.1424827128648758,0.3070046901702881,93,0.014635337516665459,0.061198536306619644,167,0.030080646276474,0.08154789358377457,192,0.004195775371044874,0.0012116719735786319,151,0.003992332145571709,0.0033593715634196997,91,0.012564549222588539,0.08107016980648041,78,0.9757387042045593,0.14137250185012817,79
+98,0.11787455528974533,0.1411629319190979,0.3054828643798828,93,0.014632630161941051,0.06344300508499146,167,0.02963249944150448,0.08115552365779877,192,0.004101671278476715,0.0011973984073847532,151,0.003925092983990908,0.003334728768095374,91,0.012269790284335613,0.07926729321479797,78,0.9761572480201721,0.1393057256937027,79
+99,0.11759176105260849,0.139820858836174,0.30389928817749023,93,0.014643372036516666,0.06531362980604172,167,0.029012998566031456,0.08094100654125214,192,0.003994297701865435,0.0011739636538550258,151,0.0038444078527390957,0.0032673701643943787,91,0.011997565627098083,0.07751781493425369,78,0.9767407774925232,0.13645285367965698,79
+100,0.11731333285570145,0.13836130499839783,0.3021310865879059,93,0.01471888367086649,0.06719660013914108,167,0.028344938531517982,0.08070316910743713,192,0.0038884144742041826,0.0011538966791704297,151,0.003767747199162841,0.003201874205842614,91,0.011724129319190979,0.07560145109891891,78,0.977484405040741,0.1326824426651001,79

logs/train_log.csv

@@ -0,0 +1,101 @@
+epoch,lr,train_loss,train_acc1,val_loss,val_acc1,val_macro_f1,val_micro_f1,val_weighted_f1,val_macro_precision,val_macro_recall,val_macro_auc,val_gate_global_mean
+1,0.00010080000000000005,1.3908220419369393,0.5608274503365621,1.9776228380539442,0.11750881316098707,0.0857364529437987,0.11750881316098707,0.07106468301573295,0.06896633019396721,0.14942776117891324,-1.0,0.1382163017988205
+2,0.00020059999999999967,0.7350130474561541,0.8740765063208011,1.967904258558809,0.13396004700352526,0.10748057599607762,0.13396004700352526,0.08712407571643092,0.08686108015652448,0.1729560206389306,-1.0,0.14228865504264832
+3,0.00030039999999999917,0.6729173774379479,0.903135774092924,1.955628945880715,0.15746180963572268,0.1339313955782462,0.15746180963572268,0.1071536321191382,0.10747777232081232,0.2070745764618628,-1.0,0.1456960141658783
+4,0.0004001999999999987,0.6638125806112357,0.9080610737153176,1.9391662428718055,0.17861339600470036,0.15792789444930241,0.17861339600470036,0.12567937074838642,0.19754503339270393,0.2380472863851467,-1.0,0.14981620013713837
+5,0.0005,0.6618826663899238,0.9134789032999507,1.9189131830049597,0.2209165687426557,0.2019788047874198,0.2209165687426557,0.16500091598602545,0.2854132840613012,0.2939614903070111,-1.0,0.15626448392868042
+6,0.0004998635877065607,0.6724555298938009,0.9108520768346741,1.894214528007597,0.2796709753231492,0.26873323155397577,0.2796709753231492,0.22695587194544628,0.35732123120924353,0.3677915335804873,-1.0,0.16254860162734985
+7,0.000499454499991084,0.6357897746555129,0.9174191429978658,1.8647351888315937,0.32667450058754405,0.3271663346927411,0.32667450058754405,0.28937573283091866,0.5540637851138671,0.4163632284138659,-1.0,0.17011359333992004
+8,0.0004987731841849799,0.6168189207882427,0.9277622722048925,1.8301400881956662,0.3889541715628672,0.39940501961102204,0.3889541715628672,0.3708948066962515,0.6164568397210423,0.47269595621756305,-1.0,0.17948782444000244
+9,0.0004978203852970802,0.5967175014068724,0.9412247578394353,1.7898118449714293,0.4606345475910693,0.48132553374020814,0.4606345475910693,0.46042588250612876,0.666655933101118,0.5363662536117123,-1.0,0.1898115873336792
+10,0.000496597145198979,0.5803039723616127,0.9458217041536694,1.74392947326956,0.5346650998824912,0.5548696658815543,0.5346650998824912,0.5472031860945265,0.7029668179420936,0.5963371318960896,-1.0,0.19942344725131989
+11,0.0004951048014857652,0.5740164745971499,0.9486127072730258,1.693804628980706,0.5969447708578144,0.612440856583573,0.5969447708578144,0.6144258070948417,0.7285087106276679,0.6462874982820217,-1.0,0.21003882586956024
+12,0.0004933449860133774,0.5689613245876527,0.9520604170087014,1.6383264620912061,0.6556991774383079,0.6646891817515135,0.6556991774383079,0.6735817104414558,0.7550082597612464,0.6935087114362685,-1.0,0.2219642847776413
+13,0.0004913196231141983,0.5772521313218095,0.9477918240026268,1.5794607470934876,0.6897767332549941,0.6985055125023216,0.6897767332549941,0.7097977075176422,0.7753300115669416,0.723231466088059,-1.0,0.2360948920249939
+14,0.0004890309274928212,0.5454350984697368,0.9610901329830898,1.5178145159005278,0.7238542890716804,0.73267507712127,0.7238542890716804,0.745366998576213,0.7967393458098038,0.7526822953380515,-1.0,0.25121352076530457
+15,0.0004864814018043065,0.5651946111845669,0.9551797734362174,1.4526872078764452,0.7567567567567568,0.7628865147302958,0.7567567567567568,0.7769662280982038,0.8124991454803011,0.7816566165818104,-1.0,0.26236122846603394
+16,0.0004836738339175637,0.5555888061706625,0.9576424232474142,1.382095862359474,0.7849588719153937,0.7851252594569665,0.7849588719153937,0.8025754066495304,0.8219763043533582,0.8022281156937804,-1.0,0.27411791682243347
+17,0.00048061129386685553,0.5468882523505212,0.9628960761779675,1.3122997548689432,0.8061104582843713,0.8023137215111129,0.8061104582843713,0.8206294905056574,0.8300095188511882,0.8176069322346338,-1.0,0.2863384485244751
+18,0.00047729713049476415,0.5396052404413707,0.9647020193728452,1.2425811030189524,0.827262044653349,0.8215690511230023,0.827262044653349,0.8396570364306877,0.8458190659486118,0.8335138401234189,-1.0,0.29785749316215515
+19,0.0004737349677902782,0.5251606226339264,0.9696273189952389,1.1733587782755581,0.8507638072855465,0.8418504736088678,0.8507638072855465,0.8590994750201377,0.8577215560013275,0.8526380894093207,-1.0,0.30571117997169495
+20,0.00046992870092601687,0.5201644873192424,0.9748809719257922,1.1089710605130492,0.8648648648648649,0.8558151459409572,0.8648648648648649,0.871645232570063,0.8679825043168294,0.8664130000882919,-1.0,0.3118903636932373
+21,0.00046588249199891235,0.5190874668857457,0.9732392053849942,1.0479686148858658,0.8766157461809636,0.8660829017243604,0.8766157461809636,0.8823940705888519,0.8763717645820647,0.8759506359625403,-1.0,0.31569111347198486
+22,0.0004616007654790212,0.5258363440615695,0.9686422590707602,0.989844783072466,0.8871915393654524,0.8755285546355972,0.8871915393654524,0.8915394360583828,0.8841051975912958,0.8837395079451158,-1.0,0.31790709495544434
+23,0.00045708820337142904,0.5170383159602693,0.9738959120013134,0.9340034252608285,0.8977673325499412,0.8855344669371382,0.8977673325499412,0.9011008305564551,0.8921089241717732,0.8921157314700611,-1.0,0.31786802411079407
+24,0.0004523497400965474,0.5230878312987777,0.9709407322278772,0.8826805344759787,0.9071680376028202,0.8961494471150614,0.9071680376028202,0.909947734115002,0.9006002150393757,0.902580298550515,-1.0,0.31493332982063293
+25,0.0004473905570943985,0.5034866668167735,0.9794779182400263,0.836172174062628,0.9118683901292597,0.9008139999332839,0.9118683901292597,0.9142233589123873,0.9048052301236789,0.9065713084661676,-1.0,0.31246694922447205
+26,0.00044221607715878457,0.5140769486404996,0.9747167952717124,0.7940767085230028,0.9142185663924794,0.9027133847786325,0.9142185663924794,0.916538478313084,0.9070594389857538,0.9081707880783717,-1.0,0.30819734930992126
+27,0.0004368319585075445,0.510354492493479,0.9763585618125102,0.7557552379110585,0.9200940070505288,0.9082050051944606,0.9200940070505288,0.9220834366239532,0.9116890224251016,0.9125222412852405,-1.0,0.3016328811645508
+28,0.00043124408859537453,0.49755851624243325,0.9802988015104253,0.7201850099793332,0.927144535840188,0.9145233724078364,0.927144535840188,0.928436573733253,0.9183316873729713,0.917320680121853,-1.0,0.2927577495574951
+29,0.0004254585776759835,0.49273654275918277,0.983582334592021,0.6878663562019339,0.9283196239717979,0.9161148055200766,0.9283196239717979,0.929506268631284,0.9192065803606331,0.9188567784321447,-1.0,0.2841106951236725
+30,0.0004194817521206201,0.5041687665872632,0.977343621736989,0.6579268554402855,0.9330199764982373,0.9221610353505606,0.9330199764982373,0.9339263375878261,0.9231870405900604,0.9256594995205801,-1.0,0.2761184871196747
+31,0.00041332014750027773,0.504332787071498,0.9781645050073879,0.631423396453454,0.9330199764982373,0.9218964503582943,0.9330199764982373,0.9337507578250341,0.9234947553793543,0.9249788332034449,-1.0,0.26876914501190186
+32,0.0004069805014391451,0.5027044568913576,0.9789853882777869,0.6069516676853463,0.9341950646298472,0.9234336303108636,0.9341950646298472,0.9348348622069869,0.9245902101424358,0.9265486919161606,-1.0,0.2617676258087158
+33,0.0004004697462471122,0.4926504614640488,0.9832539812838614,0.5847853650497634,0.936545240893067,0.9261339595669195,0.936545240893067,0.9371742064418042,0.9277191825072578,0.928940222219609,-1.0,0.25493571162223816
+34,0.0003937950013393921,0.4831682262681345,0.9881792809062552,0.5643481237487703,0.9388954171562868,0.9281206064091434,0.9388954171562868,0.9392372730547887,0.9296541619512801,0.9305397018318133,-1.0,0.2467111498117447
+35,0.0003869635654515432,0.4795132137797106,0.9886718108684945,0.5457618808073947,0.9388954171562868,0.928047705137695,0.9388954171562868,0.939113900871726,0.9291356334795084,0.9305397018318133,-1.0,0.2388567477464676
+36,0.0003799829086584113,0.48336275565058623,0.9871942209817763,0.5293749063051405,0.9377203290246768,0.9265615170239351,0.9377203290246768,0.9376884320784633,0.9285272132732535,0.9281777924278531,-1.0,0.22993408143520355
+37,0.00037286066420571,0.4811554110572757,0.9894926941388934,0.5143624097283943,0.9400705052878966,0.929687062242953,0.9400705052878966,0.9401758808734244,0.9324412612390713,0.9304579383571926,-1.0,0.22201590240001678
+38,0.0003656046201631808,0.48410041832587264,0.9858808077491381,0.5008288156832147,0.9400705052878966,0.929687062242953,0.9400705052878966,0.9401758808734244,0.9324412612390713,0.9304579383571926,-1.0,0.2146243005990982
+39,0.000358222710908452,0.4860124641051141,0.9870300443276966,0.48860801407929455,0.9400705052878966,0.9297495593780835,0.9400705052878966,0.9402486861745648,0.9321017142376878,0.93018571840347,-1.0,0.20899856090545654
+40,0.00035072300845091396,0.4813997786956318,0.9883434575603349,0.4793005480959609,0.9388954171562868,0.9280961335264944,0.9388954171562868,0.9390658219878123,0.9306673937973579,0.9286496200931781,-1.0,0.20374590158462524
+41,0.0003431137136050944,0.47526996022393364,0.9894926941388934,0.46523104658838443,0.9412455934195064,0.9309057255786265,0.9412455934195064,0.9412896627617495,0.9344237185779632,0.9296814762304498,-1.0,0.19626393914222717
+42,0.0003354031470231884,0.47401191310095014,0.991462813987851,0.4535493067302659,0.9424206815511164,0.9320195072205976,0.9424206815511164,0.9424102660073712,0.9359422316156557,0.9305369082236064,-1.0,0.18987086415290833
+43,0.00032759974009654777,0.4754933392917399,0.9886718108684945,0.4444352014364563,0.9412455934195064,0.9305677145036074,0.9412455934195064,0.9411975540344428,0.9349882223756716,0.9287054063921045,-1.0,0.18478913605213165
+44,0.00031971202573607835,0.47548810089384075,0.9901494007552126,0.4351572833876492,0.9388954171562868,0.9278627509350917,0.9388954171562868,0.9388513461609955,0.9314280016744136,0.9261530405090426,-1.0,0.17954014241695404
+45,0.0003117486290416276,0.46562098976309674,0.9942538171072074,0.42729801899117675,0.9388954171562868,0.9278627509350917,0.9388954171562868,0.9388513461609955,0.9314280016744136,0.9261530405090426,-1.0,0.175197571516037
+46,0.00030371825787056314,0.4750240609830206,0.9899852241011328,0.42263410269303553,0.9388954171562868,0.9283592914831633,0.9388954171562868,0.9388893802456078,0.9318270389916694,0.9272173111540092,-1.0,0.17310455441474915
+47,0.00029562969331585953,0.4730196357840017,0.9909702840256116,0.41438596351586554,0.9412455934195064,0.9308301582965137,0.9412455934195064,0.9411469094282957,0.9342022078113356,0.9297696770370714,-1.0,0.16952265799045563
+48,0.00028749178010410264,0.46827438992832393,0.9931045805286488,0.40884833337277277,0.9388954171562868,0.9283592914831633,0.9388954171562868,0.9388893802456078,0.9318270389916694,0.9272173111540092,-1.0,0.16739462316036224
+49,0.00027931341692390995,0.4654188673546349,0.9942538171072074,0.40121484951463066,0.9377203290246768,0.9267395195785004,0.9377203290246768,0.9377602254018582,0.9300985018753444,0.9254089928899949,-1.0,0.16403499245643616
+50,0.00027110354669534516,0.46340707349103966,0.9945821704153669,0.3955816849278227,0.9388954171562868,0.9278627509350917,0.9388954171562868,0.9388513461609955,0.9314280016744136,0.9261530405090426,-1.0,0.16126936674118042
+51,0.0002628711467909672,0.4664463660504074,0.9931045805286488,0.38971112088927373,0.9388954171562868,0.9278876559418864,0.9388954171562868,0.9387603059688735,0.9314842379434031,0.9264252604627651,-1.0,0.1583871990442276
+52,0.0002546252192192047,0.46133796413472045,0.9957314069939255,0.38316493946292846,0.9400705052878966,0.9296593044061181,0.9400705052878966,0.9399430469219383,0.9329834493263356,0.927961358773057,-1.0,0.15487226843833923
+53,0.0002463747807807917,0.4621405169707843,0.9950747003776063,0.37845999839443156,0.9377203290246768,0.927240436306392,0.9377203290246768,0.937581146509989,0.9305343088714146,0.9255698284696086,-1.0,0.15255261957645416
+54,0.0002381288532090291,0.465658201869535,0.993761287144968,0.3752673771910886,0.9377203290246768,0.927240436306392,0.9377203290246768,0.937581146509989,0.9305343088714146,0.9255698284696086,-1.0,0.1514977067708969
+55,0.00022989645330465135,0.46201838069783313,0.9945821704153669,0.37156452972236165,0.9400705052878966,0.9295519605886957,0.9400705052878966,0.9398963548524386,0.9327550132072734,0.9279951191754809,-1.0,0.15029790997505188
+56,0.00022168658307608662,0.4616780016119307,0.9949105237235265,0.36793412248900575,0.9412455934195064,0.9311552489005301,0.9412455934195064,0.9410137770848085,0.9344232906272426,0.9298034374394953,-1.0,0.14947760105133057
+57,0.00021350821989589381,0.4631414296855638,0.9950747003776063,0.3630745484461936,0.9412455934195064,0.9311552489005301,0.9412455934195064,0.9410137770848085,0.9344232906272426,0.9298034374394953,-1.0,0.1469230353832245
+58,0.00020537030668413706,0.45906062555555793,0.9958955836480052,0.3588960429896479,0.9435957696827262,0.9340637828945987,0.9435957696827262,0.9434309223155862,0.9368096949872929,0.9329093944625031,-1.0,0.1452769637107849
+59,0.00019728174212943383,0.45365352578555357,0.9985224101132819,0.3543484571327474,0.9447708578143361,0.9357575739243817,0.9447708578143361,0.9446376271923798,0.9384330716106694,0.934445492772795,-1.0,0.14340779185295105
+60,0.00018925137095836925,0.4530364658969158,0.998358233459202,0.35025946118296525,0.9459459459459459,0.9366126795926831,0.9459459459459459,0.9458301521129965,0.9393805266160635,0.9351895403918425,-1.0,0.14178405702114105
+61,0.0001812879742639184,0.45400755740292376,0.998358233459202,0.3472150193328163,0.9447708578143361,0.9349188885629001,0.9447708578143361,0.9446234472362027,0.9377571499926869,0.9336534420815507,-1.0,0.14071416854858398
+62,0.00017340025990344944,0.4523577916810047,0.9988507634214415,0.3437663520194949,0.9459459459459459,0.9361249838716558,0.9459459459459459,0.9458279678967195,0.9385490953983215,0.9352233007942665,-1.0,0.13936010003089905
+63,0.000165596852976809,0.45490874769143175,0.9981940568051223,0.3407002502196544,0.9459459459459459,0.9361249838716558,0.9459459459459459,0.9458279678967195,0.9385490953983215,0.9352233007942665,-1.0,0.13819602131843567
+64,0.00015788628639490284,0.45141753279366253,0.9988507634214415,0.3378794372922245,0.9459459459459459,0.936393167724364,0.9459459459459459,0.9457915406418724,0.9392210754974986,0.9352233007942665,-1.0,0.13725410401821136
+65,0.00015027699154908296,0.4530874482811505,0.9981940568051223,0.33475452272788336,0.9459459459459459,0.936393167724364,0.9459459459459459,0.9457915406418724,0.9392210754974986,0.9352233007942665,-1.0,0.1359560191631317
+66,0.000142777289091545,0.45203483032780983,0.998358233459202,0.3323103916841164,0.9447708578143361,0.935252627404518,0.9447708578143361,0.9446164673524895,0.9376545775382885,0.934479253175219,-1.0,0.13505986332893372
+67,0.00013539537983681642,0.4533498602495897,0.9981940568051223,0.32910113283526043,0.9435957696827262,0.9343954186154145,0.9435957696827262,0.9434206209343896,0.9367107594463906,0.9337352055561713,-1.0,0.13395564258098602
+68,0.00012813933579428725,0.4550192875729819,0.9980298801510425,0.32667779936493496,0.9424206815511164,0.9330526564993298,0.9424206815511164,0.9422624492876921,0.9357167944622777,0.9319268872921568,-1.0,0.13318076729774475
+69,0.00012101709134158592,0.45059922376185524,0.9990149400755213,0.32468149731358126,0.9435957696827262,0.9342803430290356,0.9435957696827262,0.943428076615384,0.9365395253251372,0.9334967460048726,-1.0,0.1325986683368683
+70,0.0001140364345484537,0.4514251067228964,0.9990149400755213,0.3222819770545153,0.9435957696827262,0.9342803430290356,0.9435957696827262,0.943428076615384,0.9365395253251372,0.9334967460048726,-1.0,0.1317313015460968
+71,0.00010720499866060473,0.4505591306354079,0.999179116729601,0.3195850983629776,0.9435957696827262,0.9342803430290356,0.9435957696827262,0.943428076615384,0.9365395253251372,0.9334967460048726,-1.0,0.13057154417037964
+72,0.00010053025375288452,0.45125170012174853,0.9988507634214415,0.31734078324079235,0.9435957696827262,0.9342803430290356,0.9435957696827262,0.943428076615384,0.9365395253251372,0.9334967460048726,-1.0,0.1297169327735901
+73,9.401949856085179e-05,0.4510819575622582,0.9990149400755213,0.3159670251937086,0.9447708578143361,0.9359744929630723,0.9447708578143361,0.9446351783313952,0.9381624680066203,0.9350328443151644,-1.0,0.12969650328159332
+74,8.767985249971935e-05,0.45085413682795455,0.999179116729601,0.3134864605831904,0.9459459459459459,0.9371175153416351,0.9459459459459459,0.9458138052622572,0.939712519129874,0.9357768919342122,-1.0,0.12862172722816467
+75,8.151824787937694e-05,0.4497990544510483,0.9995074700377606,0.31147868173102794,0.9459459459459459,0.9371175153416351,0.9459459459459459,0.9458138052622572,0.939712519129874,0.9357768919342122,-1.0,0.12804833054542542
+76,7.554142232401354e-05,0.4499944573825773,0.9995074700377606,0.30941840610548976,0.9459459459459459,0.9371175153416351,0.9459459459459459,0.9458138052622572,0.939712519129874,0.9357768919342122,-1.0,0.12729601562023163
+77,6.975591140462275e-05,0.4489256559951105,0.9998358233459202,0.30803503717714975,0.9459459459459459,0.9371175153416351,0.9459459459459459,0.9458138052622572,0.939712519129874,0.9357768919342122,-1.0,0.1268409639596939
+78,6.416804149245308e-05,0.4494119148509997,0.9995074700377606,0.3061724248770121,0.9459459459459459,0.9371175153416351,0.9459459459459459,0.9458138052622572,0.939712519129874,0.9357768919342122,-1.0,0.126071035861969
+79,5.8783922841213185e-05,0.4497741232353211,0.9993432933836809,0.30415030336828264,0.9459459459459459,0.9371175153416351,0.9459459459459459,0.9458138052622572,0.939712519129874,0.9357768919342122,-1.0,0.12516917288303375
+80,5.360944290559942e-05,0.4499496643550289,0.9995074700377606,0.302954130146114,0.9471210340775558,0.938740704997402,0.9471210340775558,0.9469462022232932,0.9413927421412918,0.9375852101982266,-1.0,0.12470585107803345
+81,4.8650259903450266e-05,0.4490372505539882,0.9998358233459202,0.3014898592941069,0.9459459459459459,0.9379318152112467,0.9459459459459459,0.9457151642833983,0.9412788625478798,0.9360153514855108,-1.0,0.12407362461090088
+82,4.3911796628568647e-05,0.4498804659409743,0.9990149400755213,0.30010921775104016,0.9447708578143361,0.9365892675571672,0.9447708578143361,0.9445587925396143,0.9403209916724362,0.9342070332214963,-1.0,0.12352821975946426
+83,3.9399234520976274e-05,0.44955352783457553,0.9993432933836809,0.29838660534484407,0.9459459459459459,0.9382935504307139,0.9459459459459459,0.94577145769875,0.9419925183424466,0.9357431315317882,-1.0,0.12286943197250366
+84,3.5117508001085116e-05,0.4485327209613727,0.9998358233459202,0.29675235629922214,0.9459459459459459,0.9381391300036953,0.9459459459459459,0.9457183250906153,0.9426729785646953,0.9349510808405439,-1.0,0.12229049950838089
+85,3.107129907398077e-05,0.4482425531814925,1.0,0.29501468163819766,0.9459459459459459,0.9385979619345765,0.9459459459459459,0.9456725897294025,0.9432985235673177,0.9349742644080313,-1.0,0.12183791399002075
+86,2.726503220971929e-05,0.4493891918596582,0.9996716466918404,0.2933808625024578,0.9459459459459459,0.9385979619345765,0.9459459459459459,0.9456725897294025,0.9432985235673177,0.9349742644080313,-1.0,0.12154067307710648
+87,2.3702869505233765e-05,0.44870492068833856,0.9996716466918404,0.291717207214667,0.9471210340775558,0.9396641236264643,0.9471210340775558,0.9468014200200555,0.9448071420518477,0.9357183120270788,-1.0,0.12106702476739883
+88,2.0388706133142422e-05,0.4483200583507267,1.0,0.29020818122405423,0.9471210340775558,0.9396641236264643,0.9471210340775558,0.9468014200200555,0.9448071420518477,0.9357183120270788,-1.0,0.12073215842247009
+89,1.7326166082434382e-05,0.44835791004993825,1.0,0.2886762074101826,0.9471210340775558,0.9396641236264643,0.9471210340775558,0.9468014200200555,0.9448071420518477,0.9357183120270788,-1.0,0.12033922225236893
+90,1.4518598195691546e-05,0.4486914487664795,0.9998358233459202,0.2872285750091566,0.9471210340775558,0.9396641236264643,0.9471210340775558,0.9468014200200555,0.9448071420518477,0.9357183120270788,-1.0,0.12013569474220276
+91,1.196907250717768e-05,0.4484253770963368,1.0,0.2857050651599041,0.9471210340775558,0.9396641236264643,0.9471210340775558,0.9468014200200555,0.9448071420518477,0.9357183120270788,-1.0,0.11980102211236954
+92,9.68037688580146e-06,0.4499674249196401,0.9995074700377606,0.2842704875768982,0.9459459459459459,0.9384797966290146,0.9459459459459459,0.9456742849225576,0.9432515073782316,0.9349742644080313,-1.0,0.11958242952823639
+93,7.655013986622496e-06,0.44840513204749316,0.9998358233459202,0.2826862976427784,0.9494712103407755,0.9431311364160085,0.9494712103407755,0.9492345406176157,0.948520942358427,0.9388242690500866,-1.0,0.11919249594211578
+94,5.895198514235033e-06,0.4482504137385128,0.9998358233459202,0.28139602500599503,0.9471210340775558,0.9402224789722136,0.9471210340775558,0.9468815638109709,0.9452193283115697,0.9365441231207472,-1.0,0.11891059577465057
+95,4.402854801020759e-06,0.4482248198015313,1.0,0.28011780129756547,0.9471210340775558,0.9402224789722136,0.9471210340775558,0.9468815638109709,0.9452193283115697,0.9365441231207472,-1.0,0.11867201328277588
+96,3.1796147029196916e-06,0.4481892012704321,1.0,0.278920335123878,0.9471210340775558,0.9402224789722136,0.9471210340775558,0.9468815638109709,0.9452193283115697,0.9365441231207472,-1.0,0.11844654381275177
+97,2.226815815019654e-06,0.44868434781884153,0.9998358233459202,0.2775330954657038,0.9471210340775558,0.9402224789722136,0.9471210340775558,0.9468815638109709,0.9452193283115697,0.9365441231207472,-1.0,0.11813247203826904
+98,1.5455000089157545e-06,0.4481991411057859,1.0,0.2764225577285511,0.9471210340775558,0.9402224789722136,0.9471210340775558,0.9468815638109709,0.9452193283115697,0.9365441231207472,-1.0,0.11787455528974533
+99,1.1364122934389507e-06,0.4487777463694234,0.9998358233459202,0.2753074277722037,0.9471210340775558,0.9402224789722136,0.9471210340775558,0.9468815638109709,0.9452193283115697,0.9365441231207472,-1.0,0.11759176105260849
+100,1e-06,0.448271934841482,1.0,0.2740601867472383,0.9482961222091657,0.9418746747005986,0.9482961222091657,0.9480345684268087,0.9469448377924705,0.9383524413847616,-1.0,0.11731333285570145

metrics/classification_report.txt

@@ -0,0 +1,13 @@
+                     precision    recall  f1-score   support
+
+       Brown Blight     0.9271    0.9570    0.9418        93
+        Gray Blight     0.9477    0.9760    0.9617       167
+    Green mirid bug     0.9786    0.9482    0.9632       193
+       Healthy leaf     0.9934    1.0000    0.9967       151
+         Helopeltis     0.9457    0.9560    0.9508        91
+         Red spider     0.9605    0.9481    0.9542        77
+Tea algal leaf spot     0.9351    0.9000    0.9172        80
+
+           accuracy                         0.9601       852
+          macro avg     0.9554    0.9550    0.9551       852
+       weighted avg     0.9603    0.9601    0.9600       852

metrics/confusion_matrix.npy

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1c8a56d38515db25c027fb0284eb85c14e08c354e0d49977fc4df81ea050a677
+size 520

metrics/per_class_metrics.json

@@ -0,0 +1,37 @@
+{
+  "Brown Blight": {
+    "f1": 0.9417989417989417,
+    "precision": 0.9270833333333334,
+    "recall": 0.956989247311828
+  },
+  "Gray Blight": {
+    "f1": 0.9616519174041298,
+    "precision": 0.9476744186046512,
+    "recall": 0.9760479041916168
+  },
+  "Green mirid bug": {
+    "f1": 0.9631578947368421,
+    "precision": 0.9786096256684492,
+    "recall": 0.9481865284974094
+  },
+  "Healthy leaf": {
+    "f1": 0.9966996699669967,
+    "precision": 0.993421052631579,
+    "recall": 1.0
+  },
+  "Helopeltis": {
+    "f1": 0.9508196721311475,
+    "precision": 0.9456521739130435,
+    "recall": 0.9560439560439561
+  },
+  "Red spider": {
+    "f1": 0.954248366013072,
+    "precision": 0.9605263157894737,
+    "recall": 0.948051948051948
+  },
+  "Tea algal leaf spot": {
+    "f1": 0.9171974522292994,
+    "precision": 0.935064935064935,
+    "recall": 0.9
+  }
+}

metrics/test_results.json

@@ -0,0 +1,16 @@
+{
+  "loss": 0.2719278124296609,
+  "acc1": 0.960093896713615,
+  "macro_f1": 0.9550819877543469,
+  "micro_f1": 0.960093896713615,
+  "weighted_f1": 0.9600373311834545,
+  "macro_precision": 0.9554331221436377,
+  "macro_recall": 0.9550456548709655,
+  "macro_auc": 0.9959145616219398,
+  "gate_global_mean": 0.10522443801164627,
+  "used_ema_weights": true,
+  "inference_seconds": 2.2969756059974316,
+  "throughput_img_s": 92.29527707933222,
+  "params_m": 49.47357,
+  "gflops": 17.164864688
+}

model/best_model.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d9de14c8f877be3648bc56f3a4224b40feb4538f3810240adf4812ade58574af
+size 792151199

model/ema_model_weights_only.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f87db62c0dc3619566e4418961e56c14012fb61ed21735b5021898eecb3d0a92
+size 198011974

model/model_weights_only.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a26c478f3f0bc54feeac54483013dd6ac1f74b89be9fef2d31e40461aeeb51fb
+size 198037482

raw_outputs/gate_stats_test.json

@@ -0,0 +1,37 @@
+{
+  "Brown Blight": {
+    "mean": 0.07836990058422089,
+    "std": 0.22063851356506348,
+    "n": 93
+  },
+  "Gray Blight": {
+    "mean": 0.015741290524601936,
+    "std": 0.07828688621520996,
+    "n": 167
+  },
+  "Green mirid bug": {
+    "mean": 0.03033372201025486,
+    "std": 0.07218299806118011,
+    "n": 193
+  },
+  "Healthy leaf": {
+    "mean": 0.004847533069550991,
+    "std": 0.003229882102459669,
+    "n": 151
+  },
+  "Helopeltis": {
+    "mean": 0.003911720588803291,
+    "std": 0.001458005397580564,
+    "n": 91
+  },
+  "Red spider": {
+    "mean": 0.0035547777079045773,
+    "std": 0.0016042763600125909,
+    "n": 77
+  },
+  "Tea algal leaf spot": {
+    "mean": 0.9064744114875793,
+    "std": 0.28227925300598145,
+    "n": 80
+  }
+}

raw_outputs/test_predictions.npy

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:20a44cfb6a4f8ded1badb4c48d243bed1e7955989a1ede71b6cd486584b408c1
+size 6944

raw_outputs/test_probabilities.npy

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:95d9c685145349b45b22e02c7792c9dfe0627cb8f53282bd2d450e43fe1d208c
+size 47840

raw_outputs/test_targets.npy

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6e111d57ac295dfea569b2887140e9a61e07d016cde1f9ad051ef9c725d2f54d
+size 6944