yield-estimation / config.json
Sarikaa-Sridhar's picture
Update config.json
4d3c5e2 verified
Raw
History Blame Contribute Delete
6.22 kB
{
"K": 52,
"S": 66,
"W": 6,
"architectures": [
"YieldForSequenceClassification"
],
"crop_emb_dim": 8,
"d_model": 64,
"dim_ff": 128,
"dropout": 0.4,
"dtype": "float32",
"eval_cutoffs": [
20,
24,
28,
32,
36,
40,
44,
48,
52
],
"model_type": "yield-weather-soil",
"nhead": 4,
"num_layers": 4,
"pool": "last",
"s_mean": [
138.40794372558594,
152.29786682128906,
154.56175231933594,
160.80038452148438,
164.22401428222656,
165.8285675048828,
232.1925506591797,
211.66586303710938,
208.61048889160156,
210.4038543701172,
196.25576782226562,
169.04379272460938,
16.51355743408203,
16.052227020263672,
16.29838752746582,
18.991479873657227,
26.101285934448242,
35.37116241455078,
293.3885192871094,
296.4434814453125,
315.0303039550781,
331.96435546875,
329.5356750488281,
309.2357177734375,
3859.735107421875,
2151.462158203125,
1574.3883056640625,
1057.6832275390625,
792.9168090820312,
640.056884765625,
401.9707946777344,
282.0995178222656,
218.99769592285156,
119.24910736083984,
71.83614349365234,
44.02671432495117,
61.37840270996094,
61.37840270996094,
61.37840270996094,
61.37840270996094,
61.37840270996094,
61.37840270996094,
64.16217041015625,
64.32339477539062,
64.31382751464844,
65.56702423095703,
68.18431854248047,
71.56515502929688,
197.78048706054688,
196.43638610839844,
191.7803955078125,
190.5934295654297,
205.2471466064453,
238.66366577148438,
508.832275390625,
507.11798095703125,
493.1898193359375,
477.44244384765625,
465.2177429199219,
452.10076904296875,
347.4835510253906,
207.85842895507812,
140.189453125,
88.9273681640625,
50.258174896240234,
35.850830078125
],
"s_std": [
5.658433437347412,
4.963406562805176,
4.842231273651123,
6.4795145988464355,
7.407203674316406,
8.543693542480469,
32.758174896240234,
43.4786262512207,
41.363216400146484,
36.315101623535156,
37.40601348876953,
44.32140350341797,
14.699563980102539,
14.964685440063477,
15.500894546508789,
18.559709548950195,
22.95189666748047,
28.39887046813965,
55.16016387939453,
54.736392974853516,
54.40840148925781,
57.90495681762695,
59.78011703491211,
68.35675048828125,
1374.9456787109375,
633.04736328125,
509.03704833984375,
386.8252258300781,
327.8870849609375,
311.1480407714844,
57.87489318847656,
59.43294906616211,
50.55767059326172,
27.027706146240234,
13.878527641296387,
10.90867805480957,
14.719537734985352,
14.719537734985352,
14.719537734985352,
14.719537734985352,
14.719537734985352,
14.719537734985352,
3.2984049320220947,
3.6471333503723145,
4.528029441833496,
6.537323951721191,
7.542074203491211,
6.777342796325684,
131.77452087402344,
133.4365692138672,
134.30154418945312,
137.3269500732422,
143.42730712890625,
147.23556518554688,
106.86735534667969,
106.61359405517578,
99.10890197753906,
93.6629638671875,
95.26378631591797,
95.61508178710938,
88.6065444946289,
62.81496047973633,
45.79389190673828,
31.114017486572266,
18.991355895996094,
19.752763748168945
],
"soil_vars": [
"bdod_mean_0-5cm",
"bdod_mean_5-15cm",
"bdod_mean_15-30cm",
"bdod_mean_30-60cm",
"bdod_mean_60-100cm",
"bdod_mean_100-200cm",
"cec_mean_0-5cm",
"cec_mean_5-15cm",
"cec_mean_15-30cm",
"cec_mean_30-60cm",
"cec_mean_60-100cm",
"cec_mean_100-200cm",
"cfvo_mean_0-5cm",
"cfvo_mean_5-15cm",
"cfvo_mean_15-30cm",
"cfvo_mean_30-60cm",
"cfvo_mean_60-100cm",
"cfvo_mean_100-200cm",
"clay_mean_0-5cm",
"clay_mean_5-15cm",
"clay_mean_15-30cm",
"clay_mean_30-60cm",
"clay_mean_60-100cm",
"clay_mean_100-200cm",
"nitrogen_mean_0-5cm",
"nitrogen_mean_5-15cm",
"nitrogen_mean_15-30cm",
"nitrogen_mean_30-60cm",
"nitrogen_mean_60-100cm",
"nitrogen_mean_100-200cm",
"ocd_mean_0-5cm",
"ocd_mean_5-15cm",
"ocd_mean_15-30cm",
"ocd_mean_30-60cm",
"ocd_mean_60-100cm",
"ocd_mean_100-200cm",
"ocs_mean_0-5cm",
"ocs_mean_5-15cm",
"ocs_mean_15-30cm",
"ocs_mean_30-60cm",
"ocs_mean_60-100cm",
"ocs_mean_100-200cm",
"phh2o_mean_0-5cm",
"phh2o_mean_5-15cm",
"phh2o_mean_15-30cm",
"phh2o_mean_30-60cm",
"phh2o_mean_60-100cm",
"phh2o_mean_100-200cm",
"sand_mean_0-5cm",
"sand_mean_5-15cm",
"sand_mean_15-30cm",
"sand_mean_30-60cm",
"sand_mean_60-100cm",
"sand_mean_100-200cm",
"silt_mean_0-5cm",
"silt_mean_5-15cm",
"silt_mean_15-30cm",
"silt_mean_30-60cm",
"silt_mean_60-100cm",
"silt_mean_100-200cm",
"soc_mean_0-5cm",
"soc_mean_5-15cm",
"soc_mean_15-30cm",
"soc_mean_30-60cm",
"soc_mean_60-100cm",
"soc_mean_100-200cm"
],
"train_cutoffs": [
20,
24,
28,
32,
36,
40,
44,
48,
52
],
"transformers_version": "5.13.0",
"use_crop": true,
"w_mean": [
2.6932172775268555,
319.30352783203125,
5.36029577255249,
15.82466983795166,
3.981907606124878,
1018.5076904296875
],
"w_std": [
3.367875814437866,
91.14374542236328,
13.447235107421875,
11.749135971069336,
11.175808906555176,
684.6181030273438
],
"weather_vars": [
"prcp",
"srad",
"swe",
"tmax",
"tmin",
"vp"
],
"y_mean": 166.3075714111328,
"y_std": 30.291048049926758,
"problem_type": "regression",
"num_labels": 1,
"id2label": {
"0": "YIELD_BU_ACRE"
},
"label2id": {
"YIELD_BU_ACRE": 0
},
"tokenizer_class": "YieldTokenizer",
"auto_map": {
"AutoConfig": "configuration_yield.YieldConfig",
"AutoModelForSequenceClassification": "modeling_yield.YieldForSequenceClassification",
"AutoTokenizer": [
"tokenization_yield.YieldTokenizer",
null
]
}
}