polyedit-mips-retrained / training_report.json
luca0621's picture
Add retrained PolyEdit checkpoint and audit artifacts
27c81e3 verified
Raw
History Blame Contribute Delete
8.67 kB
{
"model_label": "MIPS-retrained (public architecture, no released pretraining weights)",
"split_audit": {
"seed": 42,
"validation_fraction": 0.15,
"properties": [
"Egc",
"Egb",
"Eea",
"Ei",
"EPS",
"Nc",
"Xc",
"Eat"
],
"records": {
"train": 1300,
"valid": 75,
"test": 953
},
"observed_labels": {
"train": {
"Egc": 1260,
"Egb": 245,
"Eea": 144,
"Ei": 145,
"EPS": 148,
"Nc": 148,
"Xc": 67,
"Eat": 150
},
"valid": {
"Egc": 73,
"Egb": 17,
"Eea": 14,
"Ei": 14,
"EPS": 15,
"Nc": 15,
"Xc": 6,
"Eat": 15
},
"test": {
"Egc": 917,
"Egb": 191,
"Eea": 156,
"Ei": 156,
"EPS": 158,
"Nc": 158,
"Xc": 105,
"Eat": 160
}
},
"components": {
"train": 140,
"valid": 22,
"test": 12
},
"component_overlap": 0,
"canonical_label_conflicts": {
"Egc": 2,
"Egb": 0,
"Eea": 0,
"Ei": 0,
"EPS": 0,
"Nc": 0,
"Xc": 0,
"Eat": 0
},
"featurization_failures": {
"train": 0,
"valid": 0,
"test": 0
}
},
"best_epoch": 15,
"best_valid_balanced_mse": 0.2277407869696617,
"history": [
{
"epoch": 1,
"train_balanced_mse": 0.823832509063539,
"valid_balanced_mse": 0.695390373468399
},
{
"epoch": 2,
"train_balanced_mse": 0.5804621186994371,
"valid_balanced_mse": 0.7774837017059326
},
{
"epoch": 3,
"train_balanced_mse": 0.44400371540160405,
"valid_balanced_mse": 0.6737748980522156
},
{
"epoch": 4,
"train_balanced_mse": 0.4038538159359069,
"valid_balanced_mse": 0.6498120427131653
},
{
"epoch": 5,
"train_balanced_mse": 0.3874354188640912,
"valid_balanced_mse": 0.47673121094703674
},
{
"epoch": 6,
"train_balanced_mse": 0.3302447313354129,
"valid_balanced_mse": 0.4952659010887146
},
{
"epoch": 7,
"train_balanced_mse": 0.3341101039023626,
"valid_balanced_mse": 0.3928861692547798
},
{
"epoch": 8,
"train_balanced_mse": 0.2897623571611586,
"valid_balanced_mse": 0.25733907520771027
},
{
"epoch": 9,
"train_balanced_mse": 0.2766442291793369,
"valid_balanced_mse": 0.33025388419628143
},
{
"epoch": 10,
"train_balanced_mse": 0.26466881412835347,
"valid_balanced_mse": 0.2535265013575554
},
{
"epoch": 11,
"train_balanced_mse": 0.22542981732459294,
"valid_balanced_mse": 0.28802742809057236
},
{
"epoch": 12,
"train_balanced_mse": 0.19928185712723506,
"valid_balanced_mse": 0.27805954962968826
},
{
"epoch": 13,
"train_balanced_mse": 0.22888986482506707,
"valid_balanced_mse": 0.3971787840127945
},
{
"epoch": 14,
"train_balanced_mse": 0.19854760063546045,
"valid_balanced_mse": 0.3791179656982422
},
{
"epoch": 15,
"train_balanced_mse": 0.2075319520774342,
"valid_balanced_mse": 0.2277407869696617
},
{
"epoch": 16,
"train_balanced_mse": 0.19584149086759203,
"valid_balanced_mse": 0.2900788113474846
},
{
"epoch": 17,
"train_balanced_mse": 0.19131148429143996,
"valid_balanced_mse": 0.2720031626522541
},
{
"epoch": 18,
"train_balanced_mse": 0.15825964386264482,
"valid_balanced_mse": 0.3209325149655342
},
{
"epoch": 19,
"train_balanced_mse": 0.21145633962892352,
"valid_balanced_mse": 0.3759225383400917
},
{
"epoch": 20,
"train_balanced_mse": 0.19808851395334517,
"valid_balanced_mse": 0.28116101399064064
},
{
"epoch": 21,
"train_balanced_mse": 0.19104890951088496,
"valid_balanced_mse": 0.3503660410642624
},
{
"epoch": 22,
"train_balanced_mse": 0.1721404262241863,
"valid_balanced_mse": 0.34436123818159103
},
{
"epoch": 23,
"train_balanced_mse": 0.16209240364176886,
"valid_balanced_mse": 0.33224186301231384
},
{
"epoch": 24,
"train_balanced_mse": 0.15665275497095926,
"valid_balanced_mse": 0.3485347107052803
},
{
"epoch": 25,
"train_balanced_mse": 0.14901041381415867,
"valid_balanced_mse": 0.32935764640569687
},
{
"epoch": 26,
"train_balanced_mse": 0.15354844750393004,
"valid_balanced_mse": 0.2708689011633396
},
{
"epoch": 27,
"train_balanced_mse": 0.14133372831912266,
"valid_balanced_mse": 0.2737532742321491
},
{
"epoch": 28,
"train_balanced_mse": 0.1276883092664537,
"valid_balanced_mse": 0.2544044181704521
},
{
"epoch": 29,
"train_balanced_mse": 0.12970404504310518,
"valid_balanced_mse": 0.24341677874326706
},
{
"epoch": 30,
"train_balanced_mse": 0.13190407980056035,
"valid_balanced_mse": 0.26462970674037933
},
{
"epoch": 31,
"train_balanced_mse": 0.1177525193918319,
"valid_balanced_mse": 0.2867523841559887
},
{
"epoch": 32,
"train_balanced_mse": 0.10043083361926533,
"valid_balanced_mse": 0.3219228461384773
},
{
"epoch": 33,
"train_balanced_mse": 0.10154936427161806,
"valid_balanced_mse": 0.23854781314730644
},
{
"epoch": 34,
"train_balanced_mse": 0.10463355410666693,
"valid_balanced_mse": 0.28424525260925293
},
{
"epoch": 35,
"train_balanced_mse": 0.12242750433229264,
"valid_balanced_mse": 0.2922641634941101
},
{
"epoch": 36,
"train_balanced_mse": 0.10169008303256262,
"valid_balanced_mse": 0.2771267108619213
},
{
"epoch": 37,
"train_balanced_mse": 0.09358360582873934,
"valid_balanced_mse": 0.2986114788800478
},
{
"epoch": 38,
"train_balanced_mse": 0.09119904910524686,
"valid_balanced_mse": 0.24989525228738785
},
{
"epoch": 39,
"train_balanced_mse": 0.10186363366388139,
"valid_balanced_mse": 0.2841300815343857
},
{
"epoch": 40,
"train_balanced_mse": 0.09794175979636964,
"valid_balanced_mse": 0.2678869664669037
},
{
"epoch": 41,
"train_balanced_mse": 0.07286352611013822,
"valid_balanced_mse": 0.2866135314106941
},
{
"epoch": 42,
"train_balanced_mse": 0.07449667670187496,
"valid_balanced_mse": 0.28048381581902504
},
{
"epoch": 43,
"train_balanced_mse": 0.07956392362359024,
"valid_balanced_mse": 0.29140742868185043
},
{
"epoch": 44,
"train_balanced_mse": 0.07215168061000961,
"valid_balanced_mse": 0.27722878381609917
},
{
"epoch": 45,
"train_balanced_mse": 0.08117080763691947,
"valid_balanced_mse": 0.26927550137043
}
],
"heldout_property_metrics": {
"Egc": {
"examples": 917,
"mae": 0.521049514137301,
"rmse": 0.6358568894994391,
"within_target_tolerance": 0.04689203925845147,
"tolerance": 0.05
},
"Egb": {
"examples": 191,
"mae": 0.707791957135064,
"rmse": 0.9666989891085025,
"within_target_tolerance": 0.04712041884816754,
"tolerance": 0.05
},
"Eea": {
"examples": 156,
"mae": 0.4107183844479394,
"rmse": 0.5128703700667199,
"within_target_tolerance": 0.07051282051282051,
"tolerance": 0.05
},
"Ei": {
"examples": 156,
"mae": 0.5060547432904271,
"rmse": 0.6440749112329941,
"within_target_tolerance": 0.019230769230769232,
"tolerance": 0.05
},
"EPS": {
"examples": 158,
"mae": 0.4330520880036501,
"rmse": 0.612281045854852,
"within_target_tolerance": 0.10126582278481013,
"tolerance": 0.05
},
"Nc": {
"examples": 158,
"mae": 0.07196369473652443,
"rmse": 0.10136330483734457,
"within_target_tolerance": 0.27848101265822783,
"tolerance": 0.025
},
"Xc": {
"examples": 105,
"mae": 20.83052285156752,
"rmse": 25.61243106865855,
"within_target_tolerance": 0.0380952380952381,
"tolerance": 1.0
},
"Eat": {
"examples": 160,
"mae": 0.11537404623516348,
"rmse": 0.1356859144409758,
"within_target_tolerance": 0.1625,
"tolerance": 0.05
}
}
}