recommit cross validation training

A9/all_models_config.py

@@ -0,0 +1,75 @@
+from models import (
+    N_INPUT, N_OUTPUT, N_JOINTS, JOINTS,
+    load_all_sequences, flatten_sequences, make_windowed_sequences,
+    build_dense_model, build_conv1d_model, build_lstm_model, build_gru_model
+)
+
+ALL_MODEL_CONFIGS = {
+    # 'dense': {
+    #     'build_fn': build_dense_model,
+    #     'params': {
+    #         'hidden_units': (256, 128, 64),
+    #         'activation': 'relu',
+    #         'dropout_rate': 0.3,
+    #         'l2_reg': 1e-4,
+    #     },
+    #     'data_type': 'flat',
+    # },
+    'conv1d': {
+        'build_fn': build_conv1d_model,
+        'params': {
+            'filters': (64, 128),
+            'kernel_size': 3,
+            'pool_size': 2,
+            'dense_units': (64,),
+            'activation': 'relu',
+            'dropout_rate': 0.3,
+        },
+        'data_type': 'windowed',
+    },
+    'conv1d_v2': {
+        'build_fn': build_conv1d_model,
+        'params': {
+            'filters': (32, 64, 128),
+            'kernel_size': 5,
+            'pool_size': 2,
+            'dense_units': (128, 64),
+            'activation': 'relu',
+            'dropout_rate': 0.4,
+        },
+        'data_type': 'windowed',
+    },
+    'conv1d_v3': {
+        'build_fn': build_conv1d_model,
+        'params': {
+            'filters': (128, 256),
+            'kernel_size': 3,
+            'pool_size': 3,
+            'dense_units': (256, 128, 64),
+            'activation': 'relu',
+            'dropout_rate': 0.2,
+        },
+        'data_type': 'windowed',
+    },
+    # 'lstm': {
+    #     'build_fn': build_lstm_model,
+    #     'params': {
+    #         'lstm_units': (64, 32),
+    #         'dense_units': (32,),
+    #         'activation': 'tanh',
+    #         'dropout_rate': 0.3,
+    #         'recurrent_dropout': 0.1,
+    #     },
+    #     'data_type': 'windowed',
+    # },
+    # 'gru': {
+    #     'build_fn': build_gru_model,
+    #     'params': {
+    #         'gru_units': (64, 32),
+    #         'dense_units': (32,),
+    #         'dropout_rate': 0.3,
+    #         'recurrent_dropout': 0.1,
+    #     },
+    #     'data_type': 'windowed',
+    # },
+}

A9/all_optimizery.py

@@ -0,0 +1,35 @@
+from tensorflow import keras
+
+OPTIMIZER_CONFIGS = {
+    'sgd': {
+        'name': 'SGD',
+        'optimizer_fn': lambda lr: keras.optimizers.SGD(
+            learning_rate=lr,
+            momentum=0.9,
+            nesterov=True
+        ),
+        'default_lr': 0.01,
+        'description': 'Stochastic Gradient Descent with momentum (0.9) and Nesterov acceleration'
+    },
+    'rmsprop': {
+        'name': 'RMSprop',
+        'optimizer_fn': lambda lr: keras.optimizers.RMSprop(
+            learning_rate=lr,
+            rho=0.9,
+            epsilon=1e-7
+        ),
+        'default_lr': 0.001,
+        'description': 'Root Mean Square propagation with rho=0.9'
+    },
+    'adam': {
+        'name': 'Adam',
+        'optimizer_fn': lambda lr: keras.optimizers.Adam(
+            learning_rate=lr,
+            beta_1=0.9,
+            beta_2=0.999,
+            epsilon=1e-7
+        ),
+        'default_lr': 0.001,
+        'description': 'Adaptive Moment Estimation with default beta parameters'
+    },
+}

A9/cv_results_conv1D_variants/conv1d_cv_results.json

@@ -0,0 +1,1243 @@
+{
+  "model_name": "conv1d",
+  "n_splits": 5,
+  "epochs": 100,
+  "batch_size": 32,
+  "patience": 10,
+  "aggregate_results": {
+    "n_folds": 5,
+    "train_metrics": {
+      "train_mse": {
+        "mean": 0.0013674775836989284,
+        "std": 0.00010447232296381574
+      },
+      "train_rmse": {
+        "mean": 0.036953366365008275,
+        "std": 0.0013879113777194468
+      },
+      "train_mae": {
+        "mean": 0.027265655994415283,
+        "std": 0.0008485562269601597
+      },
+      "train_r2": {
+        "mean": 0.9352981090545655,
+        "std": 0.005146068609952537
+      },
+      "train_head_mse": {
+        "mean": 0.001326314453035593,
+        "std": 0.00010827707955034441
+      },
+      "train_head_rmse": {
+        "mean": 0.036388519255212776,
+        "std": 0.0014799052160873556
+      },
+      "train_head_mae": {
+        "mean": 0.02745846062898636,
+        "std": 0.0010056064368367265
+      },
+      "train_head_r2": {
+        "mean": 0.8666204452514649,
+        "std": 0.01171166456874094
+      },
+      "train_left_shoulder_mse": {
+        "mean": 0.0008329122909344732,
+        "std": 6.056707645602337e-05
+      },
+      "train_left_shoulder_rmse": {
+        "mean": 0.028841305473485374,
+        "std": 0.001044695898132998
+      },
+      "train_left_shoulder_mae": {
+        "mean": 0.02195923924446106,
+        "std": 0.00076727230426843
+      },
+      "train_left_shoulder_r2": {
+        "mean": 0.8501842737197876,
+        "std": 0.010668693025846542
+      },
+      "train_left_elbow_mse": {
+        "mean": 0.0013519973494112492,
+        "std": 0.00011219314193855145
+      },
+      "train_left_elbow_rmse": {
+        "mean": 0.03673886672639777,
+        "std": 0.0015010066859385007
+      },
+      "train_left_elbow_mae": {
+        "mean": 0.028505989164114,
+        "std": 0.0012124902025995418
+      },
+      "train_left_elbow_r2": {
+        "mean": 0.9021070837974549,
+        "std": 0.007715037041884288
+      },
+      "train_right_shoulder_mse": {
+        "mean": 0.0008276826934888959,
+        "std": 9.674879556254987e-05
+      },
+      "train_right_shoulder_rmse": {
+        "mean": 0.0287229085262231,
+        "std": 0.0016362210373651093
+      },
+      "train_right_shoulder_mae": {
+        "mean": 0.021864095702767372,
+        "std": 0.0011259970885969966
+      },
+      "train_right_shoulder_r2": {
+        "mean": 0.8730275392532348,
+        "std": 0.01580822450524276
+      },
+      "train_right_elbow_mse": {
+        "mean": 0.0013868886977434157,
+        "std": 0.00015041280966255474
+      },
+      "train_right_elbow_rmse": {
+        "mean": 0.037188949903753925,
+        "std": 0.0019674101757123992
+      },
+      "train_right_elbow_mae": {
+        "mean": 0.028557825461030008,
+        "std": 0.0012342444869196845
+      },
+      "train_right_elbow_r2": {
+        "mean": 0.9120388269424439,
+        "std": 0.009928038649956726
+      },
+      "train_left_hand_mse": {
+        "mean": 0.002239828696474433,
+        "std": 0.00021475067723644066
+      },
+      "train_left_hand_rmse": {
+        "mean": 0.047274550480854424,
+        "std": 0.0022238644984753793
+      },
+      "train_left_hand_mae": {
+        "mean": 0.03682841882109642,
+        "std": 0.0015976905996233789
+      },
+      "train_left_hand_r2": {
+        "mean": 0.9148225903511047,
+        "std": 0.0078140631862792
+      },
+      "train_right_hand_mse": {
+        "mean": 0.002316287951543927,
+        "std": 0.00026727040652174867
+      },
+      "train_right_hand_rmse": {
+        "mean": 0.048050892754787,
+        "std": 0.0027202310585473168
+      },
+      "train_right_hand_mae": {
+        "mean": 0.03665531724691391,
+        "std": 0.0016860183397093065
+      },
+      "train_right_hand_r2": {
+        "mean": 0.9168192863464355,
+        "std": 0.009716877044885027
+      },
+      "train_left_hip_mse": {
+        "mean": 0.0011226138565689325,
+        "std": 4.133683613808709e-05
+      },
+      "train_left_hip_rmse": {
+        "mean": 0.033499784465900226,
+        "std": 0.0006150587835016061
+      },
+      "train_left_hip_mae": {
+        "mean": 0.025444645062088968,
+        "std": 0.0004618677940483075
+      },
+      "train_left_hip_r2": {
+        "mean": 0.8559396982192993,
+        "std": 0.006698427240212163
+      },
+      "train_right_hip_mse": {
+        "mean": 0.0011105953482910991,
+        "std": 4.231008955827923e-05
+      },
+      "train_right_hip_rmse": {
+        "mean": 0.0333195215548874,
+        "std": 0.0006362638167404992
+      },
+      "train_right_hip_mae": {
+        "mean": 0.025051160901784896,
+        "std": 0.000479124496781922
+      },
+      "train_right_hip_r2": {
+        "mean": 0.8630171537399292,
+        "std": 0.007315151454351005
+      },
+      "train_left_knee_mse": {
+        "mean": 0.0011312809539958835,
+        "std": 8.171554908713267e-05
+      },
+      "train_left_knee_rmse": {
+        "mean": 0.03361294276586063,
+        "std": 0.0012045881515524144
+      },
+      "train_left_knee_mae": {
+        "mean": 0.02520242817699909,
+        "std": 0.0008166047094808891
+      },
+      "train_left_knee_r2": {
+        "mean": 0.8806431770324707,
+        "std": 0.009944797100160612
+      },
+      "train_right_knee_mse": {
+        "mean": 0.0011635047616437078,
+        "std": 9.244940639927664e-05
+      },
+      "train_right_knee_rmse": {
+        "mean": 0.03408356264695508,
+        "std": 0.0013474119395330865
+      },
+      "train_right_knee_mae": {
+        "mean": 0.025705544650554656,
+        "std": 0.001032434348173573
+      },
+      "train_right_knee_r2": {
+        "mean": 0.8918254852294922,
+        "std": 0.008600290212284832
+      },
+      "train_left_foot_mse": {
+        "mean": 0.0014805194921791553,
+        "std": 0.00014788250224631065
+      },
+      "train_left_foot_rmse": {
+        "mean": 0.038429950578845816,
+        "std": 0.001912691999937151
+      },
+      "train_left_foot_mae": {
+        "mean": 0.025140823796391487,
+        "std": 0.0006705197794394937
+      },
+      "train_left_foot_r2": {
+        "mean": 0.6612395167350769,
+        "std": 0.034462092569137244
+      },
+      "train_right_foot_mse": {
+        "mean": 0.0014867823105305434,
+        "std": 0.00011272420003709216
+      },
+      "train_right_foot_rmse": {
+        "mean": 0.038530801416684385,
+        "std": 0.0014695756933808847
+      },
+      "train_right_foot_mae": {
+        "mean": 0.0260795745998621,
+        "std": 0.0007658809940115817
+      },
+      "train_right_foot_r2": {
+        "mean": 0.6683424949645996,
+        "std": 0.025697304933476956
+      }
+    },
+    "val_metrics": {
+      "val_mse": {
+        "mean": 0.00163096459582448,
+        "std": 0.000192285467393062
+      },
+      "val_rmse": {
+        "mean": 0.04031585464939032,
+        "std": 0.0023656837729742627
+      },
+      "val_mae": {
+        "mean": 0.02908308058977127,
+        "std": 0.0015594302764251832
+      },
+      "val_r2": {
+        "mean": 0.9229234099388123,
+        "std": 0.007842932831262816
+      },
+      "val_head_mse": {
+        "mean": 0.0015287821181118489,
+        "std": 0.00022410896487630975
+      },
+      "val_head_rmse": {
+        "mean": 0.039001549317438516,
+        "std": 0.0027678997364897575
+      },
+      "val_head_mae": {
+        "mean": 0.029091638326644898,
+        "std": 0.0017916138906603218
+      },
+      "val_head_r2": {
+        "mean": 0.8467427015304565,
+        "std": 0.01605789625016368
+      },
+      "val_left_shoulder_mse": {
+        "mean": 0.0009633486159145832,
+        "std": 0.00012496241041069073
+      },
+      "val_left_shoulder_rmse": {
+        "mean": 0.03097261330260233,
+        "std": 0.0020114276328131405
+      },
+      "val_left_shoulder_mae": {
+        "mean": 0.02330469712615013,
+        "std": 0.0013370952187728065
+      },
+      "val_left_shoulder_r2": {
+        "mean": 0.8269778251647949,
+        "std": 0.016900077063892457
+      },
+      "val_left_elbow_mse": {
+        "mean": 0.0015759425004944205,
+        "std": 0.00019954299353251644
+      },
+      "val_left_elbow_rmse": {
+        "mean": 0.039616519961909014,
+        "std": 0.0025443754837062528
+      },
+      "val_left_elbow_mae": {
+        "mean": 0.030469732731580733,
+        "std": 0.0015930399152496563
+      },
+      "val_left_elbow_r2": {
+        "mean": 0.8856167793273926,
+        "std": 0.015235446893660343
+      },
+      "val_right_shoulder_mse": {
+        "mean": 0.0009883235092274846,
+        "std": 0.00020681584342622494
+      },
+      "val_right_shoulder_rmse": {
+        "mean": 0.03128451612329376,
+        "std": 0.0030987997932840406
+      },
+      "val_right_shoulder_mae": {
+        "mean": 0.023347513750195503,
+        "std": 0.001963987988768579
+      },
+      "val_right_shoulder_r2": {
+        "mean": 0.8491879820823669,
+        "std": 0.025334490341769184
+      },
+      "val_right_elbow_mse": {
+        "mean": 0.0017225256422534585,
+        "std": 0.0003106508896136305
+      },
+      "val_right_elbow_rmse": {
+        "mean": 0.04134244018287825,
+        "std": 0.0036507919659427655
+      },
+      "val_right_elbow_mae": {
+        "mean": 0.030990320816636087,
+        "std": 0.0020072850039217225
+      },
+      "val_right_elbow_r2": {
+        "mean": 0.8909905672073364,
+        "std": 0.017207211053953238
+      },
+      "val_left_hand_mse": {
+        "mean": 0.002681449754163623,
+        "std": 0.00034906254614534143
+      },
+      "val_left_hand_rmse": {
+        "mean": 0.05167535469652275,
+        "std": 0.0033327872947710433
+      },
+      "val_left_hand_mae": {
+        "mean": 0.039669263362884524,
+        "std": 0.0022678190469317084
+      },
+      "val_left_hand_r2": {
+        "mean": 0.8975192785263062,
+        "std": 0.015380536754565594
+      },
+      "val_right_hand_mse": {
+        "mean": 0.0028814102057367562,
+        "std": 0.00046343489645050555
+      },
+      "val_right_hand_rmse": {
+        "mean": 0.05350924991995184,
+        "std": 0.004262672722704691
+      },
+      "val_right_hand_mae": {
+        "mean": 0.03993181809782982,
+        "std": 0.00216742171090448
+      },
+      "val_right_hand_r2": {
+        "mean": 0.8965159773826599,
+        "std": 0.015386297615275744
+      },
+      "val_left_hip_mse": {
+        "mean": 0.0012839170405641199,
+        "std": 0.0001460708284439856
+      },
+      "val_left_hip_rmse": {
+        "mean": 0.0357723073284611,
+        "std": 0.0020637511870979005
+      },
+      "val_left_hip_mae": {
+        "mean": 0.026940759271383286,
+        "std": 0.0014969976588809874
+      },
+      "val_left_hip_r2": {
+        "mean": 0.835548734664917,
+        "std": 0.011550085431343364
+      },
+      "val_right_hip_mse": {
+        "mean": 0.0012803857447579504,
+        "std": 0.00016185634627738672
+      },
+      "val_right_hip_rmse": {
+        "mean": 0.035711558624281506,
+        "std": 0.0022517383023910624
+      },
+      "val_right_hip_mae": {
+        "mean": 0.026605210825800897,
+        "std": 0.0016736514590391936
+      },
+      "val_right_hip_r2": {
+        "mean": 0.8426500797271729,
+        "std": 0.009381944232058508
+      },
+      "val_left_knee_mse": {
+        "mean": 0.0013555370038375258,
+        "std": 0.00022403886973122852
+      },
+      "val_left_knee_rmse": {
+        "mean": 0.03669825835814189,
+        "std": 0.002962235189277587
+      },
+      "val_left_knee_mae": {
+        "mean": 0.027006322517991066,
+        "std": 0.0021506688874656293
+      },
+      "val_left_knee_r2": {
+        "mean": 0.8575150728225708,
+        "std": 0.017900681723992107
+      },
+      "val_right_knee_mse": {
+        "mean": 0.0014224715530872345,
+        "std": 0.00015865633511808594
+      },
+      "val_right_knee_rmse": {
+        "mean": 0.037657203068201986,
+        "std": 0.0020991927418496452
+      },
+      "val_right_knee_mae": {
+        "mean": 0.027371355146169663,
+        "std": 0.0015477201779498157
+      },
+      "val_right_knee_r2": {
+        "mean": 0.8675998687744141,
+        "std": 0.013914238272642479
+      },
+      "val_left_foot_mse": {
+        "mean": 0.0017160111805424095,
+        "std": 0.0003584292301293265
+      },
+      "val_left_foot_rmse": {
+        "mean": 0.04117133646892844,
+        "std": 0.004575175833199406
+      },
+      "val_left_foot_mae": {
+        "mean": 0.026308052614331244,
+        "std": 0.002102907487694088
+      },
+      "val_left_foot_r2": {
+        "mean": 0.6116551876068115,
+        "std": 0.03904958921136898
+      },
+      "val_right_foot_mse": {
+        "mean": 0.0018024345161393284,
+        "std": 0.0002053024187996034
+      },
+      "val_right_foot_rmse": {
+        "mean": 0.04238566844065751,
+        "std": 0.0024268553681624353
+      },
+      "val_right_foot_mae": {
+        "mean": 0.027043362334370612,
+        "std": 0.0008874200296480356
+      },
+      "val_right_foot_r2": {
+        "mean": 0.5955031394958497,
+        "std": 0.057589277119730366
+      }
+    },
+    "best_fold": "1",
+    "best_fold_metrics": {
+      "fold": 1,
+      "train_metrics": {
+        "train_mse": 0.0012734184274449944,
+        "train_rmse": 0.035684988825064726,
+        "train_mae": 0.026550384238362312,
+        "train_r2": 0.9401790499687195,
+        "train_head_mse": 0.0012963401386514306,
+        "train_head_rmse": 0.036004723838010906,
+        "train_head_mae": 0.027559440582990646,
+        "train_head_r2": 0.871606707572937,
+        "train_left_shoulder_mse": 0.0008136550313793123,
+        "train_left_shoulder_rmse": 0.028524639022769634,
+        "train_left_shoulder_mae": 0.02167939394712448,
+        "train_left_shoulder_r2": 0.8559739589691162,
+        "train_left_elbow_mse": 0.001271183486096561,
+        "train_left_elbow_rmse": 0.035653660206163416,
+        "train_left_elbow_mae": 0.02764335460960865,
+        "train_left_elbow_r2": 0.9078975319862366,
+        "train_right_shoulder_mse": 0.0007877332391217351,
+        "train_right_shoulder_rmse": 0.02806658581163258,
+        "train_right_shoulder_mae": 0.0213646050542593,
+        "train_right_shoulder_r2": 0.8807903528213501,
+        "train_right_elbow_mse": 0.0012887748889625072,
+        "train_right_elbow_rmse": 0.03589951098500518,
+        "train_right_elbow_mae": 0.027788985520601273,
+        "train_right_elbow_r2": 0.9190410375595093,
+        "train_left_hand_mse": 0.0021139057353138924,
+        "train_left_hand_rmse": 0.04597723061814285,
+        "train_left_hand_mae": 0.035943202674388885,
+        "train_left_hand_r2": 0.9188737273216248,
+        "train_right_hand_mse": 0.0021140018943697214,
+        "train_right_hand_rmse": 0.0459782763309992,
+        "train_right_hand_mae": 0.0354146845638752,
+        "train_right_hand_r2": 0.924353301525116,
+        "train_left_hip_mse": 0.001063541043549776,
+        "train_left_hip_rmse": 0.03261197699541958,
+        "train_left_hip_mae": 0.024693956598639488,
+        "train_left_hip_r2": 0.8669582605361938,
+        "train_right_hip_mse": 0.0010461374185979366,
+        "train_right_hip_rmse": 0.03234404765328447,
+        "train_right_hip_mae": 0.02425871230661869,
+        "train_right_hip_r2": 0.8739422559738159,
+        "train_left_knee_mse": 0.001019526505842805,
+        "train_left_knee_rmse": 0.03193002514629146,
+        "train_left_knee_mae": 0.023999694734811783,
+        "train_left_knee_r2": 0.8933584690093994,
+        "train_right_knee_mse": 0.0010396437719464302,
+        "train_right_knee_rmse": 0.03224350743865236,
+        "train_right_knee_mae": 0.02448125183582306,
+        "train_right_knee_r2": 0.9033292531967163,
+        "train_left_foot_mse": 0.0012758304364979267,
+        "train_left_foot_rmse": 0.03571876868675524,
+        "train_left_foot_mae": 0.024458788335323334,
+        "train_left_foot_r2": 0.7103928923606873,
+        "train_right_foot_mse": 0.0014241670724004507,
+        "train_right_foot_rmse": 0.037738138168177435,
+        "train_right_foot_mae": 0.025868939235806465,
+        "train_right_foot_r2": 0.6791626811027527
+      },
+      "val_metrics": {
+        "val_mse": 0.0013669708278030157,
+        "val_rmse": 0.03697256858541229,
+        "val_mae": 0.027096066623926163,
+        "val_r2": 0.9334621429443359,
+        "val_head_mse": 0.0013267889153212309,
+        "val_head_rmse": 0.03642511379970186,
+        "val_head_mae": 0.026944724842905998,
+        "val_head_r2": 0.8578331470489502,
+        "val_left_shoulder_mse": 0.0007886486710049212,
+        "val_left_shoulder_rmse": 0.028082889292323915,
+        "val_left_shoulder_mae": 0.021231085062026978,
+        "val_left_shoulder_r2": 0.8479042053222656,
+        "val_left_elbow_mse": 0.0012750765308737755,
+        "val_left_elbow_rmse": 0.03570821377321716,
+        "val_left_elbow_mae": 0.02827148325741291,
+        "val_left_elbow_r2": 0.9075881838798523,
+        "val_right_shoulder_mse": 0.0008302742498926818,
+        "val_right_shoulder_rmse": 0.028814479865038028,
+        "val_right_shoulder_mae": 0.021549466997385025,
+        "val_right_shoulder_r2": 0.8656619787216187,
+        "val_right_elbow_mse": 0.0013753921957686543,
+        "val_right_elbow_rmse": 0.037086280425093245,
+        "val_right_elbow_mae": 0.029128871858119965,
+        "val_right_elbow_r2": 0.909455418586731,
+        "val_left_hand_mse": 0.002249638084322214,
+        "val_left_hand_rmse": 0.04743034982289519,
+        "val_left_hand_mae": 0.037715133279561996,
+        "val_left_hand_r2": 0.9171456694602966,
+        "val_right_hand_mse": 0.0022658936213701963,
+        "val_right_hand_rmse": 0.04760140356512817,
+        "val_right_hand_mae": 0.03771165758371353,
+        "val_right_hand_r2": 0.9174893498420715,
+        "val_left_hip_mse": 0.0010675908997654915,
+        "val_left_hip_rmse": 0.03267400954528678,
+        "val_left_hip_mae": 0.024262193590402603,
+        "val_left_hip_r2": 0.8477381467819214,
+        "val_right_hip_mse": 0.0010715555399656296,
+        "val_right_hip_rmse": 0.03273462295438317,
+        "val_right_hip_mae": 0.023899253457784653,
+        "val_right_hip_r2": 0.8548038601875305,
+        "val_left_knee_mse": 0.0011254898272454739,
+        "val_left_knee_rmse": 0.033548320781306984,
+        "val_left_knee_mae": 0.02400522492825985,
+        "val_left_knee_r2": 0.8774427175521851,
+        "val_right_knee_mse": 0.0012089716037735343,
+        "val_right_knee_rmse": 0.034770268963203815,
+        "val_right_knee_mae": 0.02567751519382,
+        "val_right_knee_r2": 0.8875686526298523,
+        "val_left_foot_mse": 0.001618761569261551,
+        "val_left_foot_rmse": 0.040233836124107664,
+        "val_left_foot_mae": 0.025808263570070267,
+        "val_left_foot_r2": 0.6187542676925659,
+        "val_right_foot_mse": 0.001566539634950459,
+        "val_right_foot_rmse": 0.039579535557538556,
+        "val_right_foot_mae": 0.026043979451060295,
+        "val_right_foot_r2": 0.664306640625
+      },
+      "best_epoch": 100
+    }
+  },
+  "fold_details": [
+    {
+      "fold": 1,
+      "train_metrics": {
+        "train_mse": 0.0012734184274449944,
+        "train_rmse": 0.035684988825064726,
+        "train_mae": 0.026550384238362312,
+        "train_r2": 0.9401790499687195,
+        "train_head_mse": 0.0012963401386514306,
+        "train_head_rmse": 0.036004723838010906,
+        "train_head_mae": 0.027559440582990646,
+        "train_head_r2": 0.871606707572937,
+        "train_left_shoulder_mse": 0.0008136550313793123,
+        "train_left_shoulder_rmse": 0.028524639022769634,
+        "train_left_shoulder_mae": 0.02167939394712448,
+        "train_left_shoulder_r2": 0.8559739589691162,
+        "train_left_elbow_mse": 0.001271183486096561,
+        "train_left_elbow_rmse": 0.035653660206163416,
+        "train_left_elbow_mae": 0.02764335460960865,
+        "train_left_elbow_r2": 0.9078975319862366,
+        "train_right_shoulder_mse": 0.0007877332391217351,
+        "train_right_shoulder_rmse": 0.02806658581163258,
+        "train_right_shoulder_mae": 0.0213646050542593,
+        "train_right_shoulder_r2": 0.8807903528213501,
+        "train_right_elbow_mse": 0.0012887748889625072,
+        "train_right_elbow_rmse": 0.03589951098500518,
+        "train_right_elbow_mae": 0.027788985520601273,
+        "train_right_elbow_r2": 0.9190410375595093,
+        "train_left_hand_mse": 0.0021139057353138924,
+        "train_left_hand_rmse": 0.04597723061814285,
+        "train_left_hand_mae": 0.035943202674388885,
+        "train_left_hand_r2": 0.9188737273216248,
+        "train_right_hand_mse": 0.0021140018943697214,
+        "train_right_hand_rmse": 0.0459782763309992,
+        "train_right_hand_mae": 0.0354146845638752,
+        "train_right_hand_r2": 0.924353301525116,
+        "train_left_hip_mse": 0.001063541043549776,
+        "train_left_hip_rmse": 0.03261197699541958,
+        "train_left_hip_mae": 0.024693956598639488,
+        "train_left_hip_r2": 0.8669582605361938,
+        "train_right_hip_mse": 0.0010461374185979366,
+        "train_right_hip_rmse": 0.03234404765328447,
+        "train_right_hip_mae": 0.02425871230661869,
+        "train_right_hip_r2": 0.8739422559738159,
+        "train_left_knee_mse": 0.001019526505842805,
+        "train_left_knee_rmse": 0.03193002514629146,
+        "train_left_knee_mae": 0.023999694734811783,
+        "train_left_knee_r2": 0.8933584690093994,
+        "train_right_knee_mse": 0.0010396437719464302,
+        "train_right_knee_rmse": 0.03224350743865236,
+        "train_right_knee_mae": 0.02448125183582306,
+        "train_right_knee_r2": 0.9033292531967163,
+        "train_left_foot_mse": 0.0012758304364979267,
+        "train_left_foot_rmse": 0.03571876868675524,
+        "train_left_foot_mae": 0.024458788335323334,
+        "train_left_foot_r2": 0.7103928923606873,
+        "train_right_foot_mse": 0.0014241670724004507,
+        "train_right_foot_rmse": 0.037738138168177435,
+        "train_right_foot_mae": 0.025868939235806465,
+        "train_right_foot_r2": 0.6791626811027527
+      },
+      "val_metrics": {
+        "val_mse": 0.0013669708278030157,
+        "val_rmse": 0.03697256858541229,
+        "val_mae": 0.027096066623926163,
+        "val_r2": 0.9334621429443359,
+        "val_head_mse": 0.0013267889153212309,
+        "val_head_rmse": 0.03642511379970186,
+        "val_head_mae": 0.026944724842905998,
+        "val_head_r2": 0.8578331470489502,
+        "val_left_shoulder_mse": 0.0007886486710049212,
+        "val_left_shoulder_rmse": 0.028082889292323915,
+        "val_left_shoulder_mae": 0.021231085062026978,
+        "val_left_shoulder_r2": 0.8479042053222656,
+        "val_left_elbow_mse": 0.0012750765308737755,
+        "val_left_elbow_rmse": 0.03570821377321716,
+        "val_left_elbow_mae": 0.02827148325741291,
+        "val_left_elbow_r2": 0.9075881838798523,
+        "val_right_shoulder_mse": 0.0008302742498926818,
+        "val_right_shoulder_rmse": 0.028814479865038028,
+        "val_right_shoulder_mae": 0.021549466997385025,
+        "val_right_shoulder_r2": 0.8656619787216187,
+        "val_right_elbow_mse": 0.0013753921957686543,
+        "val_right_elbow_rmse": 0.037086280425093245,
+        "val_right_elbow_mae": 0.029128871858119965,
+        "val_right_elbow_r2": 0.909455418586731,
+        "val_left_hand_mse": 0.002249638084322214,
+        "val_left_hand_rmse": 0.04743034982289519,
+        "val_left_hand_mae": 0.037715133279561996,
+        "val_left_hand_r2": 0.9171456694602966,
+        "val_right_hand_mse": 0.0022658936213701963,
+        "val_right_hand_rmse": 0.04760140356512817,
+        "val_right_hand_mae": 0.03771165758371353,
+        "val_right_hand_r2": 0.9174893498420715,
+        "val_left_hip_mse": 0.0010675908997654915,
+        "val_left_hip_rmse": 0.03267400954528678,
+        "val_left_hip_mae": 0.024262193590402603,
+        "val_left_hip_r2": 0.8477381467819214,
+        "val_right_hip_mse": 0.0010715555399656296,
+        "val_right_hip_rmse": 0.03273462295438317,
+        "val_right_hip_mae": 0.023899253457784653,
+        "val_right_hip_r2": 0.8548038601875305,
+        "val_left_knee_mse": 0.0011254898272454739,
+        "val_left_knee_rmse": 0.033548320781306984,
+        "val_left_knee_mae": 0.02400522492825985,
+        "val_left_knee_r2": 0.8774427175521851,
+        "val_right_knee_mse": 0.0012089716037735343,
+        "val_right_knee_rmse": 0.034770268963203815,
+        "val_right_knee_mae": 0.02567751519382,
+        "val_right_knee_r2": 0.8875686526298523,
+        "val_left_foot_mse": 0.001618761569261551,
+        "val_left_foot_rmse": 0.040233836124107664,
+        "val_left_foot_mae": 0.025808263570070267,
+        "val_left_foot_r2": 0.6187542676925659,
+        "val_right_foot_mse": 0.001566539634950459,
+        "val_right_foot_rmse": 0.039579535557538556,
+        "val_right_foot_mae": 0.026043979451060295,
+        "val_right_foot_r2": 0.664306640625
+      },
+      "best_epoch": 100
+    },
+    {
+      "fold": 2,
+      "train_metrics": {
+        "train_mse": 0.0015614606672897935,
+        "train_rmse": 0.03951532193073711,
+        "train_mae": 0.02883830852806568,
+        "train_r2": 0.9255533814430237,
+        "train_head_mse": 0.0015013174852356315,
+        "train_head_rmse": 0.038746838390191676,
+        "train_head_mae": 0.029118983075022697,
+        "train_head_r2": 0.8464443683624268,
+        "train_left_shoulder_mse": 0.0009345314465463161,
+        "train_left_shoulder_rmse": 0.030570107074498712,
+        "train_left_shoulder_mae": 0.023350900039076805,
+        "train_left_shoulder_r2": 0.8310527801513672,
+        "train_left_elbow_mse": 0.0015524427872151136,
+        "train_left_elbow_rmse": 0.03940105058516985,
+        "train_left_elbow_mae": 0.03067999705672264,
+        "train_left_elbow_r2": 0.8883272409439087,
+        "train_right_shoulder_mse": 0.0010097706690430641,
+        "train_right_shoulder_rmse": 0.031776888913848444,
+        "train_right_shoulder_mae": 0.024040350690484047,
+        "train_right_shoulder_r2": 0.8426597118377686,
+        "train_right_elbow_mse": 0.0016670682234689593,
+        "train_right_elbow_rmse": 0.04082974679653254,
+        "train_right_elbow_mae": 0.03075317107141018,
+        "train_right_elbow_r2": 0.8934776186943054,
+        "train_left_hand_mse": 0.0026304807979613543,
+        "train_left_hand_rmse": 0.05128821305096673,
+        "train_left_hand_mae": 0.03958764299750328,
+        "train_left_hand_r2": 0.9004061222076416,
+        "train_right_hand_mse": 0.002782791154459119,
+        "train_right_hand_rmse": 0.05275216729632176,
+        "train_right_hand_mae": 0.0392480194568634,
+        "train_right_hand_r2": 0.8995761871337891,
+        "train_left_hip_mse": 0.0011238771257922053,
+        "train_left_hip_rmse": 0.03352427666322132,
+        "train_left_hip_mae": 0.025354130193591118,
+        "train_left_hip_r2": 0.851222038269043,
+        "train_right_hip_mse": 0.0011499463580548763,
+        "train_right_hip_rmse": 0.03391085899907102,
+        "train_right_hip_mae": 0.025267165154218674,
+        "train_right_hip_r2": 0.8528518676757812,
+        "train_left_knee_mse": 0.0012745781568810344,
+        "train_left_knee_rmse": 0.03570123466886033,
+        "train_left_knee_mae": 0.0265215877443552,
+        "train_left_knee_r2": 0.8628536462783813,
+        "train_right_knee_mse": 0.001316797686740756,
+        "train_right_knee_rmse": 0.036287707102278535,
+        "train_right_knee_mae": 0.027365650981664658,
+        "train_right_knee_r2": 0.877715528011322,
+        "train_left_foot_mse": 0.0017178094713017344,
+        "train_left_foot_rmse": 0.041446465124323144,
+        "train_left_foot_mae": 0.026393601670861244,
+        "train_left_foot_r2": 0.6022300720214844,
+        "train_right_foot_mse": 0.001637577428482473,
+        "train_right_foot_rmse": 0.04046699183881195,
+        "train_right_foot_mae": 0.027216792106628418,
+        "train_right_foot_r2": 0.641518771648407
+      },
+      "val_metrics": {
+        "val_mse": 0.0019491856219246984,
+        "val_rmse": 0.04414958235277768,
+        "val_mae": 0.03186870366334915,
+        "val_r2": 0.9105142951011658,
+        "val_head_mse": 0.0019519206834957004,
+        "val_head_rmse": 0.0441805464372692,
+        "val_head_mae": 0.032341692596673965,
+        "val_head_r2": 0.8161797523498535,
+        "val_left_shoulder_mse": 0.0011485640425235033,
+        "val_left_shoulder_rmse": 0.03389047126440562,
+        "val_left_shoulder_mae": 0.02533150091767311,
+        "val_left_shoulder_r2": 0.7967934608459473,
+        "val_left_elbow_mse": 0.0018383507849648595,
+        "val_left_elbow_rmse": 0.04287599310762212,
+        "val_left_elbow_mae": 0.0329643078148365,
+        "val_left_elbow_r2": 0.8627037405967712,
+        "val_right_shoulder_mse": 0.0013948227278888226,
+        "val_right_shoulder_rmse": 0.03734732557879911,
+        "val_right_shoulder_mae": 0.027158092707395554,
+        "val_right_shoulder_r2": 0.7987672686576843,
+        "val_right_elbow_mse": 0.0022705348674207926,
+        "val_right_elbow_rmse": 0.04765012977338879,
+        "val_right_elbow_mae": 0.03473043069243431,
+        "val_right_elbow_r2": 0.8600704669952393,
+        "val_left_hand_mse": 0.003247955348342657,
+        "val_left_hand_rmse": 0.056990835652257786,
+        "val_left_hand_mae": 0.043985966593027115,
+        "val_left_hand_r2": 0.8736846446990967,
+        "val_right_hand_mse": 0.003676592605188489,
+        "val_right_hand_rmse": 0.06063491242830725,
+        "val_right_hand_mae": 0.043874699622392654,
+        "val_right_hand_r2": 0.8702507615089417,
+        "val_left_hip_mse": 0.0014065408613532782,
+        "val_left_hip_rmse": 0.03750387795086367,
+        "val_left_hip_mae": 0.028496865183115005,
+        "val_left_hip_r2": 0.839414656162262,
+        "val_right_hip_mse": 0.0015354282222688198,
+        "val_right_hip_rmse": 0.03918454060300847,
+        "val_right_hip_mae": 0.029061945155262947,
+        "val_right_hip_r2": 0.8349571228027344,
+        "val_left_knee_mse": 0.0017549460753798485,
+        "val_left_knee_rmse": 0.04189207652265341,
+        "val_left_knee_mae": 0.030653607100248337,
+        "val_left_knee_r2": 0.8283937573432922,
+        "val_right_knee_mse": 0.0015137753216549754,
+        "val_right_knee_rmse": 0.03890726566664606,
+        "val_right_knee_mae": 0.029408888891339302,
+        "val_right_knee_r2": 0.858546257019043,
+        "val_left_foot_mse": 0.0017820688663050532,
+        "val_left_foot_rmse": 0.04221455751639538,
+        "val_left_foot_mae": 0.027876628562808037,
+        "val_left_foot_r2": 0.6108624339103699,
+        "val_right_foot_mse": 0.0018179105827584863,
+        "val_right_foot_rmse": 0.04263696263523571,
+        "val_right_foot_mae": 0.028408564627170563,
+        "val_right_foot_r2": 0.5614440441131592
+      },
+      "best_epoch": 98
+    },
+    {
+      "fold": 3,
+      "train_metrics": {
+        "train_mse": 0.0012790263863280416,
+        "train_rmse": 0.035763478386868934,
+        "train_mae": 0.026583004742860794,
+        "train_r2": 0.9388182163238525,
+        "train_head_mse": 0.0011763650691136718,
+        "train_head_rmse": 0.03429817880170421,
+        "train_head_mae": 0.026254532858729362,
+        "train_head_r2": 0.8804605007171631,
+        "train_left_shoulder_mse": 0.0007457094034180045,
+        "train_left_shoulder_rmse": 0.027307680300933737,
+        "train_left_shoulder_mae": 0.021003369241952896,
+        "train_left_shoulder_r2": 0.8631486892700195,
+        "train_left_elbow_mse": 0.001242591766640544,
+        "train_left_elbow_rmse": 0.0352504151272087,
+        "train_left_elbow_mae": 0.027316849678754807,
+        "train_left_elbow_r2": 0.9091324806213379,
+        "train_right_shoulder_mse": 0.0007266461616382003,
+        "train_right_shoulder_rmse": 0.026956375157617174,
+        "train_right_shoulder_mae": 0.02089906856417656,
+        "train_right_shoulder_r2": 0.8870998620986938,
+        "train_right_elbow_mse": 0.0012617878383025527,
+        "train_right_elbow_rmse": 0.035521653090791715,
+        "train_right_elbow_mae": 0.027573036029934883,
+        "train_right_elbow_r2": 0.9187759757041931,
+        "train_left_hand_mse": 0.002024478744715452,
+        "train_left_hand_rmse": 0.044994207901856126,
+        "train_left_hand_mae": 0.03509342670440674,
+        "train_left_hand_r2": 0.9222167730331421,
+        "train_right_hand_mse": 0.0020789802074432373,
+        "train_right_hand_rmse": 0.04559583541775759,
+        "train_right_hand_mae": 0.03555019199848175,
+        "train_right_hand_r2": 0.9240489602088928,
+        "train_left_hip_mse": 0.0011243551271036267,
+        "train_left_hip_rmse": 0.033531405086927486,
+        "train_left_hip_mae": 0.02547801099717617,
+        "train_left_hip_r2": 0.8550944328308105,
+        "train_right_hip_mse": 0.001106569543480873,
+        "train_right_hip_rmse": 0.03326514006405013,
+        "train_right_hip_mae": 0.025070156902074814,
+        "train_right_hip_r2": 0.8635149002075195,
+        "train_left_knee_mse": 0.0011253359261900187,
+        "train_left_knee_rmse": 0.03354602698070248,
+        "train_left_knee_mae": 0.025196490809321404,
+        "train_left_knee_r2": 0.8804743885993958,
+        "train_right_knee_mse": 0.0011050159810110927,
+        "train_right_knee_rmse": 0.03324178065343511,
+        "train_right_knee_mae": 0.02475661411881447,
+        "train_right_knee_r2": 0.8966816663742065,
+        "train_left_foot_mse": 0.0014200935838744044,
+        "train_left_foot_rmse": 0.03768412907145931,
+        "train_left_foot_mae": 0.025054579600691795,
+        "train_left_foot_r2": 0.6667464971542358,
+        "train_right_foot_mse": 0.0014894143678247929,
+        "train_right_foot_rmse": 0.038592931578526044,
+        "train_right_foot_mae": 0.026332709938287735,
+        "train_right_foot_r2": 0.6649521589279175
+      },
+      "val_metrics": {
+        "val_mse": 0.001638404792174697,
+        "val_rmse": 0.04047721324615489,
+        "val_mae": 0.02911573462188244,
+        "val_r2": 0.9257985353469849,
+        "val_head_mse": 0.0014651210512965918,
+        "val_head_rmse": 0.03827689970852644,
+        "val_head_mae": 0.02846965752542019,
+        "val_head_r2": 0.8588935732841492,
+        "val_left_shoulder_mse": 0.0010537024354562163,
+        "val_left_shoulder_rmse": 0.03246078303824811,
+        "val_left_shoulder_mae": 0.023874560371041298,
+        "val_left_shoulder_r2": 0.8243831992149353,
+        "val_left_elbow_mse": 0.001650865888223052,
+        "val_left_elbow_rmse": 0.04063084897246244,
+        "val_left_elbow_mae": 0.03040950372815132,
+        "val_left_elbow_r2": 0.8847611546516418,
+        "val_right_shoulder_mse": 0.0009412385406903923,
+        "val_right_shoulder_rmse": 0.030679611156114613,
+        "val_right_shoulder_mae": 0.022934339940547943,
+        "val_right_shoulder_r2": 0.8631192445755005,
+        "val_right_elbow_mse": 0.001665425719693303,
+        "val_right_elbow_rmse": 0.04080962778185196,
+        "val_right_elbow_mae": 0.03136920928955078,
+        "val_right_elbow_r2": 0.9004873037338257,
+        "val_left_hand_mse": 0.0025085045490413904,
+        "val_left_hand_rmse": 0.05008497328582087,
+        "val_left_hand_mae": 0.03840486705303192,
+        "val_left_hand_r2": 0.9081084132194519,
+        "val_right_hand_mse": 0.002905910601839423,
+        "val_right_hand_rmse": 0.05390649869764705,
+        "val_right_hand_mae": 0.040518734604120255,
+        "val_right_hand_r2": 0.9023516774177551,
+        "val_left_hip_mse": 0.0014503411948680878,
+        "val_left_hip_rmse": 0.038083345373904426,
+        "val_left_hip_mae": 0.027614036574959755,
+        "val_left_hip_r2": 0.8175036311149597,
+        "val_right_hip_mse": 0.0013523006346076727,
+        "val_right_hip_rmse": 0.03677364048619164,
+        "val_right_hip_mae": 0.026614289730787277,
+        "val_right_hip_r2": 0.8338983058929443,
+        "val_left_knee_mse": 0.001419665408320725,
+        "val_left_knee_rmse": 0.037678447530660346,
+        "val_left_knee_mae": 0.02733062021434307,
+        "val_left_knee_r2": 0.8547688126564026,
+        "val_right_knee_mse": 0.0013064589584246278,
+        "val_right_knee_rmse": 0.03614497141269623,
+        "val_right_knee_mae": 0.025865653529763222,
+        "val_right_knee_r2": 0.881216287612915,
+        "val_left_foot_mse": 0.0020077244844287634,
+        "val_left_foot_rmse": 0.04480763868392044,
+        "val_left_foot_mae": 0.02791229635477066,
+        "val_left_foot_r2": 0.5818337202072144,
+        "val_right_foot_mse": 0.0015720025403425097,
+        "val_right_foot_rmse": 0.0396484872390172,
+        "val_right_foot_mae": 0.02718675695359707,
+        "val_right_foot_r2": 0.6608878374099731
+      },
+      "best_epoch": 91
+    },
+    {
+      "fold": 4,
+      "train_metrics": {
+        "train_mse": 0.001371898571960628,
+        "train_rmse": 0.03703914917976151,
+        "train_mae": 0.02743724174797535,
+        "train_r2": 0.9355619549751282,
+        "train_head_mse": 0.001376723637804389,
+        "train_head_rmse": 0.0371042266838212,
+        "train_head_mae": 0.027764396741986275,
+        "train_head_r2": 0.8617536425590515,
+        "train_left_shoulder_mse": 0.0008315580198541284,
+        "train_left_shoulder_rmse": 0.028836747733649304,
+        "train_left_shoulder_mae": 0.021917693316936493,
+        "train_left_shoulder_r2": 0.8512165546417236,
+        "train_left_elbow_mse": 0.0013927394757047296,
+        "train_left_elbow_rmse": 0.03731942491122726,
+        "train_left_elbow_mae": 0.028923772275447845,
+        "train_left_elbow_r2": 0.8990980386734009,
+        "train_right_shoulder_mse": 0.0008294542203657329,
+        "train_right_shoulder_rmse": 0.028800246880291373,
+        "train_right_shoulder_mae": 0.02179909497499466,
+        "train_right_shoulder_r2": 0.8732565641403198,
+        "train_right_elbow_mse": 0.0014205370098352432,
+        "train_right_elbow_rmse": 0.037690012069980065,
+        "train_right_elbow_mae": 0.029091544449329376,
+        "train_right_elbow_r2": 0.9098526835441589,
+        "train_left_hand_mse": 0.002300370717421174,
+        "train_left_hand_rmse": 0.04796218007369112,
+        "train_left_hand_mae": 0.03757936507463455,
+        "train_left_hand_r2": 0.9129942655563354,
+        "train_right_hand_mse": 0.0024476528633385897,
+        "train_right_hand_rmse": 0.049473759341074834,
+        "train_right_hand_mae": 0.038064852356910706,
+        "train_right_hand_r2": 0.9124616384506226,
+        "train_left_hip_mse": 0.0011089489562436938,
+        "train_left_hip_rmse": 0.03330088521711839,
+        "train_left_hip_mae": 0.0255591943860054,
+        "train_left_hip_r2": 0.8589164018630981,
+        "train_right_hip_mse": 0.0010876995511353016,
+        "train_right_hip_rmse": 0.032980290343405126,
+        "train_right_hip_mae": 0.024931492283940315,
+        "train_right_hip_r2": 0.8670696020126343,
+        "train_left_knee_mse": 0.001117759384214878,
+        "train_left_knee_rmse": 0.033432908701081905,
+        "train_left_knee_mae": 0.025413010269403458,
+        "train_left_knee_r2": 0.8843995332717896,
+        "train_right_knee_mse": 0.0011729779653251171,
+        "train_right_knee_rmse": 0.03424876589492119,
+        "train_right_knee_mae": 0.02610812336206436,
+        "train_right_knee_r2": 0.8931787610054016,
+        "train_left_foot_mse": 0.0014348349068313837,
+        "train_left_foot_rmse": 0.0378792147071634,
+        "train_left_foot_mae": 0.024688156321644783,
+        "train_left_foot_r2": 0.6630778312683105,
+        "train_right_foot_mse": 0.0013134240871295333,
+        "train_right_foot_rmse": 0.03624119323545423,
+        "train_right_foot_mae": 0.024843459948897362,
+        "train_right_foot_r2": 0.7116966247558594
+      },
+      "val_metrics": {
+        "val_mse": 0.0016774431569501758,
+        "val_rmse": 0.04095660089595053,
+        "val_mae": 0.02889617159962654,
+        "val_r2": 0.9183139801025391,
+        "val_head_mse": 0.0015361307887360454,
+        "val_head_rmse": 0.039193504420197554,
+        "val_head_mae": 0.02928389422595501,
+        "val_head_r2": 0.8450303673744202,
+        "val_left_shoulder_mse": 0.0009139323956333101,
+        "val_left_shoulder_rmse": 0.030231314818137005,
+        "val_left_shoulder_mae": 0.02285638079047203,
+        "val_left_shoulder_r2": 0.8318844437599182,
+        "val_left_elbow_mse": 0.0016869965475052595,
+        "val_left_elbow_rmse": 0.04107306352714951,
+        "val_left_elbow_mae": 0.031246719881892204,
+        "val_left_elbow_r2": 0.8778408169746399,
+        "val_right_shoulder_mse": 0.0009116395376622677,
+        "val_right_shoulder_rmse": 0.030193369100884843,
+        "val_right_shoulder_mae": 0.02234402857720852,
+        "val_right_shoulder_r2": 0.8584368228912354,
+        "val_right_elbow_mse": 0.0018057312117889524,
+        "val_right_elbow_rmse": 0.042493896170967334,
+        "val_right_elbow_mae": 0.02988065406680107,
+        "val_right_elbow_r2": 0.885803759098053,
+        "val_left_hand_mse": 0.002889244118705392,
+        "val_left_hand_rmse": 0.05375168944977815,
+        "val_left_hand_mae": 0.0398484505712986,
+        "val_left_hand_r2": 0.8873634934425354,
+        "val_right_hand_mse": 0.002922926563769579,
+        "val_right_hand_rmse": 0.054064096808969064,
+        "val_right_hand_mae": 0.03874985873699188,
+        "val_right_hand_r2": 0.8931021094322205,
+        "val_left_hip_mse": 0.0011623839382082224,
+        "val_left_hip_rmse": 0.03409375218728825,
+        "val_left_hip_mae": 0.026439180597662926,
+        "val_left_hip_r2": 0.845921516418457,
+        "val_right_hip_mse": 0.0011499712709337473,
+        "val_right_hip_rmse": 0.03391122632600814,
+        "val_right_hip_mae": 0.026169009506702423,
+        "val_right_hip_r2": 0.8533806800842285,
+        "val_left_knee_mse": 0.0012985683279111981,
+        "val_left_knee_rmse": 0.03603565356575621,
+        "val_left_knee_mae": 0.026868587359786034,
+        "val_left_knee_r2": 0.8516912460327148,
+        "val_right_knee_mse": 0.0014188772765919566,
+        "val_right_knee_rmse": 0.037667987424230095,
+        "val_right_knee_mae": 0.026948794722557068,
+        "val_right_knee_r2": 0.8560424447059631,
+        "val_left_foot_mse": 0.0020904801785945892,
+        "val_left_foot_rmse": 0.04572176919799352,
+        "val_left_foot_mae": 0.027545904740691185,
+        "val_left_foot_r2": 0.5668109655380249,
+        "val_right_foot_mse": 0.0020198787096887827,
+        "val_right_foot_rmse": 0.044943060751230356,
+        "val_right_foot_mae": 0.027468761429190636,
+        "val_right_foot_r2": 0.5189753770828247
+      },
+      "best_epoch": 82
+    },
+    {
+      "fold": 5,
+      "train_metrics": {
+        "train_mse": 0.0013515838654711843,
+        "train_rmse": 0.036763893502609106,
+        "train_mae": 0.02691934071481228,
+        "train_r2": 0.9363779425621033,
+        "train_head_mse": 0.0012808259343728423,
+        "train_head_rmse": 0.03578862856233586,
+        "train_head_mae": 0.026594949886202812,
+        "train_head_r2": 0.8728370070457458,
+        "train_left_shoulder_mse": 0.0008391075534746051,
+        "train_left_shoulder_rmse": 0.028967353235575477,
+        "train_left_shoulder_mae": 0.021844839677214622,
+        "train_left_shoulder_r2": 0.8495293855667114,
+        "train_left_elbow_mse": 0.0013010292313992977,
+        "train_left_elbow_rmse": 0.03606978280221961,
+        "train_left_elbow_mae": 0.02796597220003605,
+        "train_left_elbow_r2": 0.9060801267623901,
+        "train_right_shoulder_mse": 0.0007848091772757471,
+        "train_right_shoulder_rmse": 0.028014445867725942,
+        "train_right_shoulder_mae": 0.021217359229922295,
+        "train_right_shoulder_r2": 0.881331205368042,
+        "train_right_elbow_mse": 0.0012962755281478167,
+        "train_right_elbow_rmse": 0.036003826576460125,
+        "train_right_elbow_mae": 0.02758239023387432,
+        "train_right_elbow_r2": 0.9190468192100525,
+        "train_left_hand_mse": 0.002129907486960292,
+        "train_left_hand_rmse": 0.04615092075961531,
+        "train_left_hand_mae": 0.035938456654548645,
+        "train_left_hand_r2": 0.9196220636367798,
+        "train_right_hand_mse": 0.0021580136381089687,
+        "train_right_hand_rmse": 0.04645442538778161,
+        "train_right_hand_mae": 0.03499883785843849,
+        "train_right_hand_r2": 0.9236563444137573,
+        "train_left_hip_mse": 0.0011923470301553607,
+        "train_left_hip_rmse": 0.03453037836681435,
+        "train_left_hip_mae": 0.026137933135032654,
+        "train_left_hip_r2": 0.8475073575973511,
+        "train_right_hip_mse": 0.0011626238701865077,
+        "train_right_hip_rmse": 0.03409727071462623,
+        "train_right_hip_mae": 0.02572827786207199,
+        "train_right_hip_r2": 0.857707142829895,
+        "train_left_knee_mse": 0.0011192047968506813,
+        "train_left_knee_rmse": 0.033454518332367025,
+        "train_left_knee_mae": 0.024881357327103615,
+        "train_left_knee_r2": 0.8821298480033875,
+        "train_right_knee_mse": 0.0011830884031951427,
+        "train_right_knee_rmse": 0.03439605214548819,
+        "train_right_knee_mae": 0.02581608295440674,
+        "train_right_knee_r2": 0.8882222175598145,
+        "train_left_foot_mse": 0.0015540290623903275,
+        "train_left_foot_rmse": 0.03942117530452799,
+        "train_left_foot_mae": 0.02510899305343628,
+        "train_left_foot_r2": 0.6637502908706665,
+        "train_right_foot_mse": 0.0015693285968154669,
+        "train_right_foot_rmse": 0.039614752262452264,
+        "train_right_foot_mae": 0.026135971769690514,
+        "train_right_foot_r2": 0.6443822383880615
+      },
+      "val_metrics": {
+        "val_mse": 0.0015228185802698135,
+        "val_rmse": 0.03902330816665616,
+        "val_mae": 0.02843872644007206,
+        "val_r2": 0.9265280961990356,
+        "val_head_mse": 0.0013639491517096758,
+        "val_head_rmse": 0.036931682221497515,
+        "val_head_mae": 0.028418222442269325,
+        "val_head_r2": 0.8557766675949097,
+        "val_left_shoulder_mse": 0.0009118955349549651,
+        "val_left_shoulder_rmse": 0.030197608099897005,
+        "val_left_shoulder_mae": 0.02322995848953724,
+        "val_left_shoulder_r2": 0.8339238166809082,
+        "val_left_elbow_mse": 0.0014284227509051561,
+        "val_left_elbow_rmse": 0.03779448042909382,
+        "val_left_elbow_mae": 0.029456648975610733,
+        "val_left_elbow_r2": 0.8951900005340576,
+        "val_right_shoulder_mse": 0.000863642490003258,
+        "val_right_shoulder_rmse": 0.029387794915632204,
+        "val_right_shoulder_mae": 0.022751640528440475,
+        "val_right_shoulder_r2": 0.8599545955657959,
+        "val_right_elbow_mse": 0.0014955442165955901,
+        "val_right_elbow_rmse": 0.03867226676308993,
+        "val_right_elbow_mae": 0.0298424381762743,
+        "val_right_elbow_r2": 0.8991358876228333,
+        "val_left_hand_mse": 0.0025119066704064608,
+        "val_left_hand_rmse": 0.050118925271861736,
+        "val_left_hand_mae": 0.038391899317502975,
+        "val_left_hand_r2": 0.9012941718101501,
+        "val_right_hand_mse": 0.002635727636516094,
+        "val_right_hand_rmse": 0.05133933809970766,
+        "val_right_hand_mae": 0.03880413994193077,
+        "val_right_hand_r2": 0.8993859887123108,
+        "val_left_hip_mse": 0.0013327283086255193,
+        "val_left_hip_rmse": 0.03650655158496238,
+        "val_left_hip_mae": 0.02789152041077614,
+        "val_left_hip_r2": 0.8271657228469849,
+        "val_right_hip_mse": 0.0012926730560138822,
+        "val_right_hip_rmse": 0.035953762751816144,
+        "val_right_hip_mae": 0.02728155627846718,
+        "val_right_hip_r2": 0.8362104296684265,
+        "val_left_knee_mse": 0.0011790153803303838,
+        "val_left_knee_rmse": 0.03433679339033253,
+        "val_left_knee_mae": 0.02617357298731804,
+        "val_left_knee_r2": 0.8752788305282593,
+        "val_right_knee_mse": 0.0016642746049910784,
+        "val_right_knee_rmse": 0.04079552187423368,
+        "val_right_knee_mae": 0.02895592339336872,
+        "val_right_knee_r2": 0.8546257019042969,
+        "val_left_foot_mse": 0.0010810208041220903,
+        "val_left_foot_rmse": 0.03287888082222523,
+        "val_left_foot_mae": 0.022397169843316078,
+        "val_left_foot_r2": 0.6800145506858826,
+        "val_right_foot_mse": 0.0020358411129564047,
+        "val_right_foot_rmse": 0.045120296020265696,
+        "val_right_foot_mae": 0.026108749210834503,
+        "val_right_foot_r2": 0.571901798248291
+      },
+      "best_epoch": 100
+    }
+  ],
+  "test_metrics": {
+    "test_mse": 0.00152857624925673,
+    "test_rmse": 0.039097010745793984,
+    "test_mae": 0.028835684061050415,
+    "test_r2": 0.9283223152160645,
+    "test_head_mse": 0.0014639559667557478,
+    "test_head_rmse": 0.038261677521454124,
+    "test_head_mae": 0.029478298500180244,
+    "test_head_r2": 0.8563132286071777,
+    "test_left_shoulder_mse": 0.0009500239975750446,
+    "test_left_shoulder_rmse": 0.030822459304459218,
+    "test_left_shoulder_mae": 0.02318173088133335,
+    "test_left_shoulder_r2": 0.8333448171615601,
+    "test_left_elbow_mse": 0.001431874348782003,
+    "test_left_elbow_rmse": 0.03784011560212261,
+    "test_left_elbow_mae": 0.029071349650621414,
+    "test_left_elbow_r2": 0.894320547580719,
+    "test_right_shoulder_mse": 0.001005183788947761,
+    "test_right_shoulder_rmse": 0.031704633556434006,
+    "test_right_shoulder_mae": 0.023657025769352913,
+    "test_right_shoulder_r2": 0.8490133285522461,
+    "test_right_elbow_mse": 0.0016737201949581504,
+    "test_right_elbow_rmse": 0.040911125564547235,
+    "test_right_elbow_mae": 0.031102659180760384,
+    "test_right_elbow_r2": 0.8929486870765686,
+    "test_left_hand_mse": 0.0023705230560153723,
+    "test_left_hand_rmse": 0.0486880175814889,
+    "test_left_hand_mae": 0.037719372659921646,
+    "test_left_hand_r2": 0.9094857573509216,
+    "test_right_hand_mse": 0.0026207203045487404,
+    "test_right_hand_rmse": 0.051192971241653284,
+    "test_right_hand_mae": 0.03951131924986839,
+    "test_right_hand_r2": 0.9068613648414612,
+    "test_left_hip_mse": 0.0013033578870818019,
+    "test_left_hip_rmse": 0.03610204823942544,
+    "test_left_hip_mae": 0.02688746340572834,
+    "test_left_hip_r2": 0.8521227836608887,
+    "test_right_hip_mse": 0.0013514793245121837,
+    "test_right_hip_rmse": 0.036762471686655994,
+    "test_right_hip_mae": 0.026841389015316963,
+    "test_right_hip_r2": 0.8525161147117615,
+    "test_left_knee_mse": 0.001223901053890586,
+    "test_left_knee_rmse": 0.03498429724734493,
+    "test_left_knee_mae": 0.02541135437786579,
+    "test_left_knee_r2": 0.8742030262947083,
+    "test_right_knee_mse": 0.0012863799929618835,
+    "test_right_knee_rmse": 0.035866139922800216,
+    "test_right_knee_mae": 0.02693943679332733,
+    "test_right_knee_r2": 0.8800403475761414,
+    "test_left_foot_mse": 0.001381010515615344,
+    "test_left_foot_rmse": 0.03716194983602642,
+    "test_left_foot_mae": 0.026248158887028694,
+    "test_left_foot_r2": 0.7030462026596069,
+    "test_right_foot_mse": 0.0018093610415235162,
+    "test_right_foot_rmse": 0.042536584742119535,
+    "test_right_foot_mae": 0.028814321383833885,
+    "test_right_foot_r2": 0.6553394794464111
+  }
+}

A9/cv_results_conv1D_variants/conv1d_fold1_best.h5

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

A9/cv_results_conv1D_variants/conv1d_fold2_best.h5

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

A9/cv_results_conv1D_variants/conv1d_fold3_best.h5

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

A9/cv_results_conv1D_variants/conv1d_fold4_best.h5

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

A9/cv_results_conv1D_variants/conv1d_fold5_best.h5

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

A9/cv_results_conv1D_variants/conv1d_v2_cv_results.json

@@ -0,0 +1,1243 @@
+{
+  "model_name": "conv1d_v2",
+  "n_splits": 5,
+  "epochs": 100,
+  "batch_size": 32,
+  "patience": 10,
+  "aggregate_results": {
+    "n_folds": 5,
+    "train_metrics": {
+      "train_mse": {
+        "mean": 0.004189881170168519,
+        "std": 0.0006816715496811422
+      },
+      "train_rmse": {
+        "mean": 0.06451685860784324,
+        "std": 0.005239859305752642
+      },
+      "train_mae": {
+        "mean": 0.048024240881204605,
+        "std": 0.0042668215985116045
+      },
+      "train_r2": {
+        "mean": 0.8017601013183594,
+        "std": 0.03229650465686602
+      },
+      "train_head_mse": {
+        "mean": 0.003033032175153494,
+        "std": 0.0005226949683597734
+      },
+      "train_head_rmse": {
+        "mean": 0.0548712198771429,
+        "std": 0.0047097138286453
+      },
+      "train_head_mae": {
+        "mean": 0.042640633136034015,
+        "std": 0.00385829937703636
+      },
+      "train_head_r2": {
+        "mean": 0.6947275519371032,
+        "std": 0.05443656728597148
+      },
+      "train_left_shoulder_mse": {
+        "mean": 0.0019546084804460406,
+        "std": 0.00030388201774506063
+      },
+      "train_left_shoulder_rmse": {
+        "mean": 0.04407728124515775,
+        "std": 0.003435368754778054
+      },
+      "train_left_shoulder_mae": {
+        "mean": 0.034324729815125465,
+        "std": 0.0029035608121121244
+      },
+      "train_left_shoulder_r2": {
+        "mean": 0.6481709957122803,
+        "std": 0.0556167862524384
+      },
+      "train_left_elbow_mse": {
+        "mean": 0.0038434856571257113,
+        "std": 0.0007768449862696837
+      },
+      "train_left_elbow_rmse": {
+        "mean": 0.0616828321050555,
+        "std": 0.006222047944627788
+      },
+      "train_left_elbow_mae": {
+        "mean": 0.04853492826223373,
+        "std": 0.005036077021894072
+      },
+      "train_left_elbow_r2": {
+        "mean": 0.7215797901153564,
+        "std": 0.056427117091941446
+      },
+      "train_right_shoulder_mse": {
+        "mean": 0.0023689745226874947,
+        "std": 0.000352698129654781
+      },
+      "train_right_shoulder_rmse": {
+        "mean": 0.04853704789599529,
+        "std": 0.0036234657759324257
+      },
+      "train_right_shoulder_mae": {
+        "mean": 0.0377688467502594,
+        "std": 0.003189959178407126
+      },
+      "train_right_shoulder_r2": {
+        "mean": 0.6364410758018494,
+        "std": 0.05636254608200453
+      },
+      "train_right_elbow_mse": {
+        "mean": 0.004395840456709266,
+        "std": 0.0008234554156598675
+      },
+      "train_right_elbow_rmse": {
+        "mean": 0.06601565158258048,
+        "std": 0.0061460721470388
+      },
+      "train_right_elbow_mae": {
+        "mean": 0.051519157737493514,
+        "std": 0.005556882378088948
+      },
+      "train_right_elbow_r2": {
+        "mean": 0.7209933757781982,
+        "std": 0.05370125540341452
+      },
+      "train_left_hand_mse": {
+        "mean": 0.007526151556521654,
+        "std": 0.0015520370762865335
+      },
+      "train_left_hand_rmse": {
+        "mean": 0.08629743391887165,
+        "std": 0.008882817995412967
+      },
+      "train_left_hand_mae": {
+        "mean": 0.06772734895348549,
+        "std": 0.0070509189129446944
+      },
+      "train_left_hand_r2": {
+        "mean": 0.7137839198112488,
+        "std": 0.058396331277407475
+      },
+      "train_right_hand_mse": {
+        "mean": 0.008008821401745081,
+        "std": 0.001653979176201106
+      },
+      "train_right_hand_rmse": {
+        "mean": 0.08902119976472678,
+        "std": 0.009167736590548984
+      },
+      "train_right_hand_mae": {
+        "mean": 0.06833488419651985,
+        "std": 0.00796811904226451
+      },
+      "train_right_hand_r2": {
+        "mean": 0.7122123956680297,
+        "std": 0.06008807982816971
+      },
+      "train_left_hip_mse": {
+        "mean": 0.003786041075363755,
+        "std": 0.00077319508608496
+      },
+      "train_left_hip_rmse": {
+        "mean": 0.06119236658138614,
+        "std": 0.0064447922800515535
+      },
+      "train_left_hip_mae": {
+        "mean": 0.04556304141879082,
+        "std": 0.005946569958882972
+      },
+      "train_left_hip_r2": {
+        "mean": 0.5142432689666748,
+        "std": 0.09932022340080027
+      },
+      "train_right_hip_mse": {
+        "mean": 0.0039231596048921345,
+        "std": 0.000837088335155621
+      },
+      "train_right_hip_rmse": {
+        "mean": 0.06225503809566539,
+        "std": 0.0068898357454576
+      },
+      "train_right_hip_mae": {
+        "mean": 0.046081209927797316,
+        "std": 0.00636589003664032
+      },
+      "train_right_hip_r2": {
+        "mean": 0.5161603450775146,
+        "std": 0.10385024731703577
+      },
+      "train_left_knee_mse": {
+        "mean": 0.0036639925092458725,
+        "std": 0.0006557887853387359
+      },
+      "train_left_knee_rmse": {
+        "mean": 0.06028615744446756,
+        "std": 0.0054379895022647
+      },
+      "train_left_knee_mae": {
+        "mean": 0.04464539662003517,
+        "std": 0.003909836716461259
+      },
+      "train_left_knee_r2": {
+        "mean": 0.6140255689620971,
+        "std": 0.06683211702699614
+      },
+      "train_right_knee_mse": {
+        "mean": 0.003805920807644725,
+        "std": 0.0006887173044141474
+      },
+      "train_right_knee_rmse": {
+        "mean": 0.06142187110845705,
+        "std": 0.005768410281941993
+      },
+      "train_right_knee_mae": {
+        "mean": 0.045633669942617416,
+        "std": 0.004325583519887196
+      },
+      "train_right_knee_r2": {
+        "mean": 0.6468051910400391,
+        "std": 0.06048327830690329
+      },
+      "train_left_foot_mse": {
+        "mean": 0.0039350977167487144,
+        "std": 9.225520943953724e-05
+      },
+      "train_left_foot_rmse": {
+        "mean": 0.06272602873632047,
+        "std": 0.0007369095731598132
+      },
+      "train_left_foot_mae": {
+        "mean": 0.04483037814497948,
+        "std": 0.0008143586041787857
+      },
+      "train_left_foot_r2": {
+        "mean": 0.09900742769241333,
+        "std": 0.04143879705252255
+      },
+      "train_right_foot_mse": {
+        "mean": 0.0042233305051922795,
+        "std": 0.00017425802184662187
+      },
+      "train_right_foot_rmse": {
+        "mean": 0.06497314943422627,
+        "std": 0.001349206355596279
+      },
+      "train_right_foot_mae": {
+        "mean": 0.04671090617775917,
+        "std": 0.0006156563318552435
+      },
+      "train_right_foot_r2": {
+        "mean": 0.05848702192306519,
+        "std": 0.027442780672513212
+      }
+    },
+    "val_metrics": {
+      "val_mse": {
+        "mean": 0.004232735652476549,
+        "std": 0.0006971365522500197
+      },
+      "val_rmse": {
+        "mean": 0.06483074769308018,
+        "std": 0.005450670237019158
+      },
+      "val_mae": {
+        "mean": 0.04814460128545761,
+        "std": 0.003990709316652403
+      },
+      "val_r2": {
+        "mean": 0.7999907493591308,
+        "std": 0.031160747771842207
+      },
+      "val_head_mse": {
+        "mean": 0.003062773309648037,
+        "std": 0.0005710391956606941
+      },
+      "val_head_rmse": {
+        "mean": 0.05509508354534458,
+        "std": 0.00522542618161795
+      },
+      "val_head_mae": {
+        "mean": 0.04282923862338066,
+        "std": 0.004275146766264847
+      },
+      "val_head_r2": {
+        "mean": 0.6936161637306213,
+        "std": 0.04562635362339771
+      },
+      "val_left_shoulder_mse": {
+        "mean": 0.001972725451923907,
+        "std": 0.00027919613217668523
+      },
+      "val_left_shoulder_rmse": {
+        "mean": 0.04429980547197143,
+        "std": 0.0032019817409529375
+      },
+      "val_left_shoulder_mae": {
+        "mean": 0.03440824560821056,
+        "std": 0.00262568683471993
+      },
+      "val_left_shoulder_r2": {
+        "mean": 0.6456697106361389,
+        "std": 0.04073817121400206
+      },
+      "val_left_elbow_mse": {
+        "mean": 0.003892540652304888,
+        "std": 0.0007087172366264799
+      },
+      "val_left_elbow_rmse": {
+        "mean": 0.06213262028392545,
+        "std": 0.00566375749466922
+      },
+      "val_left_elbow_mae": {
+        "mean": 0.04876239076256752,
+        "std": 0.004066627894983642
+      },
+      "val_left_elbow_r2": {
+        "mean": 0.717805278301239,
+        "std": 0.05068272727155428
+      },
+      "val_right_shoulder_mse": {
+        "mean": 0.0023743516532704232,
+        "std": 0.00040351181278143597
+      },
+      "val_right_shoulder_rmse": {
+        "mean": 0.048550101554458054,
+        "std": 0.004152022678434351
+      },
+      "val_right_shoulder_mae": {
+        "mean": 0.03745114952325821,
+        "std": 0.0034407391251764343
+      },
+      "val_right_shoulder_r2": {
+        "mean": 0.6373598337173462,
+        "std": 0.048555836598630595
+      },
+      "val_right_elbow_mse": {
+        "mean": 0.0044254438485950235,
+        "std": 0.0009904146408568702
+      },
+      "val_right_elbow_rmse": {
+        "mean": 0.06609364605578236,
+        "std": 0.007554720355379651
+      },
+      "val_right_elbow_mae": {
+        "mean": 0.05146452784538269,
+        "std": 0.005391730930656805
+      },
+      "val_right_elbow_r2": {
+        "mean": 0.7205804467201233,
+        "std": 0.055989945893025285
+      },
+      "val_left_hand_mse": {
+        "mean": 0.007698719669133425,
+        "std": 0.0014424490042073974
+      },
+      "val_left_hand_rmse": {
+        "mean": 0.08735188282695923,
+        "std": 0.00826850867560826
+      },
+      "val_left_hand_mae": {
+        "mean": 0.06851950362324714,
+        "std": 0.006088665495251121
+      },
+      "val_left_hand_r2": {
+        "mean": 0.7061087250709533,
+        "std": 0.057988467868729285
+      },
+      "val_right_hand_mse": {
+        "mean": 0.008156843949109316,
+        "std": 0.0019101860897293762
+      },
+      "val_right_hand_rmse": {
+        "mean": 0.0896570823513265,
+        "std": 0.010883544154216303
+      },
+      "val_right_hand_mae": {
+        "mean": 0.0689016379415989,
+        "std": 0.00747019507090763
+      },
+      "val_right_hand_r2": {
+        "mean": 0.7077235102653503,
+        "std": 0.06452178715359633
+      },
+      "val_left_hip_mse": {
+        "mean": 0.0038185957819223404,
+        "std": 0.0008673056697701263
+      },
+      "val_left_hip_rmse": {
+        "mean": 0.06136601480322237,
+        "std": 0.00726691193651297
+      },
+      "val_left_hip_mae": {
+        "mean": 0.04575123265385628,
+        "std": 0.005729932539817455
+      },
+      "val_left_hip_r2": {
+        "mean": 0.5122388005256653,
+        "std": 0.09507911405578336
+      },
+      "val_right_hip_mse": {
+        "mean": 0.003961405111476779,
+        "std": 0.0009751191034544559
+      },
+      "val_right_hip_rmse": {
+        "mean": 0.062425669385723526,
+        "std": 0.00802750977708025
+      },
+      "val_right_hip_mae": {
+        "mean": 0.04628140702843666,
+        "std": 0.00637386081166704
+      },
+      "val_right_hip_r2": {
+        "mean": 0.5147492170333863,
+        "std": 0.09989324855613299
+      },
+      "val_left_knee_mse": {
+        "mean": 0.0036820633336901664,
+        "std": 0.0007623879166605281
+      },
+      "val_left_knee_rmse": {
+        "mean": 0.060345732661520726,
+        "std": 0.006360493945790588
+      },
+      "val_left_knee_mae": {
+        "mean": 0.044499056041240694,
+        "std": 0.0035912897930051786
+      },
+      "val_left_knee_r2": {
+        "mean": 0.6121258735656738,
+        "std": 0.07350370231964652
+      },
+      "val_right_knee_mse": {
+        "mean": 0.003761627431958914,
+        "std": 0.00025590417754942497
+      },
+      "val_right_knee_rmse": {
+        "mean": 0.06129672870608678,
+        "std": 0.002082902275986774
+      },
+      "val_right_knee_mae": {
+        "mean": 0.04525262713432312,
+        "std": 0.002069468595645557
+      },
+      "val_right_knee_r2": {
+        "mean": 0.649232280254364,
+        "std": 0.029181963743745005
+      },
+      "val_left_foot_mse": {
+        "mean": 0.003976805601269007,
+        "std": 0.0006406339939907319
+      },
+      "val_left_foot_rmse": {
+        "mean": 0.06283428603605966,
+        "std": 0.0053533260322574374
+      },
+      "val_left_foot_mae": {
+        "mean": 0.04497457668185234,
+        "std": 0.0028319605969746586
+      },
+      "val_left_foot_r2": {
+        "mean": 0.09387181997299195,
+        "std": 0.049758276927747924
+      },
+      "val_right_foot_mse": {
+        "mean": 0.004241668153554201,
+        "std": 0.00023221735671322577
+      },
+      "val_right_foot_rmse": {
+        "mean": 0.06510346364771302,
+        "std": 0.001790858627896126
+      },
+      "val_right_foot_mae": {
+        "mean": 0.04678419381380081,
+        "std": 0.0011237403694175805
+      },
+      "val_right_foot_r2": {
+        "mean": 0.0530170202255249,
+        "std": 0.016706412276843975
+      }
+    },
+    "best_fold": "5",
+    "best_fold_metrics": {
+      "fold": 5,
+      "train_metrics": {
+        "train_mse": 0.0033102582674473524,
+        "train_rmse": 0.057534843942148244,
+        "train_mae": 0.042325250804424286,
+        "train_r2": 0.8441787362098694,
+        "train_head_mse": 0.0024612394627183676,
+        "train_head_rmse": 0.0496108804872315,
+        "train_head_mae": 0.03801852837204933,
+        "train_head_r2": 0.7556431293487549,
+        "train_left_shoulder_mse": 0.0015651942230761051,
+        "train_left_shoulder_rmse": 0.039562535599681994,
+        "train_left_shoulder_mae": 0.030392082408070564,
+        "train_left_shoulder_r2": 0.7193259000778198,
+        "train_left_elbow_mse": 0.0030404275748878717,
+        "train_left_elbow_rmse": 0.05514007231485893,
+        "train_left_elbow_mae": 0.04346305504441261,
+        "train_left_elbow_r2": 0.7805148363113403,
+        "train_right_shoulder_mse": 0.0019232443301007152,
+        "train_right_shoulder_rmse": 0.04385480965755883,
+        "train_right_shoulder_mae": 0.0336100235581398,
+        "train_right_shoulder_r2": 0.7091915607452393,
+        "train_right_elbow_mse": 0.00364261819049716,
+        "train_right_elbow_rmse": 0.060354106658098754,
+        "train_right_elbow_mae": 0.04654518887400627,
+        "train_right_elbow_r2": 0.7725162506103516,
+        "train_left_hand_mse": 0.005895042791962624,
+        "train_left_hand_rmse": 0.07677918202196884,
+        "train_left_hand_mae": 0.06003705412149429,
+        "train_left_hand_r2": 0.7775342464447021,
+        "train_right_hand_mse": 0.006368337664753199,
+        "train_right_hand_rmse": 0.07980186504558148,
+        "train_right_hand_mae": 0.06009024381637573,
+        "train_right_hand_r2": 0.7747085094451904,
+        "train_left_hip_mse": 0.002523666014894843,
+        "train_left_hip_rmse": 0.050236102704079695,
+        "train_left_hip_mae": 0.0361558236181736,
+        "train_left_hip_r2": 0.6772412657737732,
+        "train_right_hip_mse": 0.002538986038416624,
+        "train_right_hip_rmse": 0.050388352209777845,
+        "train_right_hip_mae": 0.03592711687088013,
+        "train_right_hip_r2": 0.6892550587654114,
+        "train_left_knee_mse": 0.0026853561867028475,
+        "train_left_knee_rmse": 0.05182042248672667,
+        "train_left_knee_mae": 0.038079895079135895,
+        "train_left_knee_r2": 0.7171890735626221,
+        "train_right_knee_mse": 0.0026471319142729044,
+        "train_right_knee_rmse": 0.05145028585219818,
+        "train_right_knee_mae": 0.03821215778589249,
+        "train_right_knee_r2": 0.7498998641967773,
+        "train_left_foot_mse": 0.0038137855008244514,
+        "train_left_foot_rmse": 0.0617558539802054,
+        "train_left_foot_mae": 0.04399872571229935,
+        "train_left_foot_r2": 0.17480021715164185,
+        "train_right_foot_mse": 0.003928327467292547,
+        "train_right_foot_rmse": 0.06267637088482826,
+        "train_right_foot_mae": 0.045698340982198715,
+        "train_right_foot_r2": 0.1098211407661438
+      },
+      "val_metrics": {
+        "val_mse": 0.00329865631647408,
+        "val_rmse": 0.057433930010700816,
+        "val_mae": 0.04246829077601433,
+        "val_r2": 0.8408486843109131,
+        "val_head_mse": 0.0024432188365608454,
+        "val_head_rmse": 0.04942892712330347,
+        "val_head_mae": 0.03793890029191971,
+        "val_head_r2": 0.7416551113128662,
+        "val_left_shoulder_mse": 0.001684311544522643,
+        "val_left_shoulder_rmse": 0.04104036481956079,
+        "val_left_shoulder_mae": 0.030944740399718285,
+        "val_left_shoulder_r2": 0.6932499408721924,
+        "val_left_elbow_mse": 0.003190949559211731,
+        "val_left_elbow_rmse": 0.05648849050215213,
+        "val_left_elbow_mae": 0.04436313360929489,
+        "val_left_elbow_r2": 0.7658652663230896,
+        "val_right_shoulder_mse": 0.0018238489283248782,
+        "val_right_shoulder_rmse": 0.04270654432665886,
+        "val_right_shoulder_mae": 0.032911598682403564,
+        "val_right_shoulder_r2": 0.7042507529258728,
+        "val_right_elbow_mse": 0.003342842450365424,
+        "val_right_elbow_rmse": 0.0578173196400994,
+        "val_right_elbow_mae": 0.045683834701776505,
+        "val_right_elbow_r2": 0.7745484113693237,
+        "val_left_hand_mse": 0.006071725860238075,
+        "val_left_hand_rmse": 0.07792127989348016,
+        "val_left_hand_mae": 0.06114818900823593,
+        "val_left_hand_r2": 0.7614104151725769,
+        "val_right_hand_mse": 0.006216216366738081,
+        "val_right_hand_rmse": 0.07884298552653927,
+        "val_right_hand_mae": 0.061056043952703476,
+        "val_right_hand_r2": 0.7627074122428894,
+        "val_left_hip_mse": 0.002505603479221463,
+        "val_left_hip_rmse": 0.05005600342837473,
+        "val_left_hip_mae": 0.03692784160375595,
+        "val_left_hip_r2": 0.6750618815422058,
+        "val_right_hip_mse": 0.0024700025096535683,
+        "val_right_hip_rmse": 0.0496991198076341,
+        "val_right_hip_mae": 0.03634550794959068,
+        "val_right_hip_r2": 0.6870356202125549,
+        "val_left_knee_mse": 0.0025547617115080357,
+        "val_left_knee_rmse": 0.050544650671540264,
+        "val_left_knee_mae": 0.0389634408056736,
+        "val_left_knee_r2": 0.7297466993331909,
+        "val_right_knee_mse": 0.003407940734177828,
+        "val_right_knee_rmse": 0.05837757047169596,
+        "val_right_knee_mae": 0.041529685258865356,
+        "val_right_knee_r2": 0.7023165225982666,
+        "val_left_foot_mse": 0.0027765268459916115,
+        "val_left_foot_rmse": 0.05269275895217114,
+        "val_left_foot_mae": 0.03969832509756088,
+        "val_left_foot_r2": 0.17813962697982788,
+        "val_right_foot_mse": 0.004394582472741604,
+        "val_right_foot_rmse": 0.06629164708122437,
+        "val_right_foot_mae": 0.044576507061719894,
+        "val_right_foot_r2": 0.07590389251708984
+      },
+      "best_epoch": 87
+    }
+  },
+  "fold_details": [
+    {
+      "fold": 1,
+      "train_metrics": {
+        "train_mse": 0.0036423802375793457,
+        "train_rmse": 0.06035213531913635,
+        "train_mae": 0.04459811747074127,
+        "train_r2": 0.8288930654525757,
+        "train_head_mse": 0.0024861530400812626,
+        "train_head_rmse": 0.049861338129669794,
+        "train_head_mae": 0.03838302567601204,
+        "train_head_r2": 0.7537641525268555,
+        "train_left_shoulder_mse": 0.0016836737049743533,
+        "train_left_shoulder_rmse": 0.04103259320313978,
+        "train_left_shoulder_mae": 0.03163576498627663,
+        "train_left_shoulder_r2": 0.7019709348678589,
+        "train_left_elbow_mse": 0.0029813426081091166,
+        "train_left_elbow_rmse": 0.05460167220982446,
+        "train_left_elbow_mae": 0.04266366735100746,
+        "train_left_elbow_r2": 0.7839894890785217,
+        "train_right_shoulder_mse": 0.0020383817609399557,
+        "train_right_shoulder_rmse": 0.045148441401004706,
+        "train_right_shoulder_mae": 0.034506529569625854,
+        "train_right_shoulder_r2": 0.6915265917778015,
+        "train_right_elbow_mse": 0.0034374285023659468,
+        "train_right_elbow_rmse": 0.05862958726075041,
+        "train_right_elbow_mae": 0.044821929186582565,
+        "train_right_elbow_r2": 0.7840657234191895,
+        "train_left_hand_mse": 0.005827914457768202,
+        "train_left_hand_rmse": 0.07634077847237479,
+        "train_left_hand_mae": 0.059969719499349594,
+        "train_left_hand_r2": 0.776339590549469,
+        "train_right_hand_mse": 0.006132215261459351,
+        "train_right_hand_rmse": 0.07830846225957544,
+        "train_right_hand_mae": 0.059468600898981094,
+        "train_right_hand_r2": 0.780566930770874,
+        "train_left_hip_mse": 0.0037651248276233673,
+        "train_left_hip_rmse": 0.06136061299908409,
+        "train_left_hip_mae": 0.04432861879467964,
+        "train_left_hip_r2": 0.5290084481239319,
+        "train_right_hip_mse": 0.0038999945390969515,
+        "train_right_hip_rmse": 0.06244993626175251,
+        "train_right_hip_mae": 0.04483594372868538,
+        "train_right_hip_r2": 0.5300575494766235,
+        "train_left_knee_mse": 0.003411693498492241,
+        "train_left_knee_rmse": 0.05840970380418172,
+        "train_left_knee_mae": 0.04366626217961311,
+        "train_left_knee_r2": 0.6431400775909424,
+        "train_right_knee_mse": 0.003540073987096548,
+        "train_right_knee_rmse": 0.05949852088158619,
+        "train_right_knee_mae": 0.04441169276833534,
+        "train_right_knee_r2": 0.6708280444145203,
+        "train_left_foot_mse": 0.003982864785939455,
+        "train_left_foot_rmse": 0.063109942053051,
+        "train_left_foot_mae": 0.04448368400335312,
+        "train_left_foot_r2": 0.09590977430343628,
+        "train_right_foot_mse": 0.004164081532508135,
+        "train_right_foot_rmse": 0.06452969496679908,
+        "train_right_foot_mae": 0.046600084751844406,
+        "train_right_foot_r2": 0.0619128942489624
+      },
+      "val_metrics": {
+        "val_mse": 0.0035069233272224665,
+        "val_rmse": 0.059219281718224735,
+        "val_mae": 0.04442489892244339,
+        "val_r2": 0.8292991518974304,
+        "val_head_mse": 0.0023568107280880213,
+        "val_head_rmse": 0.048546995046944166,
+        "val_head_mae": 0.03762654587626457,
+        "val_head_r2": 0.7474652528762817,
+        "val_left_shoulder_mse": 0.0015835677040740848,
+        "val_left_shoulder_rmse": 0.03979406619175885,
+        "val_left_shoulder_mae": 0.031601954251527786,
+        "val_left_shoulder_r2": 0.6945991516113281,
+        "val_left_elbow_mse": 0.003050273284316063,
+        "val_left_elbow_rmse": 0.05522927923045948,
+        "val_left_elbow_mae": 0.04390987753868103,
+        "val_left_elbow_r2": 0.7789299488067627,
+        "val_right_shoulder_mse": 0.0020674807019531727,
+        "val_right_shoulder_rmse": 0.04546955796962593,
+        "val_right_shoulder_mae": 0.03509090840816498,
+        "val_right_shoulder_r2": 0.6654824018478394,
+        "val_right_elbow_mse": 0.003225933760404587,
+        "val_right_elbow_rmse": 0.05679730416493891,
+        "val_right_elbow_mae": 0.04449897259473801,
+        "val_right_elbow_r2": 0.7876308560371399,
+        "val_left_hand_mse": 0.006042634602636099,
+        "val_left_hand_rmse": 0.07773438494409085,
+        "val_left_hand_mae": 0.06162887439131737,
+        "val_left_hand_r2": 0.777449369430542,
+        "val_right_hand_mse": 0.0055483365431427956,
+        "val_right_hand_rmse": 0.074487156900655,
+        "val_right_hand_mae": 0.05878162384033203,
+        "val_right_hand_r2": 0.797961950302124,
+        "val_left_hip_mse": 0.0031134297605603933,
+        "val_left_hip_rmse": 0.05579811610225199,
+        "val_left_hip_mae": 0.04164275899529457,
+        "val_left_hip_r2": 0.5559567213058472,
+        "val_right_hip_mse": 0.0032598816324025393,
+        "val_right_hip_rmse": 0.05709537312604708,
+        "val_right_hip_mae": 0.04216572642326355,
+        "val_right_hip_r2": 0.5582848787307739,
+        "val_left_knee_mse": 0.0032232438679784536,
+        "val_left_knee_rmse": 0.05677361947223775,
+        "val_left_knee_mae": 0.04238000512123108,
+        "val_left_knee_r2": 0.6490132808685303,
+        "val_right_knee_mse": 0.0036669797264039516,
+        "val_right_knee_rmse": 0.06055559203247832,
+        "val_right_knee_mae": 0.04541962593793869,
+        "val_right_knee_r2": 0.6589800119400024,
+        "val_left_foot_mse": 0.003923279233276844,
+        "val_left_foot_rmse": 0.06263608571164743,
+        "val_left_foot_mae": 0.04505937919020653,
+        "val_left_foot_r2": 0.07600140571594238,
+        "val_right_foot_mse": 0.004528151359409094,
+        "val_right_foot_rmse": 0.06729154002851394,
+        "val_right_foot_mae": 0.047717414796352386,
+        "val_right_foot_r2": 0.029663562774658203
+      },
+      "best_epoch": 86
+    },
+    {
+      "fold": 2,
+      "train_metrics": {
+        "train_mse": 0.004433092195540667,
+        "train_rmse": 0.06658147036181063,
+        "train_mae": 0.04979822412133217,
+        "train_r2": 0.7886409759521484,
+        "train_head_mse": 0.0031831867527216673,
+        "train_head_rmse": 0.056419737262075824,
+        "train_head_mae": 0.04461611434817314,
+        "train_head_r2": 0.6744217872619629,
+        "train_left_shoulder_mse": 0.002108559478074312,
+        "train_left_shoulder_rmse": 0.04591905354070696,
+        "train_left_shoulder_mae": 0.036486852914094925,
+        "train_left_shoulder_r2": 0.6188086271286011,
+        "train_left_elbow_mse": 0.004112447611987591,
+        "train_left_elbow_rmse": 0.06412836823113147,
+        "train_left_elbow_mae": 0.05035443231463432,
+        "train_left_elbow_r2": 0.7041769027709961,
+        "train_right_shoulder_mse": 0.002525464165955782,
+        "train_right_shoulder_rmse": 0.050253996517250064,
+        "train_right_shoulder_mae": 0.04043629392981529,
+        "train_right_shoulder_r2": 0.6064876317977905,
+        "train_right_elbow_mse": 0.004740843083709478,
+        "train_right_elbow_rmse": 0.06885378046054899,
+        "train_right_elbow_mae": 0.053702570497989655,
+        "train_right_elbow_r2": 0.6970694065093994,
+        "train_left_hand_mse": 0.007980061694979668,
+        "train_left_hand_rmse": 0.08933119105317956,
+        "train_left_hand_mae": 0.06992378830909729,
+        "train_left_hand_r2": 0.6978632211685181,
+        "train_right_hand_mse": 0.008613844402134418,
+        "train_right_hand_rmse": 0.09281079895213928,
+        "train_right_hand_mae": 0.07129455357789993,
+        "train_right_hand_r2": 0.6891484260559082,
+        "train_left_hip_mse": 0.004002328030765057,
+        "train_left_hip_rmse": 0.06326395522542877,
+        "train_left_hip_mae": 0.047635119408369064,
+        "train_left_hip_r2": 0.47017502784729004,
+        "train_right_hip_mse": 0.004133956041187048,
+        "train_right_hip_rmse": 0.0642958477756305,
+        "train_right_hip_mae": 0.048478662967681885,
+        "train_right_hip_r2": 0.47101539373397827,
+        "train_left_knee_mse": 0.003696210216730833,
+        "train_left_knee_rmse": 0.0607964654953792,
+        "train_left_knee_mae": 0.04527777433395386,
+        "train_left_knee_r2": 0.6022827625274658,
+        "train_right_knee_mse": 0.004095621407032013,
+        "train_right_knee_rmse": 0.06399704217408811,
+        "train_right_knee_mae": 0.046711549162864685,
+        "train_right_knee_r2": 0.619659960269928,
+        "train_left_foot_mse": 0.004021741449832916,
+        "train_left_foot_rmse": 0.06341720152949763,
+        "train_left_foot_mae": 0.04518086463212967,
+        "train_left_foot_r2": 0.06873959302902222,
+        "train_right_foot_mse": 0.004415937699377537,
+        "train_right_foot_rmse": 0.06645252214459235,
+        "train_right_foot_mae": 0.04727834090590477,
+        "train_right_foot_r2": 0.03330940008163452
+      },
+      "val_metrics": {
+        "val_mse": 0.0048818266950547695,
+        "val_rmse": 0.06987007009481792,
+        "val_mae": 0.05211581289768219,
+        "val_r2": 0.7758788466453552,
+        "val_head_mse": 0.0037353525403887033,
+        "val_head_rmse": 0.06111753054884256,
+        "val_head_mae": 0.048024579882621765,
+        "val_head_r2": 0.6482267379760742,
+        "val_left_shoulder_mse": 0.0022262188140302896,
+        "val_left_shoulder_rmse": 0.04718282329439697,
+        "val_left_shoulder_mae": 0.03749328851699829,
+        "val_left_shoulder_r2": 0.6061323881149292,
+        "val_left_elbow_mse": 0.0040644253604114056,
+        "val_left_elbow_rmse": 0.06375284590048828,
+        "val_left_elbow_mae": 0.05059381574392319,
+        "val_left_elbow_r2": 0.6964505314826965,
+        "val_right_shoulder_mse": 0.002972251735627651,
+        "val_right_shoulder_rmse": 0.054518361453987696,
+        "val_right_shoulder_mae": 0.04303857684135437,
+        "val_right_shoulder_r2": 0.5711896419525146,
+        "val_right_elbow_mse": 0.00534110376611352,
+        "val_right_elbow_rmse": 0.07308285548686176,
+        "val_right_elbow_mae": 0.05661436915397644,
+        "val_right_elbow_r2": 0.6708359718322754,
+        "val_left_hand_mse": 0.008427683264017105,
+        "val_left_hand_rmse": 0.0918024142602857,
+        "val_left_hand_mae": 0.07193999737501144,
+        "val_left_hand_r2": 0.6722412109375,
+        "val_right_hand_mse": 0.00971507839858532,
+        "val_right_hand_rmse": 0.09856509726361214,
+        "val_right_hand_mae": 0.07412885129451752,
+        "val_right_hand_r2": 0.6571488380432129,
+        "val_left_hip_mse": 0.004789731930941343,
+        "val_left_hip_rmse": 0.06920788922472165,
+        "val_left_hip_mae": 0.05116049572825432,
+        "val_left_hip_r2": 0.45315438508987427,
+        "val_right_hip_mse": 0.005170117598026991,
+        "val_right_hip_rmse": 0.07190352980227738,
+        "val_right_hip_mae": 0.052645161747932434,
+        "val_right_hip_r2": 0.44426512718200684,
+        "val_left_knee_mse": 0.00482585933059454,
+        "val_left_knee_rmse": 0.06946840526883095,
+        "val_left_knee_mae": 0.04960314929485321,
+        "val_left_knee_r2": 0.5281065702438354,
+        "val_right_knee_mse": 0.003956494387239218,
+        "val_right_knee_rmse": 0.06290067080118636,
+        "val_right_knee_mae": 0.047786563634872437,
+        "val_right_knee_r2": 0.6302878856658936,
+        "val_left_foot_mse": 0.004332127049565315,
+        "val_left_foot_rmse": 0.06581889583976105,
+        "val_left_foot_mae": 0.047359779477119446,
+        "val_left_foot_r2": 0.054024577140808105,
+        "val_right_foot_mse": 0.003907305654138327,
+        "val_right_foot_rmse": 0.062508444662608,
+        "val_right_foot_mae": 0.04711692035198212,
+        "val_right_foot_r2": 0.05739474296569824
+      },
+      "best_epoch": 47
+    },
+    {
+      "fold": 3,
+      "train_metrics": {
+        "train_mse": 0.004286908078938723,
+        "train_rmse": 0.06547448418230359,
+        "train_mae": 0.04881489276885986,
+        "train_r2": 0.7949373126029968,
+        "train_head_mse": 0.0031648229341953993,
+        "train_head_rmse": 0.05625675900898842,
+        "train_head_mae": 0.0441947802901268,
+        "train_head_r2": 0.6783979535102844,
+        "train_left_shoulder_mse": 0.002001953311264515,
+        "train_left_shoulder_rmse": 0.04474319290422304,
+        "train_left_shoulder_mae": 0.03499903529882431,
+        "train_left_shoulder_r2": 0.6326049566268921,
+        "train_left_elbow_mse": 0.004002084489911795,
+        "train_left_elbow_rmse": 0.06326203039669051,
+        "train_left_elbow_mae": 0.04980403557419777,
+        "train_left_elbow_r2": 0.7073379755020142,
+        "train_right_shoulder_mse": 0.0024581796023994684,
+        "train_right_shoulder_rmse": 0.049580032295264476,
+        "train_right_shoulder_mae": 0.03864126652479172,
+        "train_right_shoulder_r2": 0.6180688142776489,
+        "train_right_elbow_mse": 0.004426611587405205,
+        "train_right_elbow_rmse": 0.06653278580824047,
+        "train_right_elbow_mae": 0.0520782470703125,
+        "train_right_elbow_r2": 0.7150493860244751,
+        "train_left_hand_mse": 0.007924766279757023,
+        "train_left_hand_rmse": 0.08902115636047997,
+        "train_left_hand_mae": 0.0700734555721283,
+        "train_left_hand_r2": 0.6955196857452393,
+        "train_right_hand_mse": 0.008269179612398148,
+        "train_right_hand_rmse": 0.09093502962224265,
+        "train_right_hand_mae": 0.0698436051607132,
+        "train_right_hand_r2": 0.6979034543037415,
+        "train_left_hip_mse": 0.0036954055540263653,
+        "train_left_hip_rmse": 0.06078984745848903,
+        "train_left_hip_mae": 0.04506364464759827,
+        "train_left_hip_r2": 0.5237404108047485,
+        "train_right_hip_mse": 0.0038784611970186234,
+        "train_right_hip_rmse": 0.06227729278813124,
+        "train_right_hip_mae": 0.045497797429561615,
+        "train_right_hip_r2": 0.52162766456604,
+        "train_left_knee_mse": 0.003810907481238246,
+        "train_left_knee_rmse": 0.06173254798919486,
+        "train_left_knee_mae": 0.04609154537320137,
+        "train_left_knee_r2": 0.5952310562133789,
+        "train_right_knee_mse": 0.004034105222672224,
+        "train_right_knee_rmse": 0.06351460637264647,
+        "train_right_knee_mae": 0.047519076615571976,
+        "train_right_knee_r2": 0.6228135824203491,
+        "train_left_foot_mse": 0.003833775408565998,
+        "train_left_foot_rmse": 0.06191748871333525,
+        "train_left_foot_mae": 0.04423471912741661,
+        "train_left_foot_r2": 0.10032761096954346,
+        "train_right_foot_mse": 0.00422955397516489,
+        "train_right_foot_rmse": 0.06503502114372602,
+        "train_right_foot_mae": 0.0465523786842823,
+        "train_right_foot_r2": 0.048550307750701904
+      },
+      "val_metrics": {
+        "val_mse": 0.004516200162470341,
+        "val_rmse": 0.06720267972685569,
+        "val_mae": 0.0496145524084568,
+        "val_r2": 0.7954664826393127,
+        "val_head_mse": 0.003185930661857128,
+        "val_head_rmse": 0.056444048949886014,
+        "val_head_mae": 0.0444367378950119,
+        "val_head_r2": 0.6931617259979248,
+        "val_left_shoulder_mse": 0.0022065129596740007,
+        "val_left_shoulder_rmse": 0.04697353467298369,
+        "val_left_shoulder_mae": 0.035925790667533875,
+        "val_left_shoulder_r2": 0.6322484016418457,
+        "val_left_elbow_mse": 0.004166789818555117,
+        "val_left_elbow_rmse": 0.06455067636016773,
+        "val_left_elbow_mae": 0.05036281421780586,
+        "val_left_elbow_r2": 0.7091368436813354,
+        "val_right_shoulder_mse": 0.0023906466085463762,
+        "val_right_shoulder_rmse": 0.04889423901183427,
+        "val_right_shoulder_mae": 0.03742371127009392,
+        "val_right_shoulder_r2": 0.6523372530937195,
+        "val_right_elbow_mse": 0.004597921390086412,
+        "val_right_elbow_rmse": 0.06780797438418591,
+        "val_right_elbow_mae": 0.05325713008642197,
+        "val_right_elbow_r2": 0.7252645492553711,
+        "val_left_hand_mse": 0.008194615133106709,
+        "val_left_hand_rmse": 0.09052411354499258,
+        "val_left_hand_mae": 0.07138348370790482,
+        "val_left_hand_r2": 0.6998146772384644,
+        "val_right_hand_mse": 0.009030655026435852,
+        "val_right_hand_rmse": 0.09502975863610226,
+        "val_right_hand_mae": 0.0734550803899765,
+        "val_right_hand_r2": 0.6965397596359253,
+        "val_left_hip_mse": 0.004186134785413742,
+        "val_left_hip_rmse": 0.06470034609964419,
+        "val_left_hip_mae": 0.047075942158699036,
+        "val_left_hip_r2": 0.4732587933540344,
+        "val_right_hip_mse": 0.00423840107396245,
+        "val_right_hip_rmse": 0.06510300357097551,
+        "val_right_hip_mae": 0.04717397317290306,
+        "val_right_hip_r2": 0.4794016480445862,
+        "val_left_knee_mse": 0.0038063821848481894,
+        "val_left_knee_rmse": 0.06169588466703585,
+        "val_left_knee_mae": 0.04561794549226761,
+        "val_left_knee_r2": 0.6106085777282715,
+        "val_right_knee_mse": 0.004136641509830952,
+        "val_right_knee_rmse": 0.06431672807155966,
+        "val_right_knee_mae": 0.04630318284034729,
+        "val_right_knee_r2": 0.6238951086997986,
+        "val_left_foot_mse": 0.004227760247886181,
+        "val_left_foot_rmse": 0.0650212292092835,
+        "val_left_foot_mae": 0.04523155838251114,
+        "val_left_foot_r2": 0.11944746971130371,
+        "val_right_foot_mse": 0.004342210479080677,
+        "val_right_foot_rmse": 0.06589545112586055,
+        "val_right_foot_mae": 0.04734177142381668,
+        "val_right_foot_r2": 0.06329900026321411
+      },
+      "best_epoch": 39
+    },
+    {
+      "fold": 4,
+      "train_metrics": {
+        "train_mse": 0.005276767071336508,
+        "train_rmse": 0.0726413592338174,
+        "train_mae": 0.054584719240665436,
+        "train_r2": 0.7521504163742065,
+        "train_head_mse": 0.0038697586860507727,
+        "train_head_rmse": 0.06220738449774892,
+        "train_head_mae": 0.047990716993808746,
+        "train_head_r2": 0.6114107370376587,
+        "train_left_shoulder_mse": 0.0024136616848409176,
+        "train_left_shoulder_rmse": 0.04912903097803698,
+        "train_left_shoulder_mae": 0.0381099134683609,
+        "train_left_shoulder_r2": 0.5681445598602295,
+        "train_left_elbow_mse": 0.0050811260007321835,
+        "train_left_elbow_rmse": 0.0712820173727721,
+        "train_left_elbow_mae": 0.056389451026916504,
+        "train_left_elbow_r2": 0.6318797469139099,
+        "train_right_shoulder_mse": 0.0028996027540415525,
+        "train_right_shoulder_rmse": 0.05384795960889839,
+        "train_right_shoulder_mae": 0.04165012016892433,
+        "train_right_shoulder_r2": 0.5569307804107666,
+        "train_right_elbow_mse": 0.005731700919568539,
+        "train_right_elbow_rmse": 0.07570799772526374,
+        "train_right_elbow_mae": 0.060447853058576584,
+        "train_right_elbow_r2": 0.6362661123275757,
+        "train_left_hand_mse": 0.010002972558140755,
+        "train_left_hand_rmse": 0.10001486168635516,
+        "train_left_hand_mae": 0.07863272726535797,
+        "train_left_hand_r2": 0.6216628551483154,
+        "train_right_hand_mse": 0.01066053006798029,
+        "train_right_hand_rmse": 0.10324984294409503,
+        "train_right_hand_mae": 0.0809774175286293,
+        "train_right_hand_r2": 0.6187346577644348,
+        "train_left_hip_mse": 0.004943680949509144,
+        "train_left_hip_rmse": 0.07031131451984911,
+        "train_left_hip_mae": 0.054632000625133514,
+        "train_left_hip_r2": 0.37105119228363037,
+        "train_right_hip_mse": 0.0051644002087414265,
+        "train_right_hip_rmse": 0.07186376144303488,
+        "train_right_hip_mae": 0.05566652864217758,
+        "train_right_hip_r2": 0.36884605884552,
+        "train_left_knee_mse": 0.004715795163065195,
+        "train_left_knee_rmse": 0.06867164744685535,
+        "train_left_knee_mae": 0.05011150613427162,
+        "train_left_knee_r2": 0.5122848749160767,
+        "train_right_knee_mse": 0.004712671507149935,
+        "train_right_knee_rmse": 0.06864890026176629,
+        "train_right_knee_mae": 0.05131387338042259,
+        "train_right_knee_r2": 0.5708245038986206,
+        "train_left_foot_mse": 0.004023321438580751,
+        "train_left_foot_rmse": 0.06342965740551301,
+        "train_left_foot_mae": 0.04625389724969864,
+        "train_left_foot_r2": 0.05525994300842285,
+        "train_right_foot_mse": 0.00437875185161829,
+        "train_right_foot_rmse": 0.06617213803118568,
+        "train_right_foot_mae": 0.04742538556456566,
+        "train_right_foot_r2": 0.0388413667678833
+      },
+      "val_metrics": {
+        "val_mse": 0.004960071761161089,
+        "val_rmse": 0.0704277769148018,
+        "val_mae": 0.052099451422691345,
+        "val_r2": 0.7584605813026428,
+        "val_head_mse": 0.0035925537813454866,
+        "val_head_rmse": 0.059937916057746676,
+        "val_head_mae": 0.04611942917108536,
+        "val_head_r2": 0.6375719904899597,
+        "val_left_shoulder_mse": 0.0021630162373185158,
+        "val_left_shoulder_rmse": 0.0465082383811569,
+        "val_left_shoulder_mae": 0.03607545420527458,
+        "val_left_shoulder_r2": 0.6021186709403992,
+        "val_left_elbow_mse": 0.004990265239030123,
+        "val_left_elbow_rmse": 0.0706418094263597,
+        "val_left_elbow_mae": 0.05458231270313263,
+        "val_left_elbow_r2": 0.6386438012123108,
+        "val_right_shoulder_mse": 0.0026175302919000387,
+        "val_right_shoulder_rmse": 0.05116180501018351,
+        "val_right_shoulder_mae": 0.038790952414274216,
+        "val_right_shoulder_r2": 0.5935391187667847,
+        "val_right_elbow_mse": 0.005619417876005173,
+        "val_right_elbow_rmse": 0.07496277660282584,
+        "val_right_elbow_mae": 0.057268332690000534,
+        "val_right_elbow_r2": 0.6446224451065063,
+        "val_left_hand_mse": 0.009756939485669136,
+        "val_left_hand_rmse": 0.0987772214919469,
+        "val_left_hand_mae": 0.07649697363376617,
+        "val_left_hand_r2": 0.6196279525756836,
+        "val_right_hand_mse": 0.010273933410644531,
+        "val_right_hand_rmse": 0.1013604134297238,
+        "val_right_hand_mae": 0.07708659023046494,
+        "val_right_hand_r2": 0.6242595911026001,
+        "val_left_hip_mse": 0.00449807895347476,
+        "val_left_hip_rmse": 0.06706771916111924,
+        "val_left_hip_mae": 0.05194912478327751,
+        "val_left_hip_r2": 0.40376222133636475,
+        "val_right_hip_mse": 0.0046686227433383465,
+        "val_right_hip_rmse": 0.06832732062168358,
+        "val_right_hip_mae": 0.053076665848493576,
+        "val_right_hip_r2": 0.4047588109970093,
+        "val_left_knee_mse": 0.004000069573521614,
+        "val_left_knee_rmse": 0.06324610322795875,
+        "val_left_knee_mae": 0.04593073949217796,
+        "val_left_knee_r2": 0.543154239654541,
+        "val_right_knee_mse": 0.00364008080214262,
+        "val_right_knee_rmse": 0.06033308215351359,
+        "val_right_knee_mae": 0.04522407799959183,
+        "val_right_knee_r2": 0.6306818723678589,
+        "val_left_foot_mse": 0.004624334629625082,
+        "val_left_foot_rmse": 0.06800246046743516,
+        "val_left_foot_mae": 0.04752384126186371,
+        "val_left_foot_r2": 0.04174602031707764,
+        "val_right_foot_mse": 0.004036090802401304,
+        "val_right_foot_rmse": 0.06353023534035825,
+        "val_right_foot_mae": 0.04716835543513298,
+        "val_right_foot_r2": 0.03882390260696411
+      },
+      "best_epoch": 37
+    },
+    {
+      "fold": 5,
+      "train_metrics": {
+        "train_mse": 0.0033102582674473524,
+        "train_rmse": 0.057534843942148244,
+        "train_mae": 0.042325250804424286,
+        "train_r2": 0.8441787362098694,
+        "train_head_mse": 0.0024612394627183676,
+        "train_head_rmse": 0.0496108804872315,
+        "train_head_mae": 0.03801852837204933,
+        "train_head_r2": 0.7556431293487549,
+        "train_left_shoulder_mse": 0.0015651942230761051,
+        "train_left_shoulder_rmse": 0.039562535599681994,
+        "train_left_shoulder_mae": 0.030392082408070564,
+        "train_left_shoulder_r2": 0.7193259000778198,
+        "train_left_elbow_mse": 0.0030404275748878717,
+        "train_left_elbow_rmse": 0.05514007231485893,
+        "train_left_elbow_mae": 0.04346305504441261,
+        "train_left_elbow_r2": 0.7805148363113403,
+        "train_right_shoulder_mse": 0.0019232443301007152,
+        "train_right_shoulder_rmse": 0.04385480965755883,
+        "train_right_shoulder_mae": 0.0336100235581398,
+        "train_right_shoulder_r2": 0.7091915607452393,
+        "train_right_elbow_mse": 0.00364261819049716,
+        "train_right_elbow_rmse": 0.060354106658098754,
+        "train_right_elbow_mae": 0.04654518887400627,
+        "train_right_elbow_r2": 0.7725162506103516,
+        "train_left_hand_mse": 0.005895042791962624,
+        "train_left_hand_rmse": 0.07677918202196884,
+        "train_left_hand_mae": 0.06003705412149429,
+        "train_left_hand_r2": 0.7775342464447021,
+        "train_right_hand_mse": 0.006368337664753199,
+        "train_right_hand_rmse": 0.07980186504558148,
+        "train_right_hand_mae": 0.06009024381637573,
+        "train_right_hand_r2": 0.7747085094451904,
+        "train_left_hip_mse": 0.002523666014894843,
+        "train_left_hip_rmse": 0.050236102704079695,
+        "train_left_hip_mae": 0.0361558236181736,
+        "train_left_hip_r2": 0.6772412657737732,
+        "train_right_hip_mse": 0.002538986038416624,
+        "train_right_hip_rmse": 0.050388352209777845,
+        "train_right_hip_mae": 0.03592711687088013,
+        "train_right_hip_r2": 0.6892550587654114,
+        "train_left_knee_mse": 0.0026853561867028475,
+        "train_left_knee_rmse": 0.05182042248672667,
+        "train_left_knee_mae": 0.038079895079135895,
+        "train_left_knee_r2": 0.7171890735626221,
+        "train_right_knee_mse": 0.0026471319142729044,
+        "train_right_knee_rmse": 0.05145028585219818,
+        "train_right_knee_mae": 0.03821215778589249,
+        "train_right_knee_r2": 0.7498998641967773,
+        "train_left_foot_mse": 0.0038137855008244514,
+        "train_left_foot_rmse": 0.0617558539802054,
+        "train_left_foot_mae": 0.04399872571229935,
+        "train_left_foot_r2": 0.17480021715164185,
+        "train_right_foot_mse": 0.003928327467292547,
+        "train_right_foot_rmse": 0.06267637088482826,
+        "train_right_foot_mae": 0.045698340982198715,
+        "train_right_foot_r2": 0.1098211407661438
+      },
+      "val_metrics": {
+        "val_mse": 0.00329865631647408,
+        "val_rmse": 0.057433930010700816,
+        "val_mae": 0.04246829077601433,
+        "val_r2": 0.8408486843109131,
+        "val_head_mse": 0.0024432188365608454,
+        "val_head_rmse": 0.04942892712330347,
+        "val_head_mae": 0.03793890029191971,
+        "val_head_r2": 0.7416551113128662,
+        "val_left_shoulder_mse": 0.001684311544522643,
+        "val_left_shoulder_rmse": 0.04104036481956079,
+        "val_left_shoulder_mae": 0.030944740399718285,
+        "val_left_shoulder_r2": 0.6932499408721924,
+        "val_left_elbow_mse": 0.003190949559211731,
+        "val_left_elbow_rmse": 0.05648849050215213,
+        "val_left_elbow_mae": 0.04436313360929489,
+        "val_left_elbow_r2": 0.7658652663230896,
+        "val_right_shoulder_mse": 0.0018238489283248782,
+        "val_right_shoulder_rmse": 0.04270654432665886,
+        "val_right_shoulder_mae": 0.032911598682403564,
+        "val_right_shoulder_r2": 0.7042507529258728,
+        "val_right_elbow_mse": 0.003342842450365424,
+        "val_right_elbow_rmse": 0.0578173196400994,
+        "val_right_elbow_mae": 0.045683834701776505,
+        "val_right_elbow_r2": 0.7745484113693237,
+        "val_left_hand_mse": 0.006071725860238075,
+        "val_left_hand_rmse": 0.07792127989348016,
+        "val_left_hand_mae": 0.06114818900823593,
+        "val_left_hand_r2": 0.7614104151725769,
+        "val_right_hand_mse": 0.006216216366738081,
+        "val_right_hand_rmse": 0.07884298552653927,
+        "val_right_hand_mae": 0.061056043952703476,
+        "val_right_hand_r2": 0.7627074122428894,
+        "val_left_hip_mse": 0.002505603479221463,
+        "val_left_hip_rmse": 0.05005600342837473,
+        "val_left_hip_mae": 0.03692784160375595,
+        "val_left_hip_r2": 0.6750618815422058,
+        "val_right_hip_mse": 0.0024700025096535683,
+        "val_right_hip_rmse": 0.0496991198076341,
+        "val_right_hip_mae": 0.03634550794959068,
+        "val_right_hip_r2": 0.6870356202125549,
+        "val_left_knee_mse": 0.0025547617115080357,
+        "val_left_knee_rmse": 0.050544650671540264,
+        "val_left_knee_mae": 0.0389634408056736,
+        "val_left_knee_r2": 0.7297466993331909,
+        "val_right_knee_mse": 0.003407940734177828,
+        "val_right_knee_rmse": 0.05837757047169596,
+        "val_right_knee_mae": 0.041529685258865356,
+        "val_right_knee_r2": 0.7023165225982666,
+        "val_left_foot_mse": 0.0027765268459916115,
+        "val_left_foot_rmse": 0.05269275895217114,
+        "val_left_foot_mae": 0.03969832509756088,
+        "val_left_foot_r2": 0.17813962697982788,
+        "val_right_foot_mse": 0.004394582472741604,
+        "val_right_foot_rmse": 0.06629164708122437,
+        "val_right_foot_mae": 0.044576507061719894,
+        "val_right_foot_r2": 0.07590389251708984
+      },
+      "best_epoch": 87
+    }
+  ],
+  "test_metrics": {
+    "test_mse": 0.003693027189001441,
+    "test_rmse": 0.060770282120469386,
+    "test_mae": 0.044539257884025574,
+    "test_r2": 0.8268274068832397,
+    "test_head_mse": 0.0026763868518173695,
+    "test_head_rmse": 0.0517338076292222,
+    "test_head_mae": 0.040661051869392395,
+    "test_head_r2": 0.7373135685920715,
+    "test_left_shoulder_mse": 0.0018177879974246025,
+    "test_left_shoulder_rmse": 0.04263552506331548,
+    "test_left_shoulder_mae": 0.03241756930947304,
+    "test_left_shoulder_r2": 0.6811198592185974,
+    "test_left_elbow_mse": 0.0031669889576733112,
+    "test_left_elbow_rmse": 0.056276006944996654,
+    "test_left_elbow_mae": 0.04425946995615959,
+    "test_left_elbow_r2": 0.7662605047225952,
+    "test_right_shoulder_mse": 0.0021612290292978287,
+    "test_right_shoulder_rmse": 0.04648902052418214,
+    "test_right_shoulder_mae": 0.035598043352365494,
+    "test_right_shoulder_r2": 0.6753659844398499,
+    "test_right_elbow_mse": 0.004195427987724543,
+    "test_right_elbow_rmse": 0.06477212353879208,
+    "test_right_elbow_mae": 0.048175785690546036,
+    "test_right_elbow_r2": 0.7316600680351257,
+    "test_left_hand_mse": 0.0059423684142529964,
+    "test_left_hand_rmse": 0.07708675900732237,
+    "test_left_hand_mae": 0.060452744364738464,
+    "test_left_hand_r2": 0.7731010913848877,
+    "test_right_hand_mse": 0.007265681400895119,
+    "test_right_hand_rmse": 0.0852389664466617,
+    "test_right_hand_mae": 0.06327150017023087,
+    "test_right_hand_r2": 0.7417826056480408,
+    "test_left_hip_mse": 0.003142004366964102,
+    "test_left_hip_rmse": 0.05605358478245706,
+    "test_left_hip_mae": 0.0385875478386879,
+    "test_left_hip_r2": 0.6435123682022095,
+    "test_right_hip_mse": 0.003165272995829582,
+    "test_right_hip_rmse": 0.0562607589339993,
+    "test_right_hip_mae": 0.038569748401641846,
+    "test_right_hip_r2": 0.65458083152771,
+    "test_left_knee_mse": 0.002925207605585456,
+    "test_left_knee_rmse": 0.054085188412221104,
+    "test_left_knee_mae": 0.040487684309482574,
+    "test_left_knee_r2": 0.699336588382721,
+    "test_right_knee_mse": 0.0030811612959951162,
+    "test_right_knee_rmse": 0.055508209266694204,
+    "test_right_knee_mae": 0.04153814539313316,
+    "test_right_knee_r2": 0.7126704454421997,
+    "test_left_foot_mse": 0.003813335672020912,
+    "test_left_foot_rmse": 0.061752211879583005,
+    "test_left_foot_mae": 0.045766137540340424,
+    "test_left_foot_r2": 0.1800318956375122,
+    "test_right_foot_mse": 0.004656502977013588,
+    "test_right_foot_rmse": 0.06823857396673517,
+    "test_right_foot_mae": 0.04922492429614067,
+    "test_right_foot_r2": 0.11299479007720947
+  }
+}

A9/cv_results_conv1D_variants/conv1d_v2_fold1_best.h5

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

A9/cv_results_conv1D_variants/conv1d_v2_fold2_best.h5

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

A9/cv_results_conv1D_variants/conv1d_v2_fold3_best.h5

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

A9/cv_results_conv1D_variants/conv1d_v2_fold4_best.h5

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

A9/cv_results_conv1D_variants/conv1d_v2_fold5_best.h5

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

A9/cv_results_conv1D_variants/conv1d_v3_cv_results.json

@@ -0,0 +1,1243 @@
+{
+  "model_name": "conv1d_v3",
+  "n_splits": 5,
+  "epochs": 100,
+  "batch_size": 32,
+  "patience": 10,
+  "aggregate_results": {
+    "n_folds": 5,
+    "train_metrics": {
+      "train_mse": {
+        "mean": 0.0008195941452868282,
+        "std": 8.522000248144307e-05
+      },
+      "train_rmse": {
+        "mean": 0.0285918130299355,
+        "std": 0.0014499561883171191
+      },
+      "train_mae": {
+        "mean": 0.02115652449429035,
+        "std": 0.0009938247639313328
+      },
+      "train_r2": {
+        "mean": 0.9612491011619568,
+        "std": 0.0038233759828969747
+      },
+      "train_head_mse": {
+        "mean": 0.0008750901557505131,
+        "std": 0.00011812912821382188
+      },
+      "train_head_rmse": {
+        "mean": 0.02951907776512544,
+        "std": 0.0019272269318868626
+      },
+      "train_head_mae": {
+        "mean": 0.022351517155766488,
+        "std": 0.0013525196730982763
+      },
+      "train_head_r2": {
+        "mean": 0.9120695114135742,
+        "std": 0.011641433443308376
+      },
+      "train_left_shoulder_mse": {
+        "mean": 0.0005659079179167747,
+        "std": 5.327909449097915e-05
+      },
+      "train_left_shoulder_rmse": {
+        "mean": 0.023763778893861963,
+        "std": 0.0010912060302323931
+      },
+      "train_left_shoulder_mae": {
+        "mean": 0.018041763827204703,
+        "std": 0.0007210157404257646
+      },
+      "train_left_shoulder_r2": {
+        "mean": 0.8982231974601745,
+        "std": 0.00923972819868532
+      },
+      "train_left_elbow_mse": {
+        "mean": 0.0008394661359488964,
+        "std": 8.51702879832458e-05
+      },
+      "train_left_elbow_rmse": {
+        "mean": 0.028937745509017194,
+        "std": 0.001439798876319684
+      },
+      "train_left_elbow_mae": {
+        "mean": 0.022239993885159494,
+        "std": 0.0010486314219589532
+      },
+      "train_left_elbow_r2": {
+        "mean": 0.9391974449157715,
+        "std": 0.006171283733379493
+      },
+      "train_right_shoulder_mse": {
+        "mean": 0.000496693653985858,
+        "std": 5.2685968267384126e-05
+      },
+      "train_right_shoulder_rmse": {
+        "mean": 0.022257275880251278,
+        "std": 0.0011433828651167483
+      },
+      "train_right_shoulder_mae": {
+        "mean": 0.017168228328227998,
+        "std": 0.0008125987598809944
+      },
+      "train_right_shoulder_r2": {
+        "mean": 0.9238792896270752,
+        "std": 0.007878152438655885
+      },
+      "train_right_elbow_mse": {
+        "mean": 0.0007912467932328582,
+        "std": 8.638037747771794e-05
+      },
+      "train_right_elbow_rmse": {
+        "mean": 0.028090033124052744,
+        "std": 0.0014821714888898414
+      },
+      "train_right_elbow_mae": {
+        "mean": 0.021604909002780913,
+        "std": 0.0011041224718242976
+      },
+      "train_right_elbow_r2": {
+        "mean": 0.9498387455940247,
+        "std": 0.005477781094225295
+      },
+      "train_left_hand_mse": {
+        "mean": 0.0012282680487260222,
+        "std": 0.0001568095091660827
+      },
+      "train_left_hand_rmse": {
+        "mean": 0.034978466704192857,
+        "std": 0.0021851580605745627
+      },
+      "train_left_hand_mae": {
+        "mean": 0.02744795083999634,
+        "std": 0.0017758365247434811
+      },
+      "train_left_hand_r2": {
+        "mean": 0.9533008098602295,
+        "std": 0.0057287310128754515
+      },
+      "train_right_hand_mse": {
+        "mean": 0.001363598508760333,
+        "std": 0.0001457597693770755
+      },
+      "train_right_hand_rmse": {
+        "mean": 0.03687756085133102,
+        "std": 0.0019089301759652834
+      },
+      "train_right_hand_mae": {
+        "mean": 0.028094538673758506,
+        "std": 0.0012648543740051863
+      },
+      "train_right_hand_r2": {
+        "mean": 0.9510537147521972,
+        "std": 0.005068510878177982
+      },
+      "train_left_hip_mse": {
+        "mean": 0.0006579913664609194,
+        "std": 3.847135666785556e-05
+      },
+      "train_left_hip_rmse": {
+        "mean": 0.025640809146617656,
+        "std": 0.0007350324942803048
+      },
+      "train_left_hip_mae": {
+        "mean": 0.019576072692871094,
+        "std": 0.0005317582171009088
+      },
+      "train_left_hip_r2": {
+        "mean": 0.9156060576438904,
+        "std": 0.004769001785386521
+      },
+      "train_right_hip_mse": {
+        "mean": 0.00064688038546592,
+        "std": 4.250115808947391e-05
+      },
+      "train_right_hip_rmse": {
+        "mean": 0.02542064778603107,
+        "std": 0.0008191773949968974
+      },
+      "train_right_hip_mae": {
+        "mean": 0.019312109053134918,
+        "std": 0.0005669435969385627
+      },
+      "train_right_hip_r2": {
+        "mean": 0.920278799533844,
+        "std": 0.005043086796485103
+      },
+      "train_left_knee_mse": {
+        "mean": 0.0006886297021992504,
+        "std": 6.974504794838787e-05
+      },
+      "train_left_knee_rmse": {
+        "mean": 0.02620939501624899,
+        "std": 0.0013028104549292132
+      },
+      "train_left_knee_mae": {
+        "mean": 0.019477668032050133,
+        "std": 0.0008917548225038857
+      },
+      "train_left_knee_r2": {
+        "mean": 0.9274759411811828,
+        "std": 0.006539129280202389
+      },
+      "train_right_knee_mse": {
+        "mean": 0.0006816792534664273,
+        "std": 6.827021013290147e-05
+      },
+      "train_right_knee_rmse": {
+        "mean": 0.02607606490617408,
+        "std": 0.0013107602661906138
+      },
+      "train_right_knee_mae": {
+        "mean": 0.01950543448328972,
+        "std": 0.0009483323706038759
+      },
+      "train_right_knee_r2": {
+        "mean": 0.9366542935371399,
+        "std": 0.005958460202074409
+      },
+      "train_left_foot_mse": {
+        "mean": 0.0009236005949787796,
+        "std": 0.00011904990470437431
+      },
+      "train_left_foot_rmse": {
+        "mean": 0.030328433593336625,
+        "std": 0.0019459472637639358
+      },
+      "train_left_foot_mae": {
+        "mean": 0.01970864236354828,
+        "std": 0.0011220017974361374
+      },
+      "train_left_foot_r2": {
+        "mean": 0.7887775897979736,
+        "std": 0.027225303723745767
+      },
+      "train_right_foot_mse": {
+        "mean": 0.0008956713369116187,
+        "std": 0.00012984242297746048
+      },
+      "train_right_foot_rmse": {
+        "mean": 0.02984882005811242,
+        "std": 0.002172389939684928
+      },
+      "train_right_foot_mae": {
+        "mean": 0.020505985617637633,
+        "std": 0.0014937101399877292
+      },
+      "train_right_foot_r2": {
+        "mean": 0.8001867890357971,
+        "std": 0.028967454560583708
+      }
+    },
+    "val_metrics": {
+      "val_mse": {
+        "mean": 0.0010792994755320251,
+        "std": 0.00013184790765934768
+      },
+      "val_rmse": {
+        "mean": 0.03279370229630049,
+        "std": 0.0019678834400545667
+      },
+      "val_mae": {
+        "mean": 0.02366417460143566,
+        "std": 0.0009568243790402584
+      },
+      "val_r2": {
+        "mean": 0.9488348007202149,
+        "std": 0.006920098485728935
+      },
+      "val_head_mse": {
+        "mean": 0.0010808661347255111,
+        "std": 8.265808274630489e-05
+      },
+      "val_head_rmse": {
+        "mean": 0.03285341468336509,
+        "std": 0.001232590105575428
+      },
+      "val_head_mae": {
+        "mean": 0.024595128372311592,
+        "std": 0.0009660829530098315
+      },
+      "val_head_r2": {
+        "mean": 0.8911216259002686,
+        "std": 0.00845311789724493
+      },
+      "val_left_shoulder_mse": {
+        "mean": 0.0007035996648482978,
+        "std": 8.558399011701371e-05
+      },
+      "val_left_shoulder_rmse": {
+        "mean": 0.02647639551050131,
+        "std": 0.0016124967038117582
+      },
+      "val_left_shoulder_mae": {
+        "mean": 0.019845132902264594,
+        "std": 0.000916021347836871
+      },
+      "val_left_shoulder_r2": {
+        "mean": 0.8735331058502197,
+        "std": 0.01187359652185089
+      },
+      "val_left_elbow_mse": {
+        "mean": 0.0011012219940312207,
+        "std": 0.00015562033011014148
+      },
+      "val_left_elbow_rmse": {
+        "mean": 0.033100499838072193,
+        "std": 0.002361970469968483
+      },
+      "val_left_elbow_mae": {
+        "mean": 0.025278348103165628,
+        "std": 0.001571504072522497
+      },
+      "val_left_elbow_r2": {
+        "mean": 0.9201975703239441,
+        "std": 0.010732475416879852
+      },
+      "val_right_shoulder_mse": {
+        "mean": 0.0006283880793489516,
+        "std": 4.664416051379735e-05
+      },
+      "val_right_shoulder_rmse": {
+        "mean": 0.02505011212937417,
+        "std": 0.0009380627136457735
+      },
+      "val_right_shoulder_mae": {
+        "mean": 0.01899750679731369,
+        "std": 0.0006832862341109292
+      },
+      "val_right_shoulder_r2": {
+        "mean": 0.9034538269042969,
+        "std": 0.007720910589557951
+      },
+      "val_right_elbow_mse": {
+        "mean": 0.0011237113503739238,
+        "std": 0.00014178076376119861
+      },
+      "val_right_elbow_rmse": {
+        "mean": 0.033458617006460696,
+        "std": 0.0020572550617030476
+      },
+      "val_right_elbow_mae": {
+        "mean": 0.02502993755042553,
+        "std": 0.0006716551709940101
+      },
+      "val_right_elbow_r2": {
+        "mean": 0.9286151051521301,
+        "std": 0.009059338642804437
+      },
+      "val_left_hand_mse": {
+        "mean": 0.0017177092609927057,
+        "std": 0.00021946972395404376
+      },
+      "val_left_hand_rmse": {
+        "mean": 0.04136231851242698,
+        "std": 0.002620661799860157
+      },
+      "val_left_hand_mae": {
+        "mean": 0.031912262365221974,
+        "std": 0.0017570600179395452
+      },
+      "val_left_hand_r2": {
+        "mean": 0.9343525648117066,
+        "std": 0.009713664214163498
+      },
+      "val_right_hand_mse": {
+        "mean": 0.0019347476540133358,
+        "std": 0.0002444287208313568
+      },
+      "val_right_hand_rmse": {
+        "mean": 0.043895905137120605,
+        "std": 0.0028101897100096746
+      },
+      "val_right_hand_mae": {
+        "mean": 0.0325042475014925,
+        "std": 0.0016575521740539475
+      },
+      "val_right_hand_r2": {
+        "mean": 0.930271565914154,
+        "std": 0.009831162639184982
+      },
+      "val_left_hip_mse": {
+        "mean": 0.0007955238223075866,
+        "std": 7.521734176963166e-05
+      },
+      "val_left_hip_rmse": {
+        "mean": 0.028173078380195872,
+        "std": 0.0013421910783964434
+      },
+      "val_left_hip_mae": {
+        "mean": 0.021371008828282356,
+        "std": 0.0009528261196198866
+      },
+      "val_left_hip_r2": {
+        "mean": 0.8974689483642578,
+        "std": 0.011291635513105044
+      },
+      "val_right_hip_mse": {
+        "mean": 0.000784822553396225,
+        "std": 5.840575042000056e-05
+      },
+      "val_right_hip_rmse": {
+        "mean": 0.02799532181852289,
+        "std": 0.0010413979419805555
+      },
+      "val_right_hip_mae": {
+        "mean": 0.02122575417160988,
+        "std": 0.0007689313780119353
+      },
+      "val_right_hip_r2": {
+        "mean": 0.902676773071289,
+        "std": 0.01016081607000467
+      },
+      "val_left_knee_mse": {
+        "mean": 0.0008590147248469293,
+        "std": 8.795909148904578e-05
+      },
+      "val_left_knee_rmse": {
+        "mean": 0.02926994040604864,
+        "std": 0.0015117253299757984
+      },
+      "val_left_knee_mae": {
+        "mean": 0.02148420810699463,
+        "std": 0.0010762401343415644
+      },
+      "val_left_knee_r2": {
+        "mean": 0.9089836239814758,
+        "std": 0.012050181106136634
+      },
+      "val_right_knee_mse": {
+        "mean": 0.000924004230182618,
+        "std": 0.0001113940336691769
+      },
+      "val_right_knee_rmse": {
+        "mean": 0.030342635630672438,
+        "std": 0.0018244706538788503
+      },
+      "val_right_knee_mae": {
+        "mean": 0.021709297969937324,
+        "std": 0.0010198373307759388
+      },
+      "val_right_knee_r2": {
+        "mean": 0.9138665080070496,
+        "std": 0.011214885431024281
+      },
+      "val_left_foot_mse": {
+        "mean": 0.0011574550182558597,
+        "std": 0.00036306169954031996
+      },
+      "val_left_foot_rmse": {
+        "mean": 0.033594514637409055,
+        "std": 0.005372485880183235
+      },
+      "val_left_foot_mae": {
+        "mean": 0.021299465000629424,
+        "std": 0.0021897443676170956
+      },
+      "val_left_foot_r2": {
+        "mean": 0.7389703631401062,
+        "std": 0.06080312284868038
+      },
+      "val_right_foot_mse": {
+        "mean": 0.001219828729517758,
+        "std": 0.00037586160577890676
+      },
+      "val_right_foot_rmse": {
+        "mean": 0.03455719199142922,
+        "std": 0.005062530116972935
+      },
+      "val_right_foot_mae": {
+        "mean": 0.022381974011659624,
+        "std": 0.002188743333186183
+      },
+      "val_right_foot_r2": {
+        "mean": 0.725255811214447,
+        "std": 0.09442332839332186
+      }
+    },
+    "best_fold": "1",
+    "best_fold_metrics": {
+      "fold": 1,
+      "train_metrics": {
+        "train_mse": 0.0007861112826503813,
+        "train_rmse": 0.02803767612785306,
+        "train_mae": 0.020816409960389137,
+        "train_r2": 0.9630711078643799,
+        "train_head_mse": 0.0008639706647954881,
+        "train_head_rmse": 0.029393377907200258,
+        "train_head_mae": 0.022298693656921387,
+        "train_head_r2": 0.914429783821106,
+        "train_left_shoulder_mse": 0.0005388580029830337,
+        "train_left_shoulder_rmse": 0.023213315208798456,
+        "train_left_shoulder_mae": 0.01779824309051037,
+        "train_left_shoulder_r2": 0.904616117477417,
+        "train_left_elbow_mse": 0.0007662221323698759,
+        "train_left_elbow_rmse": 0.027680717699689,
+        "train_left_elbow_mae": 0.021295204758644104,
+        "train_left_elbow_r2": 0.9444840550422668,
+        "train_right_shoulder_mse": 0.00048733223229646683,
+        "train_right_shoulder_rmse": 0.02207560264854545,
+        "train_right_shoulder_mae": 0.01691562868654728,
+        "train_right_shoulder_r2": 0.9262508153915405,
+        "train_right_elbow_mse": 0.0007603385020047426,
+        "train_right_elbow_rmse": 0.02757423619984319,
+        "train_right_elbow_mae": 0.021216168999671936,
+        "train_right_elbow_r2": 0.9522366523742676,
+        "train_left_hand_mse": 0.0011274965945631266,
+        "train_left_hand_rmse": 0.03357821607177973,
+        "train_left_hand_mae": 0.02620617300271988,
+        "train_left_hand_r2": 0.9567295908927917,
+        "train_right_hand_mse": 0.0012768891174346209,
+        "train_right_hand_rmse": 0.03573358528659867,
+        "train_right_hand_mae": 0.02717418596148491,
+        "train_right_hand_r2": 0.954308271408081,
+        "train_left_hip_mse": 0.0006451396620832384,
+        "train_left_hip_rmse": 0.02539959964415263,
+        "train_left_hip_mae": 0.01937951147556305,
+        "train_left_hip_r2": 0.9192973971366882,
+        "train_right_hip_mse": 0.0006362437852658331,
+        "train_right_hip_rmse": 0.025223873320048076,
+        "train_right_hip_mae": 0.019131945446133614,
+        "train_right_hip_r2": 0.9233337640762329,
+        "train_left_knee_mse": 0.0006518944865092635,
+        "train_left_knee_rmse": 0.02553222447240474,
+        "train_left_knee_mae": 0.018823018297553062,
+        "train_left_knee_r2": 0.9318124651908875,
+        "train_right_knee_mse": 0.0006627499242313206,
+        "train_right_knee_rmse": 0.025743929852128648,
+        "train_right_knee_mae": 0.01942424476146698,
+        "train_right_knee_r2": 0.9383745193481445,
+        "train_left_foot_mse": 0.000912507763132453,
+        "train_left_foot_rmse": 0.030207743429995777,
+        "train_left_foot_mae": 0.020114228129386902,
+        "train_left_foot_r2": 0.7928653359413147,
+        "train_right_foot_mse": 0.000889803224708885,
+        "train_right_foot_rmse": 0.029829569636668996,
+        "train_right_foot_mae": 0.02083607017993927,
+        "train_right_foot_r2": 0.7995445132255554
+      },
+      "val_metrics": {
+        "val_mse": 0.0009312412585131824,
+        "val_rmse": 0.03051624581289747,
+        "val_mae": 0.022245794534683228,
+        "val_r2": 0.9546714425086975,
+        "val_head_mse": 0.0010041429195553064,
+        "val_head_rmse": 0.031688214205841676,
+        "val_head_mae": 0.023374570533633232,
+        "val_head_r2": 0.8924050331115723,
+        "val_left_shoulder_mse": 0.0005871722241863608,
+        "val_left_shoulder_rmse": 0.02423163684496697,
+        "val_left_shoulder_mae": 0.01849154755473137,
+        "val_left_shoulder_r2": 0.8867601752281189,
+        "val_left_elbow_mse": 0.0008660847670398653,
+        "val_left_elbow_rmse": 0.029429318154518384,
+        "val_left_elbow_mae": 0.022576389834284782,
+        "val_left_elbow_r2": 0.9372300505638123,
+        "val_right_shoulder_mse": 0.0005498766549862921,
+        "val_right_shoulder_rmse": 0.023449448927134558,
+        "val_right_shoulder_mae": 0.01789841800928116,
+        "val_right_shoulder_r2": 0.9110301733016968,
+        "val_right_elbow_mse": 0.0009768627351149917,
+        "val_right_elbow_rmse": 0.03125480339267857,
+        "val_right_elbow_mae": 0.023741306737065315,
+        "val_right_elbow_r2": 0.9356913566589355,
+        "val_left_hand_mse": 0.0014157150872051716,
+        "val_left_hand_rmse": 0.03762598951795383,
+        "val_left_hand_mae": 0.029277106747031212,
+        "val_left_hand_r2": 0.9478591084480286,
+        "val_right_hand_mse": 0.0015470071230083704,
+        "val_right_hand_rmse": 0.03933201142845825,
+        "val_right_hand_mae": 0.029717104509472847,
+        "val_right_hand_r2": 0.9436669945716858,
+        "val_left_hip_mse": 0.0006925173220224679,
+        "val_left_hip_rmse": 0.026315723855187184,
+        "val_left_hip_mae": 0.019766192883253098,
+        "val_left_hip_r2": 0.9012318849563599,
+        "val_right_hip_mse": 0.0007024001679383218,
+        "val_right_hip_rmse": 0.02650283320587295,
+        "val_right_hip_mae": 0.020036572590470314,
+        "val_right_hip_r2": 0.9048244953155518,
+        "val_left_knee_mse": 0.0007193634519353509,
+        "val_left_knee_rmse": 0.026820951734331706,
+        "val_left_knee_mae": 0.019679632037878036,
+        "val_left_knee_r2": 0.9216668009757996,
+        "val_right_knee_mse": 0.0008180341683328152,
+        "val_right_knee_rmse": 0.02860129661978308,
+        "val_right_knee_mae": 0.021385936066508293,
+        "val_right_knee_r2": 0.9239248633384705,
+        "val_left_foot_mse": 0.0011842323001474142,
+        "val_left_foot_rmse": 0.034412676445568925,
+        "val_left_foot_mae": 0.021772537380456924,
+        "val_left_foot_r2": 0.7210932970046997,
+        "val_right_foot_mse": 0.001042727380990982,
+        "val_right_foot_rmse": 0.03229128955292715,
+        "val_right_foot_mae": 0.021477986127138138,
+        "val_right_foot_r2": 0.7765542268753052
+      },
+      "best_epoch": 100
+    }
+  },
+  "fold_details": [
+    {
+      "fold": 1,
+      "train_metrics": {
+        "train_mse": 0.0007861112826503813,
+        "train_rmse": 0.02803767612785306,
+        "train_mae": 0.020816409960389137,
+        "train_r2": 0.9630711078643799,
+        "train_head_mse": 0.0008639706647954881,
+        "train_head_rmse": 0.029393377907200258,
+        "train_head_mae": 0.022298693656921387,
+        "train_head_r2": 0.914429783821106,
+        "train_left_shoulder_mse": 0.0005388580029830337,
+        "train_left_shoulder_rmse": 0.023213315208798456,
+        "train_left_shoulder_mae": 0.01779824309051037,
+        "train_left_shoulder_r2": 0.904616117477417,
+        "train_left_elbow_mse": 0.0007662221323698759,
+        "train_left_elbow_rmse": 0.027680717699689,
+        "train_left_elbow_mae": 0.021295204758644104,
+        "train_left_elbow_r2": 0.9444840550422668,
+        "train_right_shoulder_mse": 0.00048733223229646683,
+        "train_right_shoulder_rmse": 0.02207560264854545,
+        "train_right_shoulder_mae": 0.01691562868654728,
+        "train_right_shoulder_r2": 0.9262508153915405,
+        "train_right_elbow_mse": 0.0007603385020047426,
+        "train_right_elbow_rmse": 0.02757423619984319,
+        "train_right_elbow_mae": 0.021216168999671936,
+        "train_right_elbow_r2": 0.9522366523742676,
+        "train_left_hand_mse": 0.0011274965945631266,
+        "train_left_hand_rmse": 0.03357821607177973,
+        "train_left_hand_mae": 0.02620617300271988,
+        "train_left_hand_r2": 0.9567295908927917,
+        "train_right_hand_mse": 0.0012768891174346209,
+        "train_right_hand_rmse": 0.03573358528659867,
+        "train_right_hand_mae": 0.02717418596148491,
+        "train_right_hand_r2": 0.954308271408081,
+        "train_left_hip_mse": 0.0006451396620832384,
+        "train_left_hip_rmse": 0.02539959964415263,
+        "train_left_hip_mae": 0.01937951147556305,
+        "train_left_hip_r2": 0.9192973971366882,
+        "train_right_hip_mse": 0.0006362437852658331,
+        "train_right_hip_rmse": 0.025223873320048076,
+        "train_right_hip_mae": 0.019131945446133614,
+        "train_right_hip_r2": 0.9233337640762329,
+        "train_left_knee_mse": 0.0006518944865092635,
+        "train_left_knee_rmse": 0.02553222447240474,
+        "train_left_knee_mae": 0.018823018297553062,
+        "train_left_knee_r2": 0.9318124651908875,
+        "train_right_knee_mse": 0.0006627499242313206,
+        "train_right_knee_rmse": 0.025743929852128648,
+        "train_right_knee_mae": 0.01942424476146698,
+        "train_right_knee_r2": 0.9383745193481445,
+        "train_left_foot_mse": 0.000912507763132453,
+        "train_left_foot_rmse": 0.030207743429995777,
+        "train_left_foot_mae": 0.020114228129386902,
+        "train_left_foot_r2": 0.7928653359413147,
+        "train_right_foot_mse": 0.000889803224708885,
+        "train_right_foot_rmse": 0.029829569636668996,
+        "train_right_foot_mae": 0.02083607017993927,
+        "train_right_foot_r2": 0.7995445132255554
+      },
+      "val_metrics": {
+        "val_mse": 0.0009312412585131824,
+        "val_rmse": 0.03051624581289747,
+        "val_mae": 0.022245794534683228,
+        "val_r2": 0.9546714425086975,
+        "val_head_mse": 0.0010041429195553064,
+        "val_head_rmse": 0.031688214205841676,
+        "val_head_mae": 0.023374570533633232,
+        "val_head_r2": 0.8924050331115723,
+        "val_left_shoulder_mse": 0.0005871722241863608,
+        "val_left_shoulder_rmse": 0.02423163684496697,
+        "val_left_shoulder_mae": 0.01849154755473137,
+        "val_left_shoulder_r2": 0.8867601752281189,
+        "val_left_elbow_mse": 0.0008660847670398653,
+        "val_left_elbow_rmse": 0.029429318154518384,
+        "val_left_elbow_mae": 0.022576389834284782,
+        "val_left_elbow_r2": 0.9372300505638123,
+        "val_right_shoulder_mse": 0.0005498766549862921,
+        "val_right_shoulder_rmse": 0.023449448927134558,
+        "val_right_shoulder_mae": 0.01789841800928116,
+        "val_right_shoulder_r2": 0.9110301733016968,
+        "val_right_elbow_mse": 0.0009768627351149917,
+        "val_right_elbow_rmse": 0.03125480339267857,
+        "val_right_elbow_mae": 0.023741306737065315,
+        "val_right_elbow_r2": 0.9356913566589355,
+        "val_left_hand_mse": 0.0014157150872051716,
+        "val_left_hand_rmse": 0.03762598951795383,
+        "val_left_hand_mae": 0.029277106747031212,
+        "val_left_hand_r2": 0.9478591084480286,
+        "val_right_hand_mse": 0.0015470071230083704,
+        "val_right_hand_rmse": 0.03933201142845825,
+        "val_right_hand_mae": 0.029717104509472847,
+        "val_right_hand_r2": 0.9436669945716858,
+        "val_left_hip_mse": 0.0006925173220224679,
+        "val_left_hip_rmse": 0.026315723855187184,
+        "val_left_hip_mae": 0.019766192883253098,
+        "val_left_hip_r2": 0.9012318849563599,
+        "val_right_hip_mse": 0.0007024001679383218,
+        "val_right_hip_rmse": 0.02650283320587295,
+        "val_right_hip_mae": 0.020036572590470314,
+        "val_right_hip_r2": 0.9048244953155518,
+        "val_left_knee_mse": 0.0007193634519353509,
+        "val_left_knee_rmse": 0.026820951734331706,
+        "val_left_knee_mae": 0.019679632037878036,
+        "val_left_knee_r2": 0.9216668009757996,
+        "val_right_knee_mse": 0.0008180341683328152,
+        "val_right_knee_rmse": 0.02860129661978308,
+        "val_right_knee_mae": 0.021385936066508293,
+        "val_right_knee_r2": 0.9239248633384705,
+        "val_left_foot_mse": 0.0011842323001474142,
+        "val_left_foot_rmse": 0.034412676445568925,
+        "val_left_foot_mae": 0.021772537380456924,
+        "val_left_foot_r2": 0.7210932970046997,
+        "val_right_foot_mse": 0.001042727380990982,
+        "val_right_foot_rmse": 0.03229128955292715,
+        "val_right_foot_mae": 0.021477986127138138,
+        "val_right_foot_r2": 0.7765542268753052
+      },
+      "best_epoch": 100
+    },
+    {
+      "fold": 2,
+      "train_metrics": {
+        "train_mse": 0.0007456785533577204,
+        "train_rmse": 0.02730711543458445,
+        "train_mae": 0.020359808579087257,
+        "train_r2": 0.9644478559494019,
+        "train_head_mse": 0.0008064030553214252,
+        "train_head_rmse": 0.028397236755033493,
+        "train_head_mae": 0.021492645144462585,
+        "train_head_r2": 0.9175206422805786,
+        "train_left_shoulder_mse": 0.0005214340053498745,
+        "train_left_shoulder_rmse": 0.022834929501749604,
+        "train_left_shoulder_mae": 0.017472142353653908,
+        "train_left_shoulder_r2": 0.9057337045669556,
+        "train_left_elbow_mse": 0.000770989281591028,
+        "train_left_elbow_rmse": 0.027766693746123755,
+        "train_left_elbow_mae": 0.021601319313049316,
+        "train_left_elbow_r2": 0.9445399641990662,
+        "train_right_shoulder_mse": 0.0004687804321292788,
+        "train_right_shoulder_rmse": 0.021651337883125808,
+        "train_right_shoulder_mae": 0.01679183356463909,
+        "train_right_shoulder_r2": 0.9269556403160095,
+        "train_right_elbow_mse": 0.0007275167154148221,
+        "train_right_elbow_rmse": 0.02697251778041535,
+        "train_right_elbow_mae": 0.020797409117221832,
+        "train_right_elbow_r2": 0.9535130858421326,
+        "train_left_hand_mse": 0.001111118937842548,
+        "train_left_hand_rmse": 0.033333450734098144,
+        "train_left_hand_mae": 0.02602473646402359,
+        "train_left_hand_r2": 0.957931399345398,
+        "train_right_hand_mse": 0.0012774013448506594,
+        "train_right_hand_rmse": 0.035740751878642105,
+        "train_right_hand_mae": 0.027278462424874306,
+        "train_right_hand_r2": 0.9539018869400024,
+        "train_left_hip_mse": 0.0006411146023310721,
+        "train_left_hip_rmse": 0.02532024096115738,
+        "train_left_hip_mae": 0.019386382773518562,
+        "train_left_hip_r2": 0.9151297807693481,
+        "train_right_hip_mse": 0.0006240523071028292,
+        "train_right_hip_rmse": 0.024981038951629477,
+        "train_right_hip_mae": 0.019072195515036583,
+        "train_right_hip_r2": 0.9201457500457764,
+        "train_left_knee_mse": 0.0006319819949567318,
+        "train_left_knee_rmse": 0.02513925207631945,
+        "train_left_knee_mae": 0.01887628249824047,
+        "train_left_knee_r2": 0.9319978952407837,
+        "train_right_knee_mse": 0.0005759704508818686,
+        "train_right_knee_rmse": 0.02399938438547682,
+        "train_right_knee_mae": 0.018420448526740074,
+        "train_right_knee_r2": 0.9465124607086182,
+        "train_left_foot_mse": 0.0008239722810685635,
+        "train_left_foot_rmse": 0.02870491736738783,
+        "train_left_foot_mae": 0.018449537456035614,
+        "train_left_foot_r2": 0.8092038631439209,
+        "train_right_foot_mse": 0.0007130855228751898,
+        "train_right_foot_rmse": 0.026703661226041453,
+        "train_right_foot_mae": 0.019014138728380203,
+        "train_right_foot_r2": 0.8438988327980042
+      },
+      "val_metrics": {
+        "val_mse": 0.000982781290076673,
+        "val_rmse": 0.03134934273755469,
+        "val_mae": 0.023252200335264206,
+        "val_r2": 0.9548811912536621,
+        "val_head_mse": 0.0010659160325303674,
+        "val_head_rmse": 0.03264836952330648,
+        "val_head_mae": 0.024518506601452827,
+        "val_head_r2": 0.89961838722229,
+        "val_left_shoulder_mse": 0.0006559884059242904,
+        "val_left_shoulder_rmse": 0.025612270612428926,
+        "val_left_shoulder_mae": 0.019161345437169075,
+        "val_left_shoulder_r2": 0.8839410543441772,
+        "val_left_elbow_mse": 0.0010436734883114696,
+        "val_left_elbow_rmse": 0.032305935806155955,
+        "val_left_elbow_mae": 0.02516184002161026,
+        "val_left_elbow_r2": 0.9220538139343262,
+        "val_right_shoulder_mse": 0.0006253825267776847,
+        "val_right_shoulder_rmse": 0.025007649365297903,
+        "val_right_shoulder_mae": 0.018766775727272034,
+        "val_right_shoulder_r2": 0.9097753167152405,
+        "val_right_elbow_mse": 0.0010318104177713394,
+        "val_right_elbow_rmse": 0.03212180595438774,
+        "val_right_elbow_mae": 0.025114841759204865,
+        "val_right_elbow_r2": 0.9364110827445984,
+        "val_left_hand_mse": 0.0016580740921199322,
+        "val_left_hand_rmse": 0.040719455940863604,
+        "val_left_hand_mae": 0.031140200793743134,
+        "val_left_hand_r2": 0.9355162978172302,
+        "val_right_hand_mse": 0.0018276158953085542,
+        "val_right_hand_rmse": 0.04275062450197136,
+        "val_right_hand_mae": 0.032092202454805374,
+        "val_right_hand_r2": 0.935502290725708,
+        "val_left_hip_mse": 0.0007238457328639925,
+        "val_left_hip_rmse": 0.02690438129494883,
+        "val_left_hip_mae": 0.020988699048757553,
+        "val_left_hip_r2": 0.9173582196235657,
+        "val_right_hip_mse": 0.00074816431151703,
+        "val_right_hip_rmse": 0.027352592409441375,
+        "val_right_hip_mae": 0.021082568913698196,
+        "val_right_hip_r2": 0.9195799827575684,
+        "val_left_knee_mse": 0.0008337528561241925,
+        "val_left_knee_rmse": 0.02887477889307886,
+        "val_left_knee_mae": 0.0215870663523674,
+        "val_left_knee_r2": 0.9184720516204834,
+        "val_right_knee_mse": 0.0007972652674652636,
+        "val_right_knee_rmse": 0.028235886163980466,
+        "val_right_knee_mae": 0.020951585844159126,
+        "val_right_knee_r2": 0.9255000352859497,
+        "val_left_foot_mse": 0.0008481299737468362,
+        "val_left_foot_rmse": 0.02912267113001203,
+        "val_left_foot_mae": 0.020923549309372902,
+        "val_left_foot_r2": 0.814799964427948,
+        "val_right_foot_mse": 0.0009165380033664405,
+        "val_right_foot_rmse": 0.030274378661938557,
+        "val_right_foot_mae": 0.020789435133337975,
+        "val_right_foot_r2": 0.7788927555084229
+      },
+      "best_epoch": 100
+    },
+    {
+      "fold": 3,
+      "train_metrics": {
+        "train_mse": 0.0007579308585263789,
+        "train_rmse": 0.027530544101531645,
+        "train_mae": 0.020333630964159966,
+        "train_r2": 0.9637446403503418,
+        "train_head_mse": 0.000764990400057286,
+        "train_head_rmse": 0.027658459828003547,
+        "train_head_mae": 0.021056337282061577,
+        "train_head_r2": 0.9222634434700012,
+        "train_left_shoulder_mse": 0.0005329269333742559,
+        "train_left_shoulder_rmse": 0.023085210273555142,
+        "train_left_shoulder_mae": 0.01748127117753029,
+        "train_left_shoulder_r2": 0.9021981358528137,
+        "train_left_elbow_mse": 0.000797020795289427,
+        "train_left_elbow_rmse": 0.0282315567280557,
+        "train_left_elbow_mae": 0.02153290994465351,
+        "train_left_elbow_r2": 0.9417159557342529,
+        "train_right_shoulder_mse": 0.00045735828462056816,
+        "train_right_shoulder_rmse": 0.021385936608448276,
+        "train_right_shoulder_mae": 0.016732478514313698,
+        "train_right_shoulder_r2": 0.9289395213127136,
+        "train_right_elbow_mse": 0.0007456714520230889,
+        "train_right_elbow_rmse": 0.027306985407091148,
+        "train_right_elbow_mae": 0.020864685997366905,
+        "train_right_elbow_r2": 0.9519994854927063,
+        "train_left_hand_mse": 0.001096230000257492,
+        "train_left_hand_rmse": 0.033109364238195396,
+        "train_left_hand_mae": 0.026092173531651497,
+        "train_left_hand_r2": 0.9578813314437866,
+        "train_right_hand_mse": 0.0012684365501627326,
+        "train_right_hand_rmse": 0.03561511687700509,
+        "train_right_hand_mae": 0.027241498231887817,
+        "train_right_hand_r2": 0.9536604285240173,
+        "train_left_hip_mse": 0.0006386404857039452,
+        "train_left_hip_rmse": 0.02527133723616432,
+        "train_left_hip_mae": 0.019365152344107628,
+        "train_left_hip_r2": 0.9176927208900452,
+        "train_right_hip_mse": 0.0006358911050483584,
+        "train_right_hip_rmse": 0.02521688135056273,
+        "train_right_hip_mae": 0.019037069752812386,
+        "train_right_hip_r2": 0.9215686917304993,
+        "train_left_knee_mse": 0.0006292670732364058,
+        "train_left_knee_rmse": 0.025085196296549202,
+        "train_left_knee_mae": 0.01872355118393898,
+        "train_left_knee_r2": 0.9331634640693665,
+        "train_right_knee_mse": 0.0006966079818084836,
+        "train_right_knee_rmse": 0.026393332146746527,
+        "train_right_knee_mae": 0.019136209040880203,
+        "train_right_knee_r2": 0.9348675608634949,
+        "train_left_foot_mse": 0.0007778523140586913,
+        "train_left_foot_rmse": 0.027890003837552465,
+        "train_left_foot_mae": 0.018378129228949547,
+        "train_left_foot_r2": 0.817461371421814,
+        "train_right_foot_mse": 0.000812207639683038,
+        "train_right_foot_rmse": 0.028499256826854943,
+        "train_right_foot_mae": 0.018695728853344917,
+        "train_right_foot_r2": 0.8172916769981384
+      },
+      "val_metrics": {
+        "val_mse": 0.0010925802635028958,
+        "val_rmse": 0.033054201903886526,
+        "val_mae": 0.023668775334954262,
+        "val_r2": 0.9505183100700378,
+        "val_head_mse": 0.0010753707028925419,
+        "val_head_rmse": 0.03279284530034779,
+        "val_head_mae": 0.02457379177212715,
+        "val_head_r2": 0.8964306116104126,
+        "val_left_shoulder_mse": 0.0008209768566302955,
+        "val_left_shoulder_rmse": 0.028652693706356747,
+        "val_left_shoulder_mae": 0.02067362517118454,
+        "val_left_shoulder_r2": 0.8631707429885864,
+        "val_left_elbow_mse": 0.001207534340210259,
+        "val_left_elbow_rmse": 0.034749594820807035,
+        "val_left_elbow_mae": 0.025961553677916527,
+        "val_left_elbow_r2": 0.9157079458236694,
+        "val_right_shoulder_mse": 0.0006329387542791665,
+        "val_right_shoulder_rmse": 0.025158274071946318,
+        "val_right_shoulder_mae": 0.01919138990342617,
+        "val_right_shoulder_r2": 0.9079540967941284,
+        "val_right_elbow_mse": 0.001118272659368813,
+        "val_right_elbow_rmse": 0.033440584016563066,
+        "val_right_elbow_mae": 0.02554670348763466,
+        "val_right_elbow_r2": 0.9331808686256409,
+        "val_left_hand_mse": 0.001667400007136166,
+        "val_left_hand_rmse": 0.04083380960841354,
+        "val_left_hand_mae": 0.03170023486018181,
+        "val_left_hand_r2": 0.9389197826385498,
+        "val_right_hand_mse": 0.0019663642160594463,
+        "val_right_hand_rmse": 0.04434370548408699,
+        "val_right_hand_mae": 0.03394725173711777,
+        "val_right_hand_r2": 0.933923602104187,
+        "val_left_hip_mse": 0.0008482676348648965,
+        "val_left_hip_rmse": 0.02912503450409796,
+        "val_left_hip_mae": 0.021484917029738426,
+        "val_left_hip_r2": 0.893262505531311,
+        "val_right_hip_mse": 0.000783855386544019,
+        "val_right_hip_rmse": 0.027997417497762522,
+        "val_right_hip_mae": 0.02088623307645321,
+        "val_right_hip_r2": 0.9037198424339294,
+        "val_left_knee_mse": 0.0009044389589689672,
+        "val_left_knee_rmse": 0.030073891649884076,
+        "val_left_knee_mae": 0.021031994372606277,
+        "val_left_knee_r2": 0.9074762463569641,
+        "val_right_knee_mse": 0.0009019335266202688,
+        "val_right_knee_rmse": 0.030032208154251144,
+        "val_right_knee_mae": 0.02054513245820999,
+        "val_right_knee_r2": 0.9179959297180176,
+        "val_left_foot_mse": 0.001346761011518538,
+        "val_left_foot_rmse": 0.036698242621664295,
+        "val_left_foot_mae": 0.02141515165567398,
+        "val_left_foot_r2": 0.719498336315155,
+        "val_right_foot_mse": 0.0009294285555370152,
+        "val_right_foot_rmse": 0.030486530723206524,
+        "val_right_foot_mae": 0.02073611319065094,
+        "val_right_foot_r2": 0.7995038032531738
+      },
+      "best_epoch": 100
+    },
+    {
+      "fold": 4,
+      "train_metrics": {
+        "train_mse": 0.00098039535805583,
+        "train_rmse": 0.031311265673169934,
+        "train_mae": 0.0230281800031662,
+        "train_r2": 0.9539508819580078,
+        "train_head_mse": 0.0011019082739949226,
+        "train_head_rmse": 0.03319500375048815,
+        "train_head_mae": 0.024920126423239708,
+        "train_head_r2": 0.8893497586250305,
+        "train_left_shoulder_mse": 0.0006678358186036348,
+        "train_left_shoulder_rmse": 0.025842519586983675,
+        "train_left_shoulder_mae": 0.019420664757490158,
+        "train_left_shoulder_r2": 0.8805099725723267,
+        "train_left_elbow_mse": 0.000992606277577579,
+        "train_left_elbow_rmse": 0.03150565469209581,
+        "train_left_elbow_mae": 0.02411818504333496,
+        "train_left_elbow_r2": 0.9280871152877808,
+        "train_right_shoulder_mse": 0.0006003073649480939,
+        "train_right_shoulder_rmse": 0.024501170685256937,
+        "train_right_shoulder_mae": 0.018782123923301697,
+        "train_right_shoulder_r2": 0.9082709550857544,
+        "train_right_elbow_mse": 0.0009623089572414756,
+        "train_right_elbow_rmse": 0.03102110502934213,
+        "train_right_elbow_mae": 0.023770926520228386,
+        "train_right_elbow_r2": 0.9389318227767944,
+        "train_left_hand_mse": 0.0015042368322610855,
+        "train_left_hand_rmse": 0.03878449216195934,
+        "train_left_hand_mae": 0.03052580915391445,
+        "train_left_hand_r2": 0.9431060552597046,
+        "train_right_hand_mse": 0.0016497756587341428,
+        "train_right_hand_rmse": 0.04061743047921844,
+        "train_right_hand_mae": 0.03048606403172016,
+        "train_right_hand_r2": 0.9409971237182617,
+        "train_left_hip_mse": 0.0007343547185882926,
+        "train_left_hip_rmse": 0.02709898002855998,
+        "train_left_hip_mae": 0.02062179334461689,
+        "train_left_hip_r2": 0.9065733551979065,
+        "train_right_hip_mse": 0.000729475577827543,
+        "train_right_hip_rmse": 0.02700880556091926,
+        "train_right_hip_mae": 0.020434221252799034,
+        "train_right_hip_r2": 0.9108489751815796,
+        "train_left_knee_mse": 0.0008128245244733989,
+        "train_left_knee_rmse": 0.02851007759500838,
+        "train_left_knee_mae": 0.021029029041528702,
+        "train_left_knee_r2": 0.9159364104270935,
+        "train_right_knee_mse": 0.0007891195709817111,
+        "train_right_knee_rmse": 0.02809127214957897,
+        "train_right_knee_mae": 0.02127259224653244,
+        "train_right_knee_r2": 0.9281361103057861,
+        "train_left_foot_mse": 0.00110964837949723,
+        "train_left_foot_rmse": 0.03331138513327283,
+        "train_left_foot_mae": 0.021241867914795876,
+        "train_left_foot_r2": 0.7394368648529053,
+        "train_right_foot_mse": 0.0010907379910349846,
+        "train_right_foot_rmse": 0.03302632269924983,
+        "train_right_foot_mae": 0.02274293079972267,
+        "train_right_foot_r2": 0.760577380657196
+      },
+      "val_metrics": {
+        "val_mse": 0.0013148579746484756,
+        "val_rmse": 0.03626097040412012,
+        "val_mae": 0.02518092654645443,
+        "val_r2": 0.935970664024353,
+        "val_head_mse": 0.0012373874196782708,
+        "val_head_rmse": 0.03517651801526511,
+        "val_head_mae": 0.026327058672904968,
+        "val_head_r2": 0.87516850233078,
+        "val_left_shoulder_mse": 0.0007814045529812574,
+        "val_left_shoulder_rmse": 0.027953614309803615,
+        "val_left_shoulder_mae": 0.02093876712024212,
+        "val_left_shoulder_r2": 0.8562626242637634,
+        "val_left_elbow_mse": 0.001324441866017878,
+        "val_left_elbow_rmse": 0.03639288207902581,
+        "val_left_elbow_mae": 0.027415448799729347,
+        "val_left_elbow_r2": 0.9040942192077637,
+        "val_right_shoulder_mse": 0.0006962199695408344,
+        "val_right_shoulder_rmse": 0.0263859805491635,
+        "val_right_shoulder_mae": 0.020007289946079254,
+        "val_right_shoulder_r2": 0.8918880820274353,
+        "val_right_elbow_mse": 0.0013883801875635982,
+        "val_right_elbow_rmse": 0.037260974055485965,
+        "val_right_elbow_mae": 0.025579355657100677,
+        "val_right_elbow_r2": 0.912197470664978,
+        "val_left_hand_mse": 0.002094997325912118,
+        "val_left_hand_rmse": 0.04577114075388681,
+        "val_left_hand_mae": 0.03452865406870842,
+        "val_left_hand_r2": 0.9183269739151001,
+        "val_right_hand_mse": 0.002285967580974102,
+        "val_right_hand_rmse": 0.047811793325225754,
+        "val_right_hand_mae": 0.03233948349952698,
+        "val_right_hand_r2": 0.9163970947265625,
+        "val_left_hip_mse": 0.000822409987449646,
+        "val_left_hip_rmse": 0.0286776914595587,
+        "val_left_hip_mae": 0.02214895747601986,
+        "val_left_hip_r2": 0.8909863829612732,
+        "val_right_hip_mse": 0.0008155871764756739,
+        "val_right_hip_rmse": 0.028558486943038034,
+        "val_right_hip_mae": 0.021995579823851585,
+        "val_right_hip_r2": 0.8960140943527222,
+        "val_left_knee_mse": 0.0009871009970083833,
+        "val_left_knee_rmse": 0.031418163488790736,
+        "val_left_knee_mae": 0.02257070504128933,
+        "val_left_knee_r2": 0.8872637152671814,
+        "val_right_knee_mse": 0.0010213336208835244,
+        "val_right_knee_rmse": 0.03195831067005145,
+        "val_right_knee_mae": 0.0222636666148901,
+        "val_right_knee_r2": 0.8963767290115356,
+        "val_left_foot_mse": 0.0017151953652501106,
+        "val_left_foot_rmse": 0.04141491718270254,
+        "val_left_foot_mae": 0.024626245722174644,
+        "val_left_foot_r2": 0.6445774435997009,
+        "val_right_foot_mse": 0.0019227283773943782,
+        "val_right_foot_rmse": 0.04384892675305039,
+        "val_right_foot_mae": 0.026610836386680603,
+        "val_right_foot_r2": 0.5421112775802612
+      },
+      "best_epoch": 83
+    },
+    {
+      "fold": 5,
+      "train_metrics": {
+        "train_mse": 0.0008278546738438308,
+        "train_rmse": 0.028772463812538385,
+        "train_mae": 0.0212445929646492,
+        "train_r2": 0.9610310196876526,
+        "train_head_mse": 0.0008381783845834434,
+        "train_head_rmse": 0.02895131058490174,
+        "train_head_mae": 0.02198978327214718,
+        "train_head_r2": 0.9167839288711548,
+        "train_left_shoulder_mse": 0.0005684848292730749,
+        "train_left_shoulder_rmse": 0.023842919898222927,
+        "train_left_shoulder_mae": 0.0180364977568388,
+        "train_left_shoulder_r2": 0.8980580568313599,
+        "train_left_elbow_mse": 0.0008704921929165721,
+        "train_left_elbow_rmse": 0.029504104679121717,
+        "train_left_elbow_mae": 0.02265235036611557,
+        "train_left_elbow_r2": 0.9371601343154907,
+        "train_right_shoulder_mse": 0.00046968995593488216,
+        "train_right_shoulder_rmse": 0.021672331575879927,
+        "train_right_shoulder_mae": 0.01661907695233822,
+        "train_right_shoulder_r2": 0.9289795160293579,
+        "train_right_elbow_mse": 0.0007603983394801617,
+        "train_right_elbow_rmse": 0.027575321203571895,
+        "train_right_elbow_mae": 0.021375354379415512,
+        "train_right_elbow_r2": 0.9525126814842224,
+        "train_left_hand_mse": 0.0013022578787058592,
+        "train_left_hand_rmse": 0.036086810314931675,
+        "train_left_hand_mae": 0.02839086204767227,
+        "train_left_hand_r2": 0.9508556723594666,
+        "train_right_hand_mse": 0.0013454898726195097,
+        "train_right_hand_rmse": 0.036680919735190796,
+        "train_right_hand_mae": 0.02829248271882534,
+        "train_right_hand_r2": 0.9524008631706238,
+        "train_left_hip_mse": 0.0006307073635980487,
+        "train_left_hip_rmse": 0.025113887863053953,
+        "train_left_hip_mae": 0.01912752352654934,
+        "train_left_hip_r2": 0.9193370342254639,
+        "train_right_hip_mse": 0.000608739152085036,
+        "train_right_hip_rmse": 0.02467263974699578,
+        "train_right_hip_mae": 0.018885113298892975,
+        "train_right_hip_r2": 0.9254968166351318,
+        "train_left_knee_mse": 0.0007171804318204522,
+        "train_left_knee_rmse": 0.026780224640963194,
+        "train_left_knee_mae": 0.01993645913898945,
+        "train_left_knee_r2": 0.9244694709777832,
+        "train_right_knee_mse": 0.0006839483394287527,
+        "train_right_knee_rmse": 0.026152405996939413,
+        "train_right_knee_mae": 0.019273677840828896,
+        "train_right_knee_r2": 0.9353808164596558,
+        "train_left_foot_mse": 0.00099402223713696,
+        "train_left_foot_rmse": 0.0315281181984742,
+        "train_left_foot_mae": 0.020359449088573456,
+        "train_left_foot_r2": 0.7849205136299133,
+        "train_right_foot_mse": 0.0009725223062559962,
+        "train_right_foot_rmse": 0.031185289901746883,
+        "train_right_foot_mae": 0.02124105952680111,
+        "train_right_foot_r2": 0.7796215415000916
+      },
+      "val_metrics": {
+        "val_mse": 0.0010750365909188986,
+        "val_rmse": 0.03278775062304364,
+        "val_mae": 0.02397317625582218,
+        "val_r2": 0.9481323957443237,
+        "val_head_mse": 0.0010215135989710689,
+        "val_head_rmse": 0.03196112637206437,
+        "val_head_mae": 0.02418171428143978,
+        "val_head_r2": 0.8919855952262878,
+        "val_left_shoulder_mse": 0.000672456284519285,
+        "val_left_shoulder_rmse": 0.0259317620789503,
+        "val_left_shoulder_mae": 0.019960379227995872,
+        "val_left_shoulder_r2": 0.8775309324264526,
+        "val_left_elbow_mse": 0.0010643755085766315,
+        "val_left_elbow_rmse": 0.0326247683298538,
+        "val_left_elbow_mae": 0.025276508182287216,
+        "val_left_elbow_r2": 0.9219018220901489,
+        "val_right_shoulder_mse": 0.0006375224911607802,
+        "val_right_shoulder_rmse": 0.025249207733328588,
+        "val_right_shoulder_mae": 0.019123660400509834,
+        "val_right_shoulder_r2": 0.8966214656829834,
+        "val_right_elbow_mse": 0.0011032307520508766,
+        "val_right_elbow_rmse": 0.03321491761318816,
+        "val_right_elbow_mae": 0.02516748011112213,
+        "val_right_elbow_r2": 0.9255947470664978,
+        "val_left_hand_mse": 0.0017523597925901413,
+        "val_left_hand_rmse": 0.041861196741017105,
+        "val_left_hand_mae": 0.03291511535644531,
+        "val_left_hand_r2": 0.931140661239624,
+        "val_right_hand_mse": 0.0020467834547162056,
+        "val_right_hand_rmse": 0.04524139094586069,
+        "val_right_hand_mae": 0.034425195306539536,
+        "val_right_hand_r2": 0.921867847442627,
+        "val_left_hip_mse": 0.0008905784343369305,
+        "val_left_hip_rmse": 0.029842560787186653,
+        "val_left_hip_mae": 0.022466277703642845,
+        "val_left_hip_r2": 0.8845057487487793,
+        "val_right_hip_mse": 0.0008741057245060802,
+        "val_right_hip_rmse": 0.029565279036499557,
+        "val_right_hip_mae": 0.022127816453576088,
+        "val_right_hip_r2": 0.8892454504966736,
+        "val_left_knee_mse": 0.0008504173601977527,
+        "val_left_knee_rmse": 0.029161916264157827,
+        "val_left_knee_mae": 0.0225516427308321,
+        "val_left_knee_r2": 0.9100393056869507,
+        "val_right_knee_mse": 0.0010814545676112175,
+        "val_right_knee_rmse": 0.03288547654529606,
+        "val_right_knee_mae": 0.023400168865919113,
+        "val_right_knee_r2": 0.9055349826812744,
+        "val_left_foot_mse": 0.000692956440616399,
+        "val_left_foot_rmse": 0.026324065807097487,
+        "val_left_foot_mae": 0.017759840935468674,
+        "val_left_foot_r2": 0.7948827743530273,
+        "val_right_foot_mse": 0.0012877213302999735,
+        "val_right_foot_rmse": 0.035884834266023485,
+        "val_right_foot_mae": 0.022295499220490456,
+        "val_right_foot_r2": 0.729216992855072
+      },
+      "best_epoch": 100
+    }
+  ],
+  "test_metrics": {
+    "test_mse": 0.0010114616015926003,
+    "test_rmse": 0.031803484110905215,
+    "test_mae": 0.023287829011678696,
+    "test_r2": 0.9525707960128784,
+    "test_head_mse": 0.001058185938745737,
+    "test_head_rmse": 0.03252977003831624,
+    "test_head_mae": 0.025076698511838913,
+    "test_head_r2": 0.8961394429206848,
+    "test_left_shoulder_mse": 0.0006880196742713451,
+    "test_left_shoulder_rmse": 0.026230129131808427,
+    "test_left_shoulder_mae": 0.019607869908213615,
+    "test_left_shoulder_r2": 0.8793061375617981,
+    "test_left_elbow_mse": 0.0009536789148114622,
+    "test_left_elbow_rmse": 0.030881692227134546,
+    "test_left_elbow_mae": 0.023212479427456856,
+    "test_left_elbow_r2": 0.9296137690544128,
+    "test_right_shoulder_mse": 0.0006721963291056454,
+    "test_right_shoulder_rmse": 0.025926749296925858,
+    "test_right_shoulder_mae": 0.01951335184276104,
+    "test_right_shoulder_r2": 0.8990306854248047,
+    "test_right_elbow_mse": 0.0011157888220623136,
+    "test_right_elbow_rmse": 0.033403425304335384,
+    "test_right_elbow_mae": 0.02502848207950592,
+    "test_right_elbow_r2": 0.9286340475082397,
+    "test_left_hand_mse": 0.0013995806220918894,
+    "test_left_hand_rmse": 0.03741096927495851,
+    "test_left_hand_mae": 0.028312556445598602,
+    "test_left_hand_r2": 0.946559488773346,
+    "test_right_hand_mse": 0.0017985906451940536,
+    "test_right_hand_rmse": 0.04240979421305948,
+    "test_right_hand_mae": 0.032232630997896194,
+    "test_right_hand_r2": 0.9360793232917786,
+    "test_left_hip_mse": 0.0007564277038909495,
+    "test_left_hip_rmse": 0.027503230790053547,
+    "test_left_hip_mae": 0.020848345011472702,
+    "test_left_hip_r2": 0.9141767024993896,
+    "test_right_hip_mse": 0.0007855687290430069,
+    "test_right_hip_rmse": 0.0280279990196055,
+    "test_right_hip_mae": 0.020996693521738052,
+    "test_right_hip_r2": 0.914272665977478,
+    "test_left_knee_mse": 0.0008229437517002225,
+    "test_left_knee_rmse": 0.02868699621257378,
+    "test_left_knee_mae": 0.02099251188337803,
+    "test_left_knee_r2": 0.9154148697853088,
+    "test_right_knee_mse": 0.0008464669226668775,
+    "test_right_knee_rmse": 0.029094104603284796,
+    "test_right_knee_mae": 0.021935109049081802,
+    "test_right_knee_r2": 0.9210638403892517,
+    "test_left_foot_mse": 0.0011088320752605796,
+    "test_left_foot_rmse": 0.03329913024780947,
+    "test_left_foot_mae": 0.022115133702754974,
+    "test_left_foot_r2": 0.7615717649459839,
+    "test_right_foot_mse": 0.0011427195277065039,
+    "test_right_foot_rmse": 0.03380413477233967,
+    "test_right_foot_mae": 0.022869901731610298,
+    "test_right_foot_r2": 0.7823262810707092
+  }
+}

A9/cv_results_conv1D_variants/conv1d_v3_fold1_best.h5

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

A9/cv_results_conv1D_variants/conv1d_v3_fold2_best.h5

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

A9/cv_results_conv1D_variants/conv1d_v3_fold3_best.h5

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

A9/cv_results_conv1D_variants/conv1d_v3_fold4_best.h5

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

A9/cv_results_conv1D_variants/conv1d_v3_fold5_best.h5

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

A9/cv_results_conv1D_variants/summary.txt

@@ -0,0 +1,388 @@
+Cross-Validation Training Summary
+==================================================
+CONV1D:
+  Best Fold: 1
+  Val RMSE: 0.040316 ± 0.002366
+  Val MAE:  0.029083 ± 0.001559
+  Val R²:   0.922923 ± 0.007843
+  Test RMSE: 0.039097
+  Test MAE:  0.028836
+  Test R²:   0.928322
+
+CONV1D_V2:
+  Best Fold: 5
+  Val RMSE: 0.064831 ± 0.005451
+  Val MAE:  0.048145 ± 0.003991
+  Val R²:   0.799991 ± 0.031161
+  Test RMSE: 0.060770
+  Test MAE:  0.044539
+  Test R²:   0.826827
+
+CONV1D_V3:
+  Best Fold: 1
+  Val RMSE: 0.032794 ± 0.001968
+  Val MAE:  0.023664 ± 0.000957
+  Val R²:   0.948835 ± 0.006920
+  Test RMSE: 0.031803
+  Test MAE:  0.023288
+  Test R²:   0.952571
+
+CONV1D
+------------------------------
+Best Fold: 1
+
+Validation Metrics (mean ± std):
+  val_mse: 0.001631 ± 0.000192
+  val_rmse: 0.040316 ± 0.002366
+  val_mae: 0.029083 ± 0.001559
+  val_r2: 0.922923 ± 0.007843
+  val_head_mse: 0.001529 ± 0.000224
+  val_head_rmse: 0.039002 ± 0.002768
+  val_head_mae: 0.029092 ± 0.001792
+  val_head_r2: 0.846743 ± 0.016058
+  val_left_shoulder_mse: 0.000963 ± 0.000125
+  val_left_shoulder_rmse: 0.030973 ± 0.002011
+  val_left_shoulder_mae: 0.023305 ± 0.001337
+  val_left_shoulder_r2: 0.826978 ± 0.016900
+  val_left_elbow_mse: 0.001576 ± 0.000200
+  val_left_elbow_rmse: 0.039617 ± 0.002544
+  val_left_elbow_mae: 0.030470 ± 0.001593
+  val_left_elbow_r2: 0.885617 ± 0.015235
+  val_right_shoulder_mse: 0.000988 ± 0.000207
+  val_right_shoulder_rmse: 0.031285 ± 0.003099
+  val_right_shoulder_mae: 0.023348 ± 0.001964
+  val_right_shoulder_r2: 0.849188 ± 0.025334
+  val_right_elbow_mse: 0.001723 ± 0.000311
+  val_right_elbow_rmse: 0.041342 ± 0.003651
+  val_right_elbow_mae: 0.030990 ± 0.002007
+  val_right_elbow_r2: 0.890991 ± 0.017207
+  val_left_hand_mse: 0.002681 ± 0.000349
+  val_left_hand_rmse: 0.051675 ± 0.003333
+  val_left_hand_mae: 0.039669 ± 0.002268
+  val_left_hand_r2: 0.897519 ± 0.015381
+  val_right_hand_mse: 0.002881 ± 0.000463
+  val_right_hand_rmse: 0.053509 ± 0.004263
+  val_right_hand_mae: 0.039932 ± 0.002167
+  val_right_hand_r2: 0.896516 ± 0.015386
+  val_left_hip_mse: 0.001284 ± 0.000146
+  val_left_hip_rmse: 0.035772 ± 0.002064
+  val_left_hip_mae: 0.026941 ± 0.001497
+  val_left_hip_r2: 0.835549 ± 0.011550
+  val_right_hip_mse: 0.001280 ± 0.000162
+  val_right_hip_rmse: 0.035712 ± 0.002252
+  val_right_hip_mae: 0.026605 ± 0.001674
+  val_right_hip_r2: 0.842650 ± 0.009382
+  val_left_knee_mse: 0.001356 ± 0.000224
+  val_left_knee_rmse: 0.036698 ± 0.002962
+  val_left_knee_mae: 0.027006 ± 0.002151
+  val_left_knee_r2: 0.857515 ± 0.017901
+  val_right_knee_mse: 0.001422 ± 0.000159
+  val_right_knee_rmse: 0.037657 ± 0.002099
+  val_right_knee_mae: 0.027371 ± 0.001548
+  val_right_knee_r2: 0.867600 ± 0.013914
+  val_left_foot_mse: 0.001716 ± 0.000358
+  val_left_foot_rmse: 0.041171 ± 0.004575
+  val_left_foot_mae: 0.026308 ± 0.002103
+  val_left_foot_r2: 0.611655 ± 0.039050
+  val_right_foot_mse: 0.001802 ± 0.000205
+  val_right_foot_rmse: 0.042386 ± 0.002427
+  val_right_foot_mae: 0.027043 ± 0.000887
+  val_right_foot_r2: 0.595503 ± 0.057589
+
+Test Metrics:
+  test_mse: 0.001529
+  test_rmse: 0.039097
+  test_mae: 0.028836
+  test_r2: 0.928322
+  test_head_mse: 0.001464
+  test_head_rmse: 0.038262
+  test_head_mae: 0.029478
+  test_head_r2: 0.856313
+  test_left_shoulder_mse: 0.000950
+  test_left_shoulder_rmse: 0.030822
+  test_left_shoulder_mae: 0.023182
+  test_left_shoulder_r2: 0.833345
+  test_left_elbow_mse: 0.001432
+  test_left_elbow_rmse: 0.037840
+  test_left_elbow_mae: 0.029071
+  test_left_elbow_r2: 0.894321
+  test_right_shoulder_mse: 0.001005
+  test_right_shoulder_rmse: 0.031705
+  test_right_shoulder_mae: 0.023657
+  test_right_shoulder_r2: 0.849013
+  test_right_elbow_mse: 0.001674
+  test_right_elbow_rmse: 0.040911
+  test_right_elbow_mae: 0.031103
+  test_right_elbow_r2: 0.892949
+  test_left_hand_mse: 0.002371
+  test_left_hand_rmse: 0.048688
+  test_left_hand_mae: 0.037719
+  test_left_hand_r2: 0.909486
+  test_right_hand_mse: 0.002621
+  test_right_hand_rmse: 0.051193
+  test_right_hand_mae: 0.039511
+  test_right_hand_r2: 0.906861
+  test_left_hip_mse: 0.001303
+  test_left_hip_rmse: 0.036102
+  test_left_hip_mae: 0.026887
+  test_left_hip_r2: 0.852123
+  test_right_hip_mse: 0.001351
+  test_right_hip_rmse: 0.036762
+  test_right_hip_mae: 0.026841
+  test_right_hip_r2: 0.852516
+  test_left_knee_mse: 0.001224
+  test_left_knee_rmse: 0.034984
+  test_left_knee_mae: 0.025411
+  test_left_knee_r2: 0.874203
+  test_right_knee_mse: 0.001286
+  test_right_knee_rmse: 0.035866
+  test_right_knee_mae: 0.026939
+  test_right_knee_r2: 0.880040
+  test_left_foot_mse: 0.001381
+  test_left_foot_rmse: 0.037162
+  test_left_foot_mae: 0.026248
+  test_left_foot_r2: 0.703046
+  test_right_foot_mse: 0.001809
+  test_right_foot_rmse: 0.042537
+  test_right_foot_mae: 0.028814
+  test_right_foot_r2: 0.655339
+
+CONV1D_V2
+------------------------------
+Best Fold: 5
+
+Validation Metrics (mean ± std):
+  val_mse: 0.004233 ± 0.000697
+  val_rmse: 0.064831 ± 0.005451
+  val_mae: 0.048145 ± 0.003991
+  val_r2: 0.799991 ± 0.031161
+  val_head_mse: 0.003063 ± 0.000571
+  val_head_rmse: 0.055095 ± 0.005225
+  val_head_mae: 0.042829 ± 0.004275
+  val_head_r2: 0.693616 ± 0.045626
+  val_left_shoulder_mse: 0.001973 ± 0.000279
+  val_left_shoulder_rmse: 0.044300 ± 0.003202
+  val_left_shoulder_mae: 0.034408 ± 0.002626
+  val_left_shoulder_r2: 0.645670 ± 0.040738
+  val_left_elbow_mse: 0.003893 ± 0.000709
+  val_left_elbow_rmse: 0.062133 ± 0.005664
+  val_left_elbow_mae: 0.048762 ± 0.004067
+  val_left_elbow_r2: 0.717805 ± 0.050683
+  val_right_shoulder_mse: 0.002374 ± 0.000404
+  val_right_shoulder_rmse: 0.048550 ± 0.004152
+  val_right_shoulder_mae: 0.037451 ± 0.003441
+  val_right_shoulder_r2: 0.637360 ± 0.048556
+  val_right_elbow_mse: 0.004425 ± 0.000990
+  val_right_elbow_rmse: 0.066094 ± 0.007555
+  val_right_elbow_mae: 0.051465 ± 0.005392
+  val_right_elbow_r2: 0.720580 ± 0.055990
+  val_left_hand_mse: 0.007699 ± 0.001442
+  val_left_hand_rmse: 0.087352 ± 0.008269
+  val_left_hand_mae: 0.068520 ± 0.006089
+  val_left_hand_r2: 0.706109 ± 0.057988
+  val_right_hand_mse: 0.008157 ± 0.001910
+  val_right_hand_rmse: 0.089657 ± 0.010884
+  val_right_hand_mae: 0.068902 ± 0.007470
+  val_right_hand_r2: 0.707724 ± 0.064522
+  val_left_hip_mse: 0.003819 ± 0.000867
+  val_left_hip_rmse: 0.061366 ± 0.007267
+  val_left_hip_mae: 0.045751 ± 0.005730
+  val_left_hip_r2: 0.512239 ± 0.095079
+  val_right_hip_mse: 0.003961 ± 0.000975
+  val_right_hip_rmse: 0.062426 ± 0.008028
+  val_right_hip_mae: 0.046281 ± 0.006374
+  val_right_hip_r2: 0.514749 ± 0.099893
+  val_left_knee_mse: 0.003682 ± 0.000762
+  val_left_knee_rmse: 0.060346 ± 0.006360
+  val_left_knee_mae: 0.044499 ± 0.003591
+  val_left_knee_r2: 0.612126 ± 0.073504
+  val_right_knee_mse: 0.003762 ± 0.000256
+  val_right_knee_rmse: 0.061297 ± 0.002083
+  val_right_knee_mae: 0.045253 ± 0.002069
+  val_right_knee_r2: 0.649232 ± 0.029182
+  val_left_foot_mse: 0.003977 ± 0.000641
+  val_left_foot_rmse: 0.062834 ± 0.005353
+  val_left_foot_mae: 0.044975 ± 0.002832
+  val_left_foot_r2: 0.093872 ± 0.049758
+  val_right_foot_mse: 0.004242 ± 0.000232
+  val_right_foot_rmse: 0.065103 ± 0.001791
+  val_right_foot_mae: 0.046784 ± 0.001124
+  val_right_foot_r2: 0.053017 ± 0.016706
+
+Test Metrics:
+  test_mse: 0.003693
+  test_rmse: 0.060770
+  test_mae: 0.044539
+  test_r2: 0.826827
+  test_head_mse: 0.002676
+  test_head_rmse: 0.051734
+  test_head_mae: 0.040661
+  test_head_r2: 0.737314
+  test_left_shoulder_mse: 0.001818
+  test_left_shoulder_rmse: 0.042636
+  test_left_shoulder_mae: 0.032418
+  test_left_shoulder_r2: 0.681120
+  test_left_elbow_mse: 0.003167
+  test_left_elbow_rmse: 0.056276
+  test_left_elbow_mae: 0.044259
+  test_left_elbow_r2: 0.766261
+  test_right_shoulder_mse: 0.002161
+  test_right_shoulder_rmse: 0.046489
+  test_right_shoulder_mae: 0.035598
+  test_right_shoulder_r2: 0.675366
+  test_right_elbow_mse: 0.004195
+  test_right_elbow_rmse: 0.064772
+  test_right_elbow_mae: 0.048176
+  test_right_elbow_r2: 0.731660
+  test_left_hand_mse: 0.005942
+  test_left_hand_rmse: 0.077087
+  test_left_hand_mae: 0.060453
+  test_left_hand_r2: 0.773101
+  test_right_hand_mse: 0.007266
+  test_right_hand_rmse: 0.085239
+  test_right_hand_mae: 0.063272
+  test_right_hand_r2: 0.741783
+  test_left_hip_mse: 0.003142
+  test_left_hip_rmse: 0.056054
+  test_left_hip_mae: 0.038588
+  test_left_hip_r2: 0.643512
+  test_right_hip_mse: 0.003165
+  test_right_hip_rmse: 0.056261
+  test_right_hip_mae: 0.038570
+  test_right_hip_r2: 0.654581
+  test_left_knee_mse: 0.002925
+  test_left_knee_rmse: 0.054085
+  test_left_knee_mae: 0.040488
+  test_left_knee_r2: 0.699337
+  test_right_knee_mse: 0.003081
+  test_right_knee_rmse: 0.055508
+  test_right_knee_mae: 0.041538
+  test_right_knee_r2: 0.712670
+  test_left_foot_mse: 0.003813
+  test_left_foot_rmse: 0.061752
+  test_left_foot_mae: 0.045766
+  test_left_foot_r2: 0.180032
+  test_right_foot_mse: 0.004657
+  test_right_foot_rmse: 0.068239
+  test_right_foot_mae: 0.049225
+  test_right_foot_r2: 0.112995
+
+CONV1D_V3
+------------------------------
+Best Fold: 1
+
+Validation Metrics (mean ± std):
+  val_mse: 0.001079 ± 0.000132
+  val_rmse: 0.032794 ± 0.001968
+  val_mae: 0.023664 ± 0.000957
+  val_r2: 0.948835 ± 0.006920
+  val_head_mse: 0.001081 ± 0.000083
+  val_head_rmse: 0.032853 ± 0.001233
+  val_head_mae: 0.024595 ± 0.000966
+  val_head_r2: 0.891122 ± 0.008453
+  val_left_shoulder_mse: 0.000704 ± 0.000086
+  val_left_shoulder_rmse: 0.026476 ± 0.001612
+  val_left_shoulder_mae: 0.019845 ± 0.000916
+  val_left_shoulder_r2: 0.873533 ± 0.011874
+  val_left_elbow_mse: 0.001101 ± 0.000156
+  val_left_elbow_rmse: 0.033100 ± 0.002362
+  val_left_elbow_mae: 0.025278 ± 0.001572
+  val_left_elbow_r2: 0.920198 ± 0.010732
+  val_right_shoulder_mse: 0.000628 ± 0.000047
+  val_right_shoulder_rmse: 0.025050 ± 0.000938
+  val_right_shoulder_mae: 0.018998 ± 0.000683
+  val_right_shoulder_r2: 0.903454 ± 0.007721
+  val_right_elbow_mse: 0.001124 ± 0.000142
+  val_right_elbow_rmse: 0.033459 ± 0.002057
+  val_right_elbow_mae: 0.025030 ± 0.000672
+  val_right_elbow_r2: 0.928615 ± 0.009059
+  val_left_hand_mse: 0.001718 ± 0.000219
+  val_left_hand_rmse: 0.041362 ± 0.002621
+  val_left_hand_mae: 0.031912 ± 0.001757
+  val_left_hand_r2: 0.934353 ± 0.009714
+  val_right_hand_mse: 0.001935 ± 0.000244
+  val_right_hand_rmse: 0.043896 ± 0.002810
+  val_right_hand_mae: 0.032504 ± 0.001658
+  val_right_hand_r2: 0.930272 ± 0.009831
+  val_left_hip_mse: 0.000796 ± 0.000075
+  val_left_hip_rmse: 0.028173 ± 0.001342
+  val_left_hip_mae: 0.021371 ± 0.000953
+  val_left_hip_r2: 0.897469 ± 0.011292
+  val_right_hip_mse: 0.000785 ± 0.000058
+  val_right_hip_rmse: 0.027995 ± 0.001041
+  val_right_hip_mae: 0.021226 ± 0.000769
+  val_right_hip_r2: 0.902677 ± 0.010161
+  val_left_knee_mse: 0.000859 ± 0.000088
+  val_left_knee_rmse: 0.029270 ± 0.001512
+  val_left_knee_mae: 0.021484 ± 0.001076
+  val_left_knee_r2: 0.908984 ± 0.012050
+  val_right_knee_mse: 0.000924 ± 0.000111
+  val_right_knee_rmse: 0.030343 ± 0.001824
+  val_right_knee_mae: 0.021709 ± 0.001020
+  val_right_knee_r2: 0.913867 ± 0.011215
+  val_left_foot_mse: 0.001157 ± 0.000363
+  val_left_foot_rmse: 0.033595 ± 0.005372
+  val_left_foot_mae: 0.021299 ± 0.002190
+  val_left_foot_r2: 0.738970 ± 0.060803
+  val_right_foot_mse: 0.001220 ± 0.000376
+  val_right_foot_rmse: 0.034557 ± 0.005063
+  val_right_foot_mae: 0.022382 ± 0.002189
+  val_right_foot_r2: 0.725256 ± 0.094423
+
+Test Metrics:
+  test_mse: 0.001011
+  test_rmse: 0.031803
+  test_mae: 0.023288
+  test_r2: 0.952571
+  test_head_mse: 0.001058
+  test_head_rmse: 0.032530
+  test_head_mae: 0.025077
+  test_head_r2: 0.896139
+  test_left_shoulder_mse: 0.000688
+  test_left_shoulder_rmse: 0.026230
+  test_left_shoulder_mae: 0.019608
+  test_left_shoulder_r2: 0.879306
+  test_left_elbow_mse: 0.000954
+  test_left_elbow_rmse: 0.030882
+  test_left_elbow_mae: 0.023212
+  test_left_elbow_r2: 0.929614
+  test_right_shoulder_mse: 0.000672
+  test_right_shoulder_rmse: 0.025927
+  test_right_shoulder_mae: 0.019513
+  test_right_shoulder_r2: 0.899031
+  test_right_elbow_mse: 0.001116
+  test_right_elbow_rmse: 0.033403
+  test_right_elbow_mae: 0.025028
+  test_right_elbow_r2: 0.928634
+  test_left_hand_mse: 0.001400
+  test_left_hand_rmse: 0.037411
+  test_left_hand_mae: 0.028313
+  test_left_hand_r2: 0.946559
+  test_right_hand_mse: 0.001799
+  test_right_hand_rmse: 0.042410
+  test_right_hand_mae: 0.032233
+  test_right_hand_r2: 0.936079
+  test_left_hip_mse: 0.000756
+  test_left_hip_rmse: 0.027503
+  test_left_hip_mae: 0.020848
+  test_left_hip_r2: 0.914177
+  test_right_hip_mse: 0.000786
+  test_right_hip_rmse: 0.028028
+  test_right_hip_mae: 0.020997
+  test_right_hip_r2: 0.914273
+  test_left_knee_mse: 0.000823
+  test_left_knee_rmse: 0.028687
+  test_left_knee_mae: 0.020993
+  test_left_knee_r2: 0.915415
+  test_right_knee_mse: 0.000846
+  test_right_knee_rmse: 0.029094
+  test_right_knee_mae: 0.021935
+  test_right_knee_r2: 0.921064
+  test_left_foot_mse: 0.001109
+  test_left_foot_rmse: 0.033299
+  test_left_foot_mae: 0.022115
+  test_left_foot_r2: 0.761572
+  test_right_foot_mse: 0.001143
+  test_right_foot_rmse: 0.033804
+  test_right_foot_mae: 0.022870
+  test_right_foot_r2: 0.782326

A9/cv_results_different_models/conv1d_cv_results.json

@@ -0,0 +1,1185 @@
+{
+  "model_name": "conv1d",
+  "n_splits": 5,
+  "epochs": 100,
+  "batch_size": 32,
+  "patience": 15,
+  "aggregate_results": {
+    "n_folds": 5,
+    "train_metrics": {
+      "train_mse": {
+        "mean": 0.001259609032422304,
+        "std": 7.550476276570708e-05
+      },
+      "train_rmse": {
+        "mean": 0.03547526018318331,
+        "std": 0.0010559106769764501
+      },
+      "train_mae": {
+        "mean": 0.026310380175709726,
+        "std": 0.0007320774790352503
+      },
+      "train_r2": {
+        "mean": 0.9404989957809449,
+        "std": 0.003872366942575071
+      },
+      "train_head_mse": {
+        "mean": 0.0012524700257927179,
+        "std": 5.36134209779224e-05
+      },
+      "train_head_rmse": {
+        "mean": 0.035382010641099235,
+        "std": 0.0007637727318113063
+      },
+      "train_head_mae": {
+        "mean": 0.026747101917862892,
+        "std": 0.0005221287067489147
+      },
+      "train_head_r2": {
+        "mean": 0.8746951818466187,
+        "std": 0.00551120655708088
+      },
+      "train_left_shoulder_mse": {
+        "mean": 0.0007708263117820024,
+        "std": 4.023474370802645e-05
+      },
+      "train_left_shoulder_rmse": {
+        "mean": 0.02775416979964716,
+        "std": 0.000729637248470758
+      },
+      "train_left_shoulder_mae": {
+        "mean": 0.021192917600274086,
+        "std": 0.0005329086014195033
+      },
+      "train_left_shoulder_r2": {
+        "mean": 0.8620152354240418,
+        "std": 0.007639330833151162
+      },
+      "train_left_elbow_mse": {
+        "mean": 0.0012883748160675168,
+        "std": 9.712364012455063e-05
+      },
+      "train_left_elbow_rmse": {
+        "mean": 0.035868259739117336,
+        "std": 0.0013574827272312912
+      },
+      "train_left_elbow_mae": {
+        "mean": 0.027828055247664452,
+        "std": 0.0010849892897577486
+      },
+      "train_left_elbow_r2": {
+        "mean": 0.90631582736969,
+        "std": 0.007279602522703073
+      },
+      "train_right_shoulder_mse": {
+        "mean": 0.0007489739800803363,
+        "std": 3.6106333368265796e-05
+      },
+      "train_right_shoulder_rmse": {
+        "mean": 0.027359244855069808,
+        "std": 0.000667608448622784
+      },
+      "train_right_shoulder_mae": {
+        "mean": 0.020959306880831717,
+        "std": 0.0004964498604249058
+      },
+      "train_right_shoulder_r2": {
+        "mean": 0.8856366157531739,
+        "std": 0.0057457757087126604
+      },
+      "train_right_elbow_mse": {
+        "mean": 0.0012683736393228173,
+        "std": 0.0001307190629120947
+      },
+      "train_right_elbow_rmse": {
+        "mean": 0.03556815111370491,
+        "std": 0.0018111503735111299
+      },
+      "train_right_elbow_mae": {
+        "mean": 0.0275300744920969,
+        "std": 0.0014273066473422119
+      },
+      "train_right_elbow_r2": {
+        "mean": 0.9194227576255798,
+        "std": 0.008503822342908543
+      },
+      "train_left_hand_mse": {
+        "mean": 0.002168831182643771,
+        "std": 0.00018565277569346935
+      },
+      "train_left_hand_rmse": {
+        "mean": 0.046528618808285224,
+        "std": 0.001979599564825643
+      },
+      "train_left_hand_mae": {
+        "mean": 0.0360553577542305,
+        "std": 0.0016096628408419473
+      },
+      "train_left_hand_r2": {
+        "mean": 0.9173852205276489,
+        "std": 0.007565631399954955
+      },
+      "train_right_hand_mse": {
+        "mean": 0.002271615946665406,
+        "std": 0.00024551003772017177
+      },
+      "train_right_hand_rmse": {
+        "mean": 0.04759315036992276,
+        "std": 0.0025510751716339764
+      },
+      "train_right_hand_mae": {
+        "mean": 0.03630002439022064,
+        "std": 0.0017420582134845008
+      },
+      "train_right_hand_r2": {
+        "mean": 0.9185583233833313,
+        "std": 0.009197658830319559
+      },
+      "train_left_hip_mse": {
+        "mean": 0.0010194688336923718,
+        "std": 1.3491613473324573e-05
+      },
+      "train_left_hip_rmse": {
+        "mean": 0.031928419076407366,
+        "std": 0.0002118701811915491
+      },
+      "train_left_hip_mae": {
+        "mean": 0.024465654417872428,
+        "std": 0.00027758570216617905
+      },
+      "train_left_hip_r2": {
+        "mean": 0.8725345492362976,
+        "std": 0.003265704253089168
+      },
+      "train_right_hip_mse": {
+        "mean": 0.0010216994676738977,
+        "std": 8.718723253455878e-06
+      },
+      "train_right_hip_rmse": {
+        "mean": 0.03196374286604521,
+        "std": 0.00013641725419423088
+      },
+      "train_right_hip_mae": {
+        "mean": 0.024226189032196997,
+        "std": 0.0002455609632160475
+      },
+      "train_right_hip_r2": {
+        "mean": 0.8772361636161804,
+        "std": 0.002357004090175088
+      },
+      "train_left_knee_mse": {
+        "mean": 0.0010100262588821352,
+        "std": 6.581146801440587e-05
+      },
+      "train_left_knee_rmse": {
+        "mean": 0.03176444230854482,
+        "std": 0.001022968088096209
+      },
+      "train_left_knee_mae": {
+        "mean": 0.02387089729309082,
+        "std": 0.0008655138371260777
+      },
+      "train_left_knee_r2": {
+        "mean": 0.894062876701355,
+        "std": 0.007051579136615254
+      },
+      "train_right_knee_mse": {
+        "mean": 0.001021817699074745,
+        "std": 5.209155981672568e-05
+      },
+      "train_right_knee_rmse": {
+        "mean": 0.031955567159870045,
+        "std": 0.0008120508394060965
+      },
+      "train_right_knee_mae": {
+        "mean": 0.02432553954422474,
+        "std": 0.0006972828066834722
+      },
+      "train_right_knee_r2": {
+        "mean": 0.9049419283866882,
+        "std": 0.004863647502354999
+      },
+      "train_left_foot_mse": {
+        "mean": 0.0012303898110985756,
+        "std": 0.00014673490187274178
+      },
+      "train_left_foot_rmse": {
+        "mean": 0.03501575395064565,
+        "std": 0.0020704555939258804
+      },
+      "train_left_foot_mae": {
+        "mean": 0.02346724011003971,
+        "std": 0.0012687842006887011
+      },
+      "train_left_foot_r2": {
+        "mean": 0.7220417022705078,
+        "std": 0.03459181770948605
+      },
+      "train_right_foot_mse": {
+        "mean": 0.0013020491925999523,
+        "std": 7.717531827969755e-05
+      },
+      "train_right_foot_rmse": {
+        "mean": 0.036068437329192835,
+        "std": 0.0010568922509133366
+      },
+      "train_right_foot_mae": {
+        "mean": 0.025066573917865754,
+        "std": 0.0005491446024528564
+      },
+      "train_right_foot_r2": {
+        "mean": 0.7189775705337524,
+        "std": 0.021267577416146592
+      }
+    },
+    "val_metrics": {
+      "val_mse": {
+        "mean": 0.0014235081616789103,
+        "std": 0.00010525706170943662
+      },
+      "val_rmse": {
+        "mean": 0.037703938997917026,
+        "std": 0.0013860540105862806
+      },
+      "val_mae": {
+        "mean": 0.027598119899630548,
+        "std": 0.0009052339259298692
+      },
+      "val_r2": {
+        "mean": 0.9326810359954834,
+        "std": 0.005506496408093385
+      },
+      "val_head_mse": {
+        "mean": 0.0013791328994557262,
+        "std": 7.868717057161147e-05
+      },
+      "val_head_rmse": {
+        "mean": 0.03712188234169263,
+        "std": 0.001048213177390558
+      },
+      "val_head_mae": {
+        "mean": 0.027978580445051193,
+        "std": 0.0006755588705626507
+      },
+      "val_head_r2": {
+        "mean": 0.861529266834259,
+        "std": 0.01083436668846819
+      },
+      "val_left_shoulder_mse": {
+        "mean": 0.0008574498118832707,
+        "std": 4.5196062404026485e-05
+      },
+      "val_left_shoulder_rmse": {
+        "mean": 0.02927222102948161,
+        "std": 0.000766086081615806
+      },
+      "val_left_shoulder_mae": {
+        "mean": 0.022188059240579604,
+        "std": 0.00031762614949704934
+      },
+      "val_left_shoulder_r2": {
+        "mean": 0.8461169242858887,
+        "std": 0.009572314458261256
+      },
+      "val_left_elbow_mse": {
+        "mean": 0.0014372184406965971,
+        "std": 6.144461276759243e-05
+      },
+      "val_left_elbow_rmse": {
+        "mean": 0.03790178863688641,
+        "std": 0.0008202797214274431
+      },
+      "val_left_elbow_mae": {
+        "mean": 0.02924802154302597,
+        "std": 0.0008014421472320217
+      },
+      "val_left_elbow_r2": {
+        "mean": 0.8952016711235047,
+        "std": 0.006488493670129211
+      },
+      "val_right_shoulder_mse": {
+        "mean": 0.0008542192867025733,
+        "std": 4.955659218450841e-05
+      },
+      "val_right_shoulder_rmse": {
+        "mean": 0.0292150711152686,
+        "std": 0.0008360062394334796
+      },
+      "val_right_shoulder_mae": {
+        "mean": 0.02205270081758499,
+        "std": 0.0006652181780130129
+      },
+      "val_right_shoulder_r2": {
+        "mean": 0.8692180514335632,
+        "std": 0.007467270827498871
+      },
+      "val_right_elbow_mse": {
+        "mean": 0.0014760931953787804,
+        "std": 0.0002008103667243887
+      },
+      "val_right_elbow_rmse": {
+        "mean": 0.038338409362176404,
+        "std": 0.0025019118603510867
+      },
+      "val_right_elbow_mae": {
+        "mean": 0.029134225472807883,
+        "std": 0.0014706873911601124
+      },
+      "val_right_elbow_r2": {
+        "mean": 0.9059633731842041,
+        "std": 0.013117923374032812
+      },
+      "val_left_hand_mse": {
+        "mean": 0.002437477372586727,
+        "std": 0.00020028214151369896
+      },
+      "val_left_hand_rmse": {
+        "mean": 0.049329031966570666,
+        "std": 0.0020307579441619025
+      },
+      "val_left_hand_mae": {
+        "mean": 0.03809351921081543,
+        "std": 0.00142834106395467
+      },
+      "val_left_hand_r2": {
+        "mean": 0.9069287538528442,
+        "std": 0.00849589901596225
+      },
+      "val_right_hand_mse": {
+        "mean": 0.0025829548947513105,
+        "std": 0.0002848451657393098
+      },
+      "val_right_hand_rmse": {
+        "mean": 0.05074828499066647,
+        "std": 0.002750720861411955
+      },
+      "val_right_hand_mae": {
+        "mean": 0.03841363936662674,
+        "std": 0.002124937228853285
+      },
+      "val_right_hand_r2": {
+        "mean": 0.9073208689689636,
+        "std": 0.009517532206691152
+      },
+      "val_left_hip_mse": {
+        "mean": 0.0011210123309865593,
+        "std": 0.00010183173742098504
+      },
+      "val_left_hip_rmse": {
+        "mean": 0.033447020884463614,
+        "std": 0.0015195805147520897
+      },
+      "val_left_hip_mae": {
+        "mean": 0.025383176654577254,
+        "std": 0.0012054206797058196
+      },
+      "val_left_hip_r2": {
+        "mean": 0.859985101222992,
+        "std": 0.005885325382351268
+      },
+      "val_right_hip_mse": {
+        "mean": 0.001134676137007773,
+        "std": 9.483295163122207e-05
+      },
+      "val_right_hip_rmse": {
+        "mean": 0.0336560016131237,
+        "std": 0.0013963138705847947
+      },
+      "val_right_hip_mae": {
+        "mean": 0.025119788944721222,
+        "std": 0.0011173710412350614
+      },
+      "val_right_hip_r2": {
+        "mean": 0.8637673616409302,
+        "std": 0.003510235878328545
+      },
+      "val_left_knee_mse": {
+        "mean": 0.0011672578053548931,
+        "std": 0.00012528477245718158
+      },
+      "val_left_knee_rmse": {
+        "mean": 0.034116045432715014,
+        "std": 0.0018311879717325208
+      },
+      "val_left_knee_mae": {
+        "mean": 0.02519553229212761,
+        "std": 0.0011097752067142444
+      },
+      "val_left_knee_r2": {
+        "mean": 0.8776954889297486,
+        "std": 0.010286755467685782
+      },
+      "val_right_knee_mse": {
+        "mean": 0.0011866023996844888,
+        "std": 7.787158740926628e-05
+      },
+      "val_right_knee_rmse": {
+        "mean": 0.03442825215169931,
+        "std": 0.0011392337176806243
+      },
+      "val_right_knee_mae": {
+        "mean": 0.025879384949803352,
+        "std": 0.0010045809299242799
+      },
+      "val_right_knee_r2": {
+        "mean": 0.8892869591712952,
+        "std": 0.009749549354033112
+      },
+      "val_left_foot_mse": {
+        "mean": 0.0014161731814965606,
+        "std": 0.0002382787487083214
+      },
+      "val_left_foot_rmse": {
+        "mean": 0.03749324851041512,
+        "std": 0.003229473274825898
+      },
+      "val_left_foot_mae": {
+        "mean": 0.02431708499789238,
+        "std": 0.0018353381445335986
+      },
+      "val_left_foot_r2": {
+        "mean": 0.681304931640625,
+        "std": 0.04559236843620631
+      },
+      "val_right_foot_mse": {
+        "mean": 0.0014553384855389595,
+        "std": 0.00026662835346986154
+      },
+      "val_right_foot_rmse": {
+        "mean": 0.03798233689706313,
+        "std": 0.0035609787105452327
+      },
+      "val_right_foot_mae": {
+        "mean": 0.025771848484873773,
+        "std": 0.0013895512607969651
+      },
+      "val_right_foot_r2": {
+        "mean": 0.6877912759780884,
+        "std": 0.039643226327803135
+      }
+    },
+    "best_fold": "2",
+    "best_fold_metrics": {
+      "fold": 2,
+      "train_metrics": {
+        "train_mse": 0.0012231831206008792,
+        "train_rmse": 0.03497403494881423,
+        "train_mae": 0.02582358382642269,
+        "train_r2": 0.9423038363456726,
+        "train_head_mse": 0.0013183472910895944,
+        "train_head_rmse": 0.03630905246752653,
+        "train_head_mae": 0.027137121185660362,
+        "train_head_r2": 0.8694280982017517,
+        "train_left_shoulder_mse": 0.0008099307306110859,
+        "train_left_shoulder_rmse": 0.02845928197637962,
+        "train_left_shoulder_mae": 0.021720649674534798,
+        "train_left_shoulder_r2": 0.8550264835357666,
+        "train_left_elbow_mse": 0.0012797850649803877,
+        "train_left_elbow_rmse": 0.03577408370567145,
+        "train_left_elbow_mae": 0.027500562369823456,
+        "train_left_elbow_r2": 0.9068232178688049,
+        "train_right_shoulder_mse": 0.0007866587257012725,
+        "train_right_shoulder_rmse": 0.02804743706118747,
+        "train_right_shoulder_mae": 0.02136330120265484,
+        "train_right_shoulder_r2": 0.8806198239326477,
+        "train_right_elbow_mse": 0.0012222615769132972,
+        "train_right_elbow_rmse": 0.03496085778285907,
+        "train_right_elbow_mae": 0.02709626406431198,
+        "train_right_elbow_r2": 0.9227523803710938,
+        "train_left_hand_mse": 0.0020251714158803225,
+        "train_left_hand_rmse": 0.045001904580587726,
+        "train_left_hand_mae": 0.03464556485414505,
+        "train_left_hand_r2": 0.9229216575622559,
+        "train_right_hand_mse": 0.0020998280961066484,
+        "train_right_hand_rmse": 0.0458238812859261,
+        "train_right_hand_mae": 0.03513193503022194,
+        "train_right_hand_r2": 0.9252057671546936,
+        "train_left_hip_mse": 0.0010171100730076432,
+        "train_left_hip_rmse": 0.03189216319109827,
+        "train_left_hip_mae": 0.02411387488245964,
+        "train_left_hip_r2": 0.8723220825195312,
+        "train_right_hip_mse": 0.001008846564218402,
+        "train_right_hip_rmse": 0.031762345067995246,
+        "train_right_hip_mae": 0.02377685159444809,
+        "train_right_hip_r2": 0.8784840106964111,
+        "train_left_knee_mse": 0.0009703963878564537,
+        "train_left_knee_rmse": 0.031151185978329197,
+        "train_left_knee_mae": 0.02302277460694313,
+        "train_left_knee_r2": 0.8986669778823853,
+        "train_right_knee_mse": 0.0009614750742912292,
+        "train_right_knee_rmse": 0.031007661541806553,
+        "train_right_knee_mae": 0.023296024650335312,
+        "train_right_knee_r2": 0.9097654819488525,
+        "train_left_foot_mse": 0.001125421724282205,
+        "train_left_foot_rmse": 0.03354730576785869,
+        "train_left_foot_mae": 0.022023992612957954,
+        "train_left_foot_r2": 0.7484492063522339,
+        "train_right_foot_mse": 0.0012761485995724797,
+        "train_right_foot_rmse": 0.03572322213312343,
+        "train_right_foot_mae": 0.024877630174160004,
+        "train_right_foot_r2": 0.7202651500701904
+      },
+      "val_metrics": {
+        "val_mse": 0.0013027341337874532,
+        "val_rmse": 0.0360934084534483,
+        "val_mae": 0.026622798293828964,
+        "val_r2": 0.9382195472717285,
+        "val_head_mse": 0.0013191502075642347,
+        "val_head_rmse": 0.036320107482828774,
+        "val_head_mae": 0.02728099189698696,
+        "val_head_r2": 0.8625984191894531,
+        "val_left_shoulder_mse": 0.0008245119825005531,
+        "val_left_shoulder_rmse": 0.028714316681762657,
+        "val_left_shoulder_mae": 0.021826883777976036,
+        "val_left_shoulder_r2": 0.8525588512420654,
+        "val_left_elbow_mse": 0.0013217764208093286,
+        "val_left_elbow_rmse": 0.03635624321639034,
+        "val_left_elbow_mae": 0.028389081358909607,
+        "val_left_elbow_r2": 0.904492974281311,
+        "val_right_shoulder_mse": 0.0008390280418097973,
+        "val_right_shoulder_rmse": 0.028965980767268994,
+        "val_right_shoulder_mae": 0.02204328030347824,
+        "val_right_shoulder_r2": 0.868829607963562,
+        "val_right_elbow_mse": 0.0013385032070800662,
+        "val_right_elbow_rmse": 0.03658556008974123,
+        "val_right_elbow_mae": 0.028251534327864647,
+        "val_right_elbow_r2": 0.9133638739585876,
+        "val_left_hand_mse": 0.0021428412292152643,
+        "val_left_hand_rmse": 0.04629083310133081,
+        "val_left_hand_mae": 0.036545149981975555,
+        "val_left_hand_r2": 0.9183455109596252,
+        "val_right_hand_mse": 0.00224794028326869,
+        "val_right_hand_rmse": 0.047412448610767724,
+        "val_right_hand_mae": 0.036398995667696,
+        "val_right_hand_r2": 0.9175090789794922,
+        "val_left_hip_mse": 0.0012204274535179138,
+        "val_left_hip_rmse": 0.034934616836569336,
+        "val_left_hip_mae": 0.025912651792168617,
+        "val_left_hip_r2": 0.850059986114502,
+        "val_right_hip_mse": 0.0011756425956264138,
+        "val_right_hip_rmse": 0.03428764494138397,
+        "val_right_hip_mae": 0.02494887448847294,
+        "val_right_hip_r2": 0.8602867126464844,
+        "val_left_knee_mse": 0.0010017354506999254,
+        "val_left_knee_rmse": 0.031650204591754626,
+        "val_left_knee_mae": 0.02339925803244114,
+        "val_left_knee_r2": 0.8931018114089966,
+        "val_right_knee_mse": 0.0010755346156656742,
+        "val_right_knee_rmse": 0.032795344420598393,
+        "val_right_knee_mae": 0.02410624735057354,
+        "val_right_knee_r2": 0.9032953977584839,
+        "val_left_foot_mse": 0.0010426408844068646,
+        "val_left_foot_rmse": 0.032289950207562486,
+        "val_left_foot_mae": 0.022291818633675575,
+        "val_left_foot_r2": 0.7546085119247437,
+        "val_right_foot_mse": 0.001385811367072165,
+        "val_right_foot_rmse": 0.0372264874393511,
+        "val_right_foot_mae": 0.024701634421944618,
+        "val_right_foot_r2": 0.719986081123352
+      },
+      "best_epoch": 100
+    }
+  },
+  "fold_details": [
+    {
+      "fold": 1,
+      "train_metrics": {
+        "train_mse": 0.0013819488231092691,
+        "train_rmse": 0.03717457226531691,
+        "train_mae": 0.027495548129081726,
+        "train_r2": 0.9346305727958679,
+        "train_head_mse": 0.0012759002856910229,
+        "train_head_rmse": 0.035719746439343926,
+        "train_head_mae": 0.027270976454019547,
+        "train_head_r2": 0.8717637062072754,
+        "train_left_shoulder_mse": 0.000807747186627239,
+        "train_left_shoulder_rmse": 0.02842089348748978,
+        "train_left_shoulder_mae": 0.021498186513781548,
+        "train_left_shoulder_r2": 0.8547090291976929,
+        "train_left_elbow_mse": 0.001397072453983128,
+        "train_left_elbow_rmse": 0.037377432415605115,
+        "train_left_elbow_mae": 0.02885683998465538,
+        "train_left_elbow_r2": 0.8988204002380371,
+        "train_right_shoulder_mse": 0.0007754864054732025,
+        "train_right_shoulder_rmse": 0.027847556544034566,
+        "train_right_shoulder_mae": 0.02135932445526123,
+        "train_right_shoulder_r2": 0.8811382055282593,
+        "train_right_elbow_mse": 0.0014850631123408675,
+        "train_right_elbow_rmse": 0.03853651660880713,
+        "train_right_elbow_mae": 0.029974034056067467,
+        "train_right_elbow_r2": 0.9058650732040405,
+        "train_left_hand_mse": 0.002399977995082736,
+        "train_left_hand_rmse": 0.048989570268402394,
+        "train_left_hand_mae": 0.03778063505887985,
+        "train_left_hand_r2": 0.9087083339691162,
+        "train_right_hand_mse": 0.0026166639290750027,
+        "train_right_hand_rmse": 0.05115333741873547,
+        "train_right_hand_mae": 0.03883263096213341,
+        "train_right_hand_r2": 0.9060562252998352,
+        "train_left_hip_mse": 0.0010326337069272995,
+        "train_left_hip_rmse": 0.032134618512241586,
+        "train_left_hip_mae": 0.024863967671990395,
+        "train_left_hip_r2": 0.8675686717033386,
+        "train_right_hip_mse": 0.0010274844244122505,
+        "train_right_hip_rmse": 0.032054397895019814,
+        "train_right_hip_mae": 0.0245201475918293,
+        "train_right_hip_r2": 0.8733859658241272,
+        "train_left_knee_mse": 0.001128386938944459,
+        "train_left_knee_rmse": 0.033591471223280156,
+        "train_left_knee_mae": 0.02539902552962303,
+        "train_left_knee_r2": 0.8810619115829468,
+        "train_right_knee_mse": 0.001101437141187489,
+        "train_right_knee_rmse": 0.03318790655024039,
+        "train_right_knee_mae": 0.02534629963338375,
+        "train_right_knee_r2": 0.8976070284843445,
+        "train_left_foot_mse": 0.0014775035670027137,
+        "train_left_foot_rmse": 0.03843830858665238,
+        "train_left_foot_mae": 0.02559483051300049,
+        "train_left_foot_r2": 0.6621752977371216,
+        "train_right_foot_mse": 0.0014399758074432611,
+        "train_right_foot_rmse": 0.037947013155757874,
+        "train_right_foot_mae": 0.02614520490169525,
+        "train_right_foot_r2": 0.6788966655731201
+      },
+      "val_metrics": {
+        "val_mse": 0.0015962395118549466,
+        "val_rmse": 0.039952966246011655,
+        "val_mae": 0.029158486053347588,
+        "val_r2": 0.9251495003700256,
+        "val_head_mse": 0.0013889566762372851,
+        "val_head_rmse": 0.03726870907661393,
+        "val_head_mae": 0.02925695665180683,
+        "val_head_r2": 0.8636744022369385,
+        "val_left_shoulder_mse": 0.0009298284421674907,
+        "val_left_shoulder_rmse": 0.030493088432749654,
+        "val_left_shoulder_mae": 0.02262694388628006,
+        "val_left_shoulder_r2": 0.8368875980377197,
+        "val_left_elbow_mse": 0.0014760642079636455,
+        "val_left_elbow_rmse": 0.03841958104877831,
+        "val_left_elbow_mae": 0.028697587549686432,
+        "val_left_elbow_r2": 0.891059160232544,
+        "val_right_shoulder_mse": 0.0009472249075770378,
+        "val_right_shoulder_rmse": 0.03077701914703628,
+        "val_right_shoulder_mae": 0.023084677755832672,
+        "val_right_shoulder_r2": 0.8577191829681396,
+        "val_right_elbow_mse": 0.0018745786510407925,
+        "val_right_elbow_rmse": 0.04329640459715786,
+        "val_right_elbow_mae": 0.03193986788392067,
+        "val_right_elbow_r2": 0.8801017999649048,
+        "val_left_hand_mse": 0.0024660832714289427,
+        "val_left_hand_rmse": 0.049659674499828756,
+        "val_left_hand_mae": 0.037693146616220474,
+        "val_left_hand_r2": 0.9058369398117065,
+        "val_right_hand_mse": 0.0030943455640226603,
+        "val_right_hand_rmse": 0.055626842118015836,
+        "val_right_hand_mae": 0.042157046496868134,
+        "val_right_hand_r2": 0.8900290727615356,
+        "val_left_hip_mse": 0.001254973583854735,
+        "val_left_hip_rmse": 0.03542560633009314,
+        "val_left_hip_mae": 0.027197016403079033,
+        "val_left_hip_r2": 0.8576123714447021,
+        "val_right_hip_mse": 0.0012930909870192409,
+        "val_right_hip_rmse": 0.03595957434424441,
+        "val_right_hip_mae": 0.026923391968011856,
+        "val_right_hip_r2": 0.8588879108428955,
+        "val_left_knee_mse": 0.001286438899114728,
+        "val_left_knee_rmse": 0.03586696110788769,
+        "val_left_knee_mae": 0.02658277191221714,
+        "val_left_knee_r2": 0.8677751421928406,
+        "val_right_knee_mse": 0.0012484438484534621,
+        "val_right_knee_rmse": 0.03533332489949767,
+        "val_right_knee_mae": 0.026852309703826904,
+        "val_right_knee_r2": 0.8835780620574951,
+        "val_left_foot_mse": 0.001700841123238206,
+        "val_left_foot_rmse": 0.04124125511230479,
+        "val_left_foot_mae": 0.027707437053322792,
+        "val_left_foot_r2": 0.6342741847038269,
+        "val_right_foot_mse": 0.0017902441322803497,
+        "val_right_foot_rmse": 0.042311276656233736,
+        "val_right_foot_mae": 0.02834113873541355,
+        "val_right_foot_r2": 0.6589809656143188
+      },
+      "best_epoch": 88
+    },
+    {
+      "fold": 2,
+      "train_metrics": {
+        "train_mse": 0.0012231831206008792,
+        "train_rmse": 0.03497403494881423,
+        "train_mae": 0.02582358382642269,
+        "train_r2": 0.9423038363456726,
+        "train_head_mse": 0.0013183472910895944,
+        "train_head_rmse": 0.03630905246752653,
+        "train_head_mae": 0.027137121185660362,
+        "train_head_r2": 0.8694280982017517,
+        "train_left_shoulder_mse": 0.0008099307306110859,
+        "train_left_shoulder_rmse": 0.02845928197637962,
+        "train_left_shoulder_mae": 0.021720649674534798,
+        "train_left_shoulder_r2": 0.8550264835357666,
+        "train_left_elbow_mse": 0.0012797850649803877,
+        "train_left_elbow_rmse": 0.03577408370567145,
+        "train_left_elbow_mae": 0.027500562369823456,
+        "train_left_elbow_r2": 0.9068232178688049,
+        "train_right_shoulder_mse": 0.0007866587257012725,
+        "train_right_shoulder_rmse": 0.02804743706118747,
+        "train_right_shoulder_mae": 0.02136330120265484,
+        "train_right_shoulder_r2": 0.8806198239326477,
+        "train_right_elbow_mse": 0.0012222615769132972,
+        "train_right_elbow_rmse": 0.03496085778285907,
+        "train_right_elbow_mae": 0.02709626406431198,
+        "train_right_elbow_r2": 0.9227523803710938,
+        "train_left_hand_mse": 0.0020251714158803225,
+        "train_left_hand_rmse": 0.045001904580587726,
+        "train_left_hand_mae": 0.03464556485414505,
+        "train_left_hand_r2": 0.9229216575622559,
+        "train_right_hand_mse": 0.0020998280961066484,
+        "train_right_hand_rmse": 0.0458238812859261,
+        "train_right_hand_mae": 0.03513193503022194,
+        "train_right_hand_r2": 0.9252057671546936,
+        "train_left_hip_mse": 0.0010171100730076432,
+        "train_left_hip_rmse": 0.03189216319109827,
+        "train_left_hip_mae": 0.02411387488245964,
+        "train_left_hip_r2": 0.8723220825195312,
+        "train_right_hip_mse": 0.001008846564218402,
+        "train_right_hip_rmse": 0.031762345067995246,
+        "train_right_hip_mae": 0.02377685159444809,
+        "train_right_hip_r2": 0.8784840106964111,
+        "train_left_knee_mse": 0.0009703963878564537,
+        "train_left_knee_rmse": 0.031151185978329197,
+        "train_left_knee_mae": 0.02302277460694313,
+        "train_left_knee_r2": 0.8986669778823853,
+        "train_right_knee_mse": 0.0009614750742912292,
+        "train_right_knee_rmse": 0.031007661541806553,
+        "train_right_knee_mae": 0.023296024650335312,
+        "train_right_knee_r2": 0.9097654819488525,
+        "train_left_foot_mse": 0.001125421724282205,
+        "train_left_foot_rmse": 0.03354730576785869,
+        "train_left_foot_mae": 0.022023992612957954,
+        "train_left_foot_r2": 0.7484492063522339,
+        "train_right_foot_mse": 0.0012761485995724797,
+        "train_right_foot_rmse": 0.03572322213312343,
+        "train_right_foot_mae": 0.024877630174160004,
+        "train_right_foot_r2": 0.7202651500701904
+      },
+      "val_metrics": {
+        "val_mse": 0.0013027341337874532,
+        "val_rmse": 0.0360934084534483,
+        "val_mae": 0.026622798293828964,
+        "val_r2": 0.9382195472717285,
+        "val_head_mse": 0.0013191502075642347,
+        "val_head_rmse": 0.036320107482828774,
+        "val_head_mae": 0.02728099189698696,
+        "val_head_r2": 0.8625984191894531,
+        "val_left_shoulder_mse": 0.0008245119825005531,
+        "val_left_shoulder_rmse": 0.028714316681762657,
+        "val_left_shoulder_mae": 0.021826883777976036,
+        "val_left_shoulder_r2": 0.8525588512420654,
+        "val_left_elbow_mse": 0.0013217764208093286,
+        "val_left_elbow_rmse": 0.03635624321639034,
+        "val_left_elbow_mae": 0.028389081358909607,
+        "val_left_elbow_r2": 0.904492974281311,
+        "val_right_shoulder_mse": 0.0008390280418097973,
+        "val_right_shoulder_rmse": 0.028965980767268994,
+        "val_right_shoulder_mae": 0.02204328030347824,
+        "val_right_shoulder_r2": 0.868829607963562,
+        "val_right_elbow_mse": 0.0013385032070800662,
+        "val_right_elbow_rmse": 0.03658556008974123,
+        "val_right_elbow_mae": 0.028251534327864647,
+        "val_right_elbow_r2": 0.9133638739585876,
+        "val_left_hand_mse": 0.0021428412292152643,
+        "val_left_hand_rmse": 0.04629083310133081,
+        "val_left_hand_mae": 0.036545149981975555,
+        "val_left_hand_r2": 0.9183455109596252,
+        "val_right_hand_mse": 0.00224794028326869,
+        "val_right_hand_rmse": 0.047412448610767724,
+        "val_right_hand_mae": 0.036398995667696,
+        "val_right_hand_r2": 0.9175090789794922,
+        "val_left_hip_mse": 0.0012204274535179138,
+        "val_left_hip_rmse": 0.034934616836569336,
+        "val_left_hip_mae": 0.025912651792168617,
+        "val_left_hip_r2": 0.850059986114502,
+        "val_right_hip_mse": 0.0011756425956264138,
+        "val_right_hip_rmse": 0.03428764494138397,
+        "val_right_hip_mae": 0.02494887448847294,
+        "val_right_hip_r2": 0.8602867126464844,
+        "val_left_knee_mse": 0.0010017354506999254,
+        "val_left_knee_rmse": 0.031650204591754626,
+        "val_left_knee_mae": 0.02339925803244114,
+        "val_left_knee_r2": 0.8931018114089966,
+        "val_right_knee_mse": 0.0010755346156656742,
+        "val_right_knee_rmse": 0.032795344420598393,
+        "val_right_knee_mae": 0.02410624735057354,
+        "val_right_knee_r2": 0.9032953977584839,
+        "val_left_foot_mse": 0.0010426408844068646,
+        "val_left_foot_rmse": 0.032289950207562486,
+        "val_left_foot_mae": 0.022291818633675575,
+        "val_left_foot_r2": 0.7546085119247437,
+        "val_right_foot_mse": 0.001385811367072165,
+        "val_right_foot_rmse": 0.0372264874393511,
+        "val_right_foot_mae": 0.024701634421944618,
+        "val_right_foot_r2": 0.719986081123352
+      },
+      "best_epoch": 100
+    },
+    {
+      "fold": 3,
+      "train_metrics": {
+        "train_mse": 0.001309339888393879,
+        "train_rmse": 0.036184801897949904,
+        "train_mae": 0.026832809671759605,
+        "train_r2": 0.9374852180480957,
+        "train_head_mse": 0.0012702005915343761,
+        "train_head_rmse": 0.035639873618383895,
+        "train_head_mae": 0.026888052001595497,
+        "train_head_r2": 0.872306764125824,
+        "train_left_shoulder_mse": 0.0007885821396484971,
+        "train_left_shoulder_rmse": 0.028081704714074913,
+        "train_left_shoulder_mae": 0.021589305251836777,
+        "train_left_shoulder_r2": 0.8589364886283875,
+        "train_left_elbow_mse": 0.0013947520637884736,
+        "train_left_elbow_rmse": 0.03734637952718407,
+        "train_left_elbow_mae": 0.029296480119228363,
+        "train_left_elbow_r2": 0.8974599838256836,
+        "train_right_shoulder_mse": 0.000750240811612457,
+        "train_right_shoulder_rmse": 0.02739052412080603,
+        "train_right_shoulder_mae": 0.021072138100862503,
+        "train_right_shoulder_r2": 0.8845542073249817,
+        "train_right_elbow_mse": 0.0013447487726807594,
+        "train_right_elbow_rmse": 0.03667081636234404,
+        "train_right_elbow_mae": 0.028193669393658638,
+        "train_right_elbow_r2": 0.9135274291038513,
+        "train_left_hand_mse": 0.0023894563782960176,
+        "train_left_hand_rmse": 0.04888206601910375,
+        "train_left_hand_mae": 0.0382351316511631,
+        "train_left_hand_r2": 0.90757155418396,
+        "train_right_hand_mse": 0.002519788220524788,
+        "train_right_hand_rmse": 0.050197492173661304,
+        "train_right_hand_mae": 0.03781503438949585,
+        "train_right_hand_r2": 0.9086599946022034,
+        "train_left_hip_mse": 0.0010313865495845675,
+        "train_left_hip_rmse": 0.03211520745043643,
+        "train_left_hip_mae": 0.024617668241262436,
+        "train_left_hip_r2": 0.8720797300338745,
+        "train_right_hip_mse": 0.0010337994899600744,
+        "train_right_hip_rmse": 0.03215275244765329,
+        "train_right_hip_mae": 0.02432132698595524,
+        "train_right_hip_r2": 0.8766830563545227,
+        "train_left_knee_mse": 0.0009940620511770248,
+        "train_left_knee_rmse": 0.031528749597423376,
+        "train_left_knee_mae": 0.023914828896522522,
+        "train_left_knee_r2": 0.8969559669494629,
+        "train_right_knee_mse": 0.001015220070257783,
+        "train_right_knee_rmse": 0.03186251826610356,
+        "train_right_knee_mae": 0.024601824581623077,
+        "train_right_knee_r2": 0.9060174822807312,
+        "train_left_foot_mse": 0.0012519133742898703,
+        "train_left_foot_rmse": 0.03538238791107619,
+        "train_left_foot_mae": 0.023582620546221733,
+        "train_left_foot_r2": 0.716678261756897,
+        "train_right_foot_mse": 0.001237267511896789,
+        "train_right_foot_rmse": 0.03517481360145053,
+        "train_right_foot_mae": 0.024698449298739433,
+        "train_right_foot_r2": 0.733269453048706
+      },
+      "val_metrics": {
+        "val_mse": 0.0014436279889196157,
+        "val_rmse": 0.037995104802061225,
+        "val_mae": 0.0280340276658535,
+        "val_r2": 0.9346422553062439,
+        "val_head_mse": 0.0013733917148783803,
+        "val_head_rmse": 0.0370592999782562,
+        "val_head_mae": 0.027961600571870804,
+        "val_head_r2": 0.8645088076591492,
+        "val_left_shoulder_mse": 0.0008267495431937277,
+        "val_left_shoulder_rmse": 0.028753252741102662,
+        "val_left_shoulder_mae": 0.022498609498143196,
+        "val_left_shoulder_r2": 0.850952684879303,
+        "val_left_elbow_mse": 0.0014977797400206327,
+        "val_left_elbow_rmse": 0.038701159414423654,
+        "val_left_elbow_mae": 0.030653031542897224,
+        "val_left_elbow_r2": 0.8955142498016357,
+        "val_right_shoulder_mse": 0.0008326507522724569,
+        "val_right_shoulder_rmse": 0.028855688386736796,
+        "val_right_shoulder_mae": 0.02244151011109352,
+        "val_right_shoulder_r2": 0.8756150603294373,
+        "val_right_elbow_mse": 0.001397577696479857,
+        "val_right_elbow_rmse": 0.03738419046174274,
+        "val_right_elbow_mae": 0.02896454744040966,
+        "val_right_elbow_r2": 0.9148768186569214,
+        "val_left_hand_mse": 0.002741666045039892,
+        "val_left_hand_rmse": 0.05236092097203689,
+        "val_left_hand_mae": 0.0407276526093483,
+        "val_left_hand_r2": 0.9015803933143616,
+        "val_right_hand_mse": 0.002644049236550927,
+        "val_right_hand_rmse": 0.05142031929647002,
+        "val_right_hand_mae": 0.03891498222947121,
+        "val_right_hand_r2": 0.9089844822883606,
+        "val_left_hip_mse": 0.0010498631745576859,
+        "val_left_hip_rmse": 0.03240159216084428,
+        "val_left_hip_mae": 0.025424083694815636,
+        "val_left_hip_r2": 0.8644111156463623,
+        "val_right_hip_mse": 0.0010889802360907197,
+        "val_right_hip_rmse": 0.032999700545470405,
+        "val_right_hip_mae": 0.025639615952968597,
+        "val_right_hip_r2": 0.8653236031532288,
+        "val_left_knee_mse": 0.0010836493456736207,
+        "val_left_knee_rmse": 0.03291882965224646,
+        "val_left_knee_mae": 0.025076430290937424,
+        "val_left_knee_r2": 0.8805466890335083,
+        "val_right_knee_mse": 0.0012372975470498204,
+        "val_right_knee_rmse": 0.03517524054004209,
+        "val_right_knee_mae": 0.026752233505249023,
+        "val_right_knee_r2": 0.8824847340583801,
+        "val_left_foot_mse": 0.001296233618631959,
+        "val_left_foot_rmse": 0.03600324455701123,
+        "val_left_foot_mae": 0.023237591609358788,
+        "val_left_foot_r2": 0.7099207043647766,
+        "val_right_foot_mse": 0.0016972750891000032,
+        "val_right_foot_rmse": 0.041197998605514846,
+        "val_right_foot_mae": 0.026150457561016083,
+        "val_right_foot_r2": 0.6341291666030884
+      },
+      "best_epoch": 100
+    },
+    {
+      "fold": 4,
+      "train_metrics": {
+        "train_mse": 0.0012081889435648918,
+        "train_rmse": 0.03475901240778989,
+        "train_mae": 0.025799937546253204,
+        "train_r2": 0.9427821040153503,
+        "train_head_mse": 0.0012404253939166665,
+        "train_head_rmse": 0.035219673393100434,
+        "train_head_mae": 0.026644131168723106,
+        "train_head_r2": 0.8748028874397278,
+        "train_left_shoulder_mse": 0.0007395746069960296,
+        "train_left_shoulder_rmse": 0.027195121014550194,
+        "train_left_shoulder_mae": 0.020838459953665733,
+        "train_left_shoulder_r2": 0.8668829202651978,
+        "train_left_elbow_mse": 0.001148787559941411,
+        "train_left_elbow_rmse": 0.03389376874797801,
+        "train_left_elbow_mae": 0.026418965309858322,
+        "train_left_elbow_r2": 0.9157516956329346,
+        "train_right_shoulder_mse": 0.0007497898186556995,
+        "train_right_shoulder_rmse": 0.027382290237591513,
+        "train_right_shoulder_mae": 0.020988766103982925,
+        "train_right_shoulder_r2": 0.8853524327278137,
+        "train_right_elbow_mse": 0.0011488201562315226,
+        "train_right_elbow_rmse": 0.03389424960419573,
+        "train_right_elbow_mae": 0.026122210547327995,
+        "train_right_elbow_r2": 0.9264993071556091,
+        "train_left_hand_mse": 0.001982173416763544,
+        "train_left_hand_rmse": 0.04452160617906259,
+        "train_left_hand_mae": 0.03455632925033569,
+        "train_left_hand_r2": 0.9240050911903381,
+        "train_right_hand_mse": 0.0020261534955352545,
+        "train_right_hand_rmse": 0.04501281479240389,
+        "train_right_hand_mae": 0.03416953235864639,
+        "train_right_hand_r2": 0.9269030690193176,
+        "train_left_hip_mse": 0.000995241804048419,
+        "train_left_hip_rmse": 0.03154745321017877,
+        "train_left_hip_mae": 0.024193106219172478,
+        "train_left_hip_r2": 0.8778557181358337,
+        "train_right_hip_mse": 0.0010158407967537642,
+        "train_right_hip_rmse": 0.03187225747815432,
+        "train_right_hip_mae": 0.024222254753112793,
+        "train_right_hip_r2": 0.8805620074272156,
+        "train_left_knee_mse": 0.001022525131702423,
+        "train_left_knee_rmse": 0.03197694687900055,
+        "train_left_knee_mae": 0.02397104911506176,
+        "train_left_knee_r2": 0.8926383256912231,
+        "train_right_knee_mse": 0.001057090936228633,
+        "train_right_knee_rmse": 0.03251293490641275,
+        "train_right_knee_mae": 0.024521201848983765,
+        "train_right_knee_r2": 0.9012427926063538,
+        "train_left_foot_mse": 0.0012527547078207135,
+        "train_left_foot_rmse": 0.03539427507126984,
+        "train_left_foot_mae": 0.02381114289164543,
+        "train_left_foot_r2": 0.7200137376785278,
+        "train_right_foot_mse": 0.0013272782089188695,
+        "train_right_foot_rmse": 0.036431829612563645,
+        "train_right_foot_mae": 0.02494203671813011,
+        "train_right_foot_r2": 0.7225971221923828
+      },
+      "val_metrics": {
+        "val_mse": 0.0013243411667644978,
+        "val_rmse": 0.03639149855068485,
+        "val_mae": 0.027093127369880676,
+        "val_r2": 0.9381906390190125,
+        "val_head_mse": 0.001293794484809041,
+        "val_head_rmse": 0.035969354801122594,
+        "val_head_mae": 0.027726639062166214,
+        "val_head_r2": 0.8750623464584351,
+        "val_left_shoulder_mse": 0.000814940663985908,
+        "val_left_shoulder_rmse": 0.02854716560336434,
+        "val_left_shoulder_mae": 0.021924827247858047,
+        "val_left_shoulder_r2": 0.8574617505073547,
+        "val_left_elbow_mse": 0.0014341652859002352,
+        "val_left_elbow_rmse": 0.03787037477897776,
+        "val_left_elbow_mae": 0.02956383489072323,
+        "val_left_elbow_r2": 0.8992534279823303,
+        "val_right_shoulder_mse": 0.0008001015521585941,
+        "val_right_shoulder_rmse": 0.028286066395994233,
+        "val_right_shoulder_mae": 0.021333569660782814,
+        "val_right_shoulder_r2": 0.8786875009536743,
+        "val_right_elbow_mse": 0.0014084238791838288,
+        "val_right_elbow_rmse": 0.03752897386265482,
+        "val_right_elbow_mae": 0.02880442515015602,
+        "val_right_elbow_r2": 0.9131644368171692,
+        "val_left_hand_mse": 0.0023192085791379213,
+        "val_left_hand_rmse": 0.048158162123755524,
+        "val_left_hand_mae": 0.03823533654212952,
+        "val_left_hand_r2": 0.9141455888748169,
+        "val_right_hand_mse": 0.00246411538682878,
+        "val_right_hand_rmse": 0.04963985683731149,
+        "val_right_hand_mae": 0.03823849931359291,
+        "val_right_hand_r2": 0.9140748977661133,
+        "val_left_hip_mse": 0.0009860540740191936,
+        "val_left_hip_rmse": 0.031401497958205654,
+        "val_left_hip_mae": 0.023538798093795776,
+        "val_left_hip_r2": 0.866960346698761,
+        "val_right_hip_mse": 0.0010114911710843444,
+        "val_right_hip_rmse": 0.03180394898568956,
+        "val_right_hip_mae": 0.02361137792468071,
+        "val_right_hip_r2": 0.8669277429580688,
+        "val_left_knee_mse": 0.0011286813532933593,
+        "val_left_knee_rmse": 0.033595853215737194,
+        "val_left_knee_mae": 0.02480640821158886,
+        "val_left_knee_r2": 0.8821997046470642,
+        "val_right_knee_mse": 0.0011097482638433576,
+        "val_right_knee_rmse": 0.03331288435190441,
+        "val_right_knee_mae": 0.025534188374876976,
+        "val_right_knee_r2": 0.8985145092010498,
+        "val_left_foot_mse": 0.0014040921814739704,
+        "val_left_foot_rmse": 0.037471218040970734,
+        "val_left_foot_mae": 0.024032775312662125,
+        "val_left_foot_r2": 0.669397234916687,
+        "val_right_foot_mse": 0.0010416185250505805,
+        "val_right_foot_rmse": 0.03227411540306845,
+        "val_right_foot_mae": 0.024859998375177383,
+        "val_right_foot_r2": 0.7433261871337891
+      },
+      "best_epoch": 100
+    },
+    {
+      "fold": 5,
+      "train_metrics": {
+        "train_mse": 0.0011753843864426017,
+        "train_rmse": 0.03428387939604562,
+        "train_mae": 0.025600021705031395,
+        "train_r2": 0.9452932476997375,
+        "train_head_mse": 0.0011574765667319298,
+        "train_head_rmse": 0.03402170728714139,
+        "train_head_mae": 0.02579522877931595,
+        "train_head_r2": 0.8851744532585144,
+        "train_left_shoulder_mse": 0.0007082968950271606,
+        "train_left_shoulder_rmse": 0.026613847805741293,
+        "train_left_shoulder_mae": 0.020317986607551575,
+        "train_left_shoulder_r2": 0.8745212554931641,
+        "train_left_elbow_mse": 0.0012214769376441836,
+        "train_left_elbow_rmse": 0.03494963429914802,
+        "train_left_elbow_mae": 0.027067428454756737,
+        "train_left_elbow_r2": 0.9127238392829895,
+        "train_right_shoulder_mse": 0.0006826941389590502,
+        "train_right_shoulder_rmse": 0.02612841631172946,
+        "train_right_shoulder_mae": 0.020013004541397095,
+        "train_right_shoulder_r2": 0.8965184092521667,
+        "train_right_elbow_mse": 0.00114097457844764,
+        "train_right_elbow_rmse": 0.033778315210318585,
+        "train_right_elbow_mae": 0.026264194399118423,
+        "train_right_elbow_r2": 0.9284695982933044,
+        "train_left_hand_mse": 0.0020473767071962357,
+        "train_left_hand_rmse": 0.04524794699426965,
+        "train_left_hand_mae": 0.0350591279566288,
+        "train_left_hand_r2": 0.9237194657325745,
+        "train_right_hand_mse": 0.0020956459920853376,
+        "train_right_hand_rmse": 0.04577822617888703,
+        "train_right_hand_mae": 0.03555098921060562,
+        "train_right_hand_r2": 0.9259665608406067,
+        "train_left_hip_mse": 0.00102097203489393,
+        "train_left_hip_rmse": 0.03195265301808177,
+        "train_left_hip_mae": 0.024539655074477196,
+        "train_left_hip_r2": 0.8728465437889099,
+        "train_right_hip_mse": 0.0010225260630249977,
+        "train_right_hip_rmse": 0.03197696144140337,
+        "train_right_hip_mae": 0.024290364235639572,
+        "train_right_hip_r2": 0.8770657777786255,
+        "train_left_knee_mse": 0.0009347607847303152,
+        "train_left_knee_rmse": 0.030573857864690794,
+        "train_left_knee_mae": 0.023046808317303658,
+        "train_left_knee_r2": 0.9009912014007568,
+        "train_right_knee_mse": 0.0009738652734085917,
+        "train_right_knee_rmse": 0.031206814534786978,
+        "train_right_knee_mae": 0.02386234700679779,
+        "train_right_knee_r2": 0.9100768566131592,
+        "train_left_foot_mse": 0.0010443556820973754,
+        "train_left_foot_rmse": 0.03231649241637117,
+        "train_left_foot_mae": 0.022323613986372948,
+        "train_left_foot_r2": 0.7628920078277588,
+        "train_right_foot_mse": 0.0012295758351683617,
+        "train_right_foot_rmse": 0.03506530814306872,
+        "train_right_foot_mae": 0.024669548496603966,
+        "train_right_foot_r2": 0.7398594617843628
+      },
+      "val_metrics": {
+        "val_mse": 0.001450598007068038,
+        "val_rmse": 0.03808671693737908,
+        "val_mae": 0.027082160115242004,
+        "val_r2": 0.9272032380104065,
+        "val_head_mse": 0.0015203714137896895,
+        "val_head_rmse": 0.03899194036964164,
+        "val_head_mae": 0.027666714042425156,
+        "val_head_r2": 0.8418023586273193,
+        "val_left_shoulder_mse": 0.0008912184275686741,
+        "val_left_shoulder_rmse": 0.02985328168842873,
+        "val_left_shoulder_mae": 0.022063031792640686,
+        "val_left_shoulder_r2": 0.8327237367630005,
+        "val_left_elbow_mse": 0.0014563065487891436,
+        "val_left_elbow_rmse": 0.038161584725861994,
+        "val_left_elbow_mae": 0.02893657237291336,
+        "val_left_elbow_r2": 0.8856885433197021,
+        "val_right_shoulder_mse": 0.0008520911796949804,
+        "val_right_shoulder_rmse": 0.029190600879306687,
+        "val_right_shoulder_mae": 0.02136046625673771,
+        "val_right_shoulder_r2": 0.8652389049530029,
+        "val_right_elbow_mse": 0.0013613825431093574,
+        "val_right_elbow_rmse": 0.03689691779958534,
+        "val_right_elbow_mae": 0.027710752561688423,
+        "val_right_elbow_r2": 0.9083099365234375,
+        "val_left_hand_mse": 0.002517587738111615,
+        "val_left_hand_rmse": 0.05017556913590134,
+        "val_left_hand_mae": 0.037266310304403305,
+        "val_left_hand_r2": 0.8947353363037109,
+        "val_right_hand_mse": 0.002464324003085494,
+        "val_right_hand_rmse": 0.04964195809076727,
+        "val_right_hand_mae": 0.03635867312550545,
+        "val_right_hand_r2": 0.9060068130493164,
+        "val_left_hip_mse": 0.0010937433689832687,
+        "val_left_hip_rmse": 0.03307179113660566,
+        "val_left_hip_mae": 0.024843333289027214,
+        "val_left_hip_r2": 0.8608816862106323,
+        "val_right_hip_mse": 0.0011041756952181458,
+        "val_right_hip_rmse": 0.03322913924883017,
+        "val_right_hip_mae": 0.024475684389472008,
+        "val_right_hip_r2": 0.8674108386039734,
+        "val_left_knee_mse": 0.0013357839779928327,
+        "val_left_knee_rmse": 0.03654837859594913,
+        "val_left_knee_mae": 0.026112793013453484,
+        "val_left_knee_r2": 0.864854097366333,
+        "val_right_knee_mse": 0.0012619877234101295,
+        "val_right_knee_rmse": 0.035524466546454005,
+        "val_right_knee_mae": 0.02615194581449032,
+        "val_right_knee_r2": 0.8785620927810669,
+        "val_left_foot_mse": 0.001637058099731803,
+        "val_left_foot_rmse": 0.04046057463422638,
+        "val_left_foot_mae": 0.02431580238044262,
+        "val_left_foot_r2": 0.6383240222930908,
+        "val_right_foot_mse": 0.001361743314191699,
+        "val_right_foot_rmse": 0.03690180638114751,
+        "val_right_foot_mae": 0.024806013330817223,
+        "val_right_foot_r2": 0.6825339794158936
+      },
+      "best_epoch": 100
+    }
+  ]
+}

A9/cv_results_different_models/conv1d_fold1_best.h5

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

A9/cv_results_different_models/conv1d_fold2_best.h5

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

A9/cv_results_different_models/conv1d_fold3_best.h5

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

A9/cv_results_different_models/conv1d_fold4_best.h5

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

A9/cv_results_different_models/conv1d_fold5_best.h5

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

A9/cv_results_different_models/dense_cv_results.json

@@ -0,0 +1,1185 @@
+{
+  "model_name": "dense",
+  "n_splits": 5,
+  "epochs": 100,
+  "batch_size": 32,
+  "patience": 15,
+  "aggregate_results": {
+    "n_folds": 5,
+    "train_metrics": {
+      "train_mse": {
+        "mean": 0.0033601932693272827,
+        "std": 2.1231350605266408e-05
+      },
+      "train_rmse": {
+        "mean": 0.05796688447156044,
+        "std": 0.00018323206064168216
+      },
+      "train_mae": {
+        "mean": 0.042676781117916104,
+        "std": 0.00012290258107191358
+      },
+      "train_r2": {
+        "mean": 0.8206043839454651,
+        "std": 0.0010221284382159754
+      },
+      "train_head_mse": {
+        "mean": 0.0022998914122581483,
+        "std": 1.159366166762918e-05
+      },
+      "train_head_rmse": {
+        "mean": 0.04795703053467808,
+        "std": 0.00012097336112686938
+      },
+      "train_head_mae": {
+        "mean": 0.036684338748455045,
+        "std": 0.0001712969266121724
+      },
+      "train_head_r2": {
+        "mean": 0.7534187793731689,
+        "std": 0.0008684047737452706
+      },
+      "train_left_shoulder_mse": {
+        "mean": 0.0016782312421128153,
+        "std": 9.653849402272084e-06
+      },
+      "train_left_shoulder_rmse": {
+        "mean": 0.04096605141700498,
+        "std": 0.00011778544952087769
+      },
+      "train_left_shoulder_mae": {
+        "mean": 0.03128715790808201,
+        "std": 4.9817426767142093e-05
+      },
+      "train_left_shoulder_r2": {
+        "mean": 0.6874250888824462,
+        "std": 0.0010424356222478357
+      },
+      "train_left_elbow_mse": {
+        "mean": 0.00252217473462224,
+        "std": 3.147737023708636e-05
+      },
+      "train_left_elbow_rmse": {
+        "mean": 0.05022027693123492,
+        "std": 0.00031387827626933214
+      },
+      "train_left_elbow_mae": {
+        "mean": 0.0387815423309803,
+        "std": 0.0002226675376374871
+      },
+      "train_left_elbow_r2": {
+        "mean": 0.8128067135810852,
+        "std": 0.002513635674506109
+      },
+      "train_right_shoulder_mse": {
+        "mean": 0.002087010256946087,
+        "std": 6.093256535867547e-06
+      },
+      "train_right_shoulder_rmse": {
+        "mean": 0.045683758680360786,
+        "std": 6.67066761838419e-05
+      },
+      "train_right_shoulder_mae": {
+        "mean": 0.03486936166882515,
+        "std": 0.00010564698760438084
+      },
+      "train_right_shoulder_r2": {
+        "mean": 0.663767421245575,
+        "std": 0.0003751647114027566
+      },
+      "train_right_elbow_mse": {
+        "mean": 0.0031260368414223194,
+        "std": 2.5807761158741632e-05
+      },
+      "train_right_elbow_rmse": {
+        "mean": 0.05591049606051642,
+        "std": 0.00023080660583303447
+      },
+      "train_right_elbow_mae": {
+        "mean": 0.043646573275327685,
+        "std": 0.00024977907374973526
+      },
+      "train_right_elbow_r2": {
+        "mean": 0.7954890847206115,
+        "std": 0.002096844504560054
+      },
+      "train_left_hand_mse": {
+        "mean": 0.004597078543156385,
+        "std": 4.237191671611994e-05
+      },
+      "train_left_hand_rmse": {
+        "mean": 0.0678010380848992,
+        "std": 0.0003126943658485542
+      },
+      "train_left_hand_mae": {
+        "mean": 0.05299176275730133,
+        "std": 0.0002475448657992511
+      },
+      "train_left_hand_r2": {
+        "mean": 0.8243893504142761,
+        "std": 0.0018724695599212082
+      },
+      "train_right_hand_mse": {
+        "mean": 0.005134709924459457,
+        "std": 4.6044790436081167e-05
+      },
+      "train_right_hand_rmse": {
+        "mean": 0.07165616903774563,
+        "std": 0.0003215016228314161
+      },
+      "train_right_hand_mae": {
+        "mean": 0.05529057458043098,
+        "std": 0.00027474344748210615
+      },
+      "train_right_hand_r2": {
+        "mean": 0.8113100528717041,
+        "std": 0.002370755014960844
+      },
+      "train_left_hip_mse": {
+        "mean": 0.003547054110094905,
+        "std": 4.105106055311062e-05
+      },
+      "train_left_hip_rmse": {
+        "mean": 0.05955615527330804,
+        "std": 0.000344208013456151
+      },
+      "train_left_hip_mae": {
+        "mean": 0.04241396933794021,
+        "std": 0.00023351184766985493
+      },
+      "train_left_hip_r2": {
+        "mean": 0.5250137329101563,
+        "std": 0.004569723933625248
+      },
+      "train_right_hip_mse": {
+        "mean": 0.0036967017222195864,
+        "std": 4.114248413758361e-05
+      },
+      "train_right_hip_rmse": {
+        "mean": 0.060799568440528994,
+        "std": 0.00033793440934126886
+      },
+      "train_right_hip_mae": {
+        "mean": 0.04293296560645103,
+        "std": 0.00022370983407835423
+      },
+      "train_right_hip_r2": {
+        "mean": 0.5241452097892761,
+        "std": 0.0043321531935909055
+      },
+      "train_left_knee_mse": {
+        "mean": 0.003278338070958853,
+        "std": 1.5616536732250873e-05
+      },
+      "train_left_knee_rmse": {
+        "mean": 0.057256610707970154,
+        "std": 0.00013638619728553646
+      },
+      "train_left_knee_mae": {
+        "mean": 0.04152142405509949,
+        "std": 9.434138077035921e-05
+      },
+      "train_left_knee_r2": {
+        "mean": 0.6538527846336365,
+        "std": 0.002013802655966968
+      },
+      "train_right_knee_mse": {
+        "mean": 0.003561070095747709,
+        "std": 1.997479425999645e-05
+      },
+      "train_right_knee_rmse": {
+        "mean": 0.05967446725156743,
+        "std": 0.00016749325155317405
+      },
+      "train_right_knee_mae": {
+        "mean": 0.043358738720417025,
+        "std": 4.656947179953403e-05
+      },
+      "train_right_knee_r2": {
+        "mean": 0.6670795321464539,
+        "std": 0.0014684650308102298
+      },
+      "train_left_foot_mse": {
+        "mean": 0.0038611240684986114,
+        "std": 1.9540192999610135e-05
+      },
+      "train_left_foot_rmse": {
+        "mean": 0.06213774527842637,
+        "std": 0.00015709936986646735
+      },
+      "train_left_foot_mae": {
+        "mean": 0.043849467486143115,
+        "std": 0.00021675202157805476
+      },
+      "train_left_foot_r2": {
+        "mean": 0.11411242485046387,
+        "std": 0.005768318046135041
+      },
+      "train_right_foot_mse": {
+        "mean": 0.004293091781437397,
+        "std": 5.113855924360809e-05
+      },
+      "train_right_foot_rmse": {
+        "mean": 0.06552051826223913,
+        "std": 0.00039175002869862253
+      },
+      "train_right_foot_mae": {
+        "mean": 0.04717030003666878,
+        "std": 0.0002449283857185567
+      },
+      "train_right_foot_r2": {
+        "mean": 0.06302239894866943,
+        "std": 0.0038587420734756397
+      }
+    },
+    "val_metrics": {
+      "val_mse": {
+        "mean": 0.0033705555833876133,
+        "std": 5.056937997275819e-05
+      },
+      "val_rmse": {
+        "mean": 0.05805485382955374,
+        "std": 0.00043535068251069904
+      },
+      "val_mae": {
+        "mean": 0.04273122474551201,
+        "std": 0.0002442470671369566
+      },
+      "val_r2": {
+        "mean": 0.8200456857681274,
+        "std": 0.002133309775574409
+      },
+      "val_head_mse": {
+        "mean": 0.0023080407176166774,
+        "std": 3.478312909697657e-05
+      },
+      "val_head_rmse": {
+        "mean": 0.048040703946626864,
+        "std": 0.0003626043701165469
+      },
+      "val_head_mae": {
+        "mean": 0.03672601506114006,
+        "std": 0.00032191875114513755
+      },
+      "val_head_r2": {
+        "mean": 0.7524327993392944,
+        "std": 0.006410956268217092
+      },
+      "val_left_shoulder_mse": {
+        "mean": 0.001682769996114075,
+        "std": 2.7980586338292917e-05
+      },
+      "val_left_shoulder_rmse": {
+        "mean": 0.04102017214666792,
+        "std": 0.00033981343676303476
+      },
+      "val_left_shoulder_mae": {
+        "mean": 0.03131417445838451,
+        "std": 0.00013620658830589937
+      },
+      "val_left_shoulder_r2": {
+        "mean": 0.686501431465149,
+        "std": 0.006398653069719818
+      },
+      "val_left_elbow_mse": {
+        "mean": 0.0025334582198411225,
+        "std": 4.3380187911035454e-05
+      },
+      "val_left_elbow_rmse": {
+        "mean": 0.05033160857446824,
+        "std": 0.00043289507693929216
+      },
+      "val_left_elbow_mae": {
+        "mean": 0.03887294754385948,
+        "std": 0.0004987753004188174
+      },
+      "val_left_elbow_r2": {
+        "mean": 0.8118991613388061,
+        "std": 0.004719111025473799
+      },
+      "val_right_shoulder_mse": {
+        "mean": 0.002093283785507083,
+        "std": 3.668371744835639e-05
+      },
+      "val_right_shoulder_rmse": {
+        "mean": 0.04575066766246747,
+        "std": 0.0004002423584982103
+      },
+      "val_right_shoulder_mae": {
+        "mean": 0.034924938529729846,
+        "std": 0.00031387277023511335
+      },
+      "val_right_shoulder_r2": {
+        "mean": 0.6626999855041504,
+        "std": 0.00679577356859904
+      },
+      "val_right_elbow_mse": {
+        "mean": 0.003148724930360913,
+        "std": 0.00010181391943579965
+      },
+      "val_right_elbow_rmse": {
+        "mean": 0.05610624232343727,
+        "std": 0.0009024980358145677
+      },
+      "val_right_elbow_mae": {
+        "mean": 0.04376704469323158,
+        "std": 0.0007019011630798436
+      },
+      "val_right_elbow_r2": {
+        "mean": 0.7939949631690979,
+        "std": 0.005160008353997292
+      },
+      "val_left_hand_mse": {
+        "mean": 0.004624483734369278,
+        "std": 0.00011429790325639228
+      },
+      "val_left_hand_rmse": {
+        "mean": 0.06799833632396218,
+        "std": 0.0008426099587616673
+      },
+      "val_left_hand_mae": {
+        "mean": 0.053160925209522245,
+        "std": 0.0008727623533700976
+      },
+      "val_left_hand_r2": {
+        "mean": 0.8232841730117798,
+        "std": 0.003936579927964769
+      },
+      "val_right_hand_mse": {
+        "mean": 0.005180982872843742,
+        "std": 0.00023824494500786727
+      },
+      "val_right_hand_rmse": {
+        "mean": 0.07196020414731058,
+        "std": 0.0016467822931787069
+      },
+      "val_right_hand_mae": {
+        "mean": 0.055471500009298326,
+        "std": 0.001105695951975313
+      },
+      "val_right_hand_r2": {
+        "mean": 0.8096322655677796,
+        "std": 0.005755466248241146
+      },
+      "val_left_hip_mse": {
+        "mean": 0.0035488918889313935,
+        "std": 0.00014362963145960313
+      },
+      "val_left_hip_rmse": {
+        "mean": 0.05956012954225322,
+        "std": 0.0012177265051763002
+      },
+      "val_left_hip_mae": {
+        "mean": 0.042415065318346025,
+        "std": 0.0004404678351278451
+      },
+      "val_left_hip_r2": {
+        "mean": 0.524887764453888,
+        "std": 0.0073232849000420775
+      },
+      "val_right_hip_mse": {
+        "mean": 0.0036986525636166336,
+        "std": 0.0001406508955900517
+      },
+      "val_right_hip_rmse": {
+        "mean": 0.060805341771030155,
+        "std": 0.0011674655133447995
+      },
+      "val_right_hip_mae": {
+        "mean": 0.04294445514678955,
+        "std": 0.0003679443549607489
+      },
+      "val_right_hip_r2": {
+        "mean": 0.5239951372146606,
+        "std": 0.00699937816938227
+      },
+      "val_left_knee_mse": {
+        "mean": 0.003280510427430272,
+        "std": 5.435508077362768e-05
+      },
+      "val_left_knee_rmse": {
+        "mean": 0.05727377077900499,
+        "std": 0.00047498229889876684
+      },
+      "val_left_knee_mae": {
+        "mean": 0.04152631908655167,
+        "std": 0.00026638391426399435
+      },
+      "val_left_knee_r2": {
+        "mean": 0.653571093082428,
+        "std": 0.004043296678426113
+      },
+      "val_right_knee_mse": {
+        "mean": 0.00356103521771729,
+        "std": 5.4381718046413296e-05
+      },
+      "val_right_knee_rmse": {
+        "mean": 0.05967266067623724,
+        "std": 0.00045693055920891163
+      },
+      "val_right_knee_mae": {
+        "mean": 0.043351197987794875,
+        "std": 0.00011392099789224755
+      },
+      "val_right_knee_r2": {
+        "mean": 0.667071270942688,
+        "std": 0.002340040991946899
+      },
+      "val_left_foot_mse": {
+        "mean": 0.0038627997972071173,
+        "std": 7.309684474133903e-05
+      },
+      "val_left_foot_rmse": {
+        "mean": 0.06214865459258538,
+        "std": 0.0005869663862886944
+      },
+      "val_left_foot_mae": {
+        "mean": 0.04386074244976044,
+        "std": 0.00041385505581637434
+      },
+      "val_left_foot_r2": {
+        "mean": 0.11349778175354004,
+        "std": 0.005054722120182058
+      },
+      "val_right_foot_mse": {
+        "mean": 0.004293589573353529,
+        "std": 0.0001887895579996689
+      },
+      "val_right_foot_rmse": {
+        "mean": 0.06551003021239464,
+        "std": 0.0014232058616626222
+      },
+      "val_right_foot_mae": {
+        "mean": 0.04717057421803474,
+        "std": 0.0007388068526269872
+      },
+      "val_right_foot_r2": {
+        "mean": 0.0629162311553955,
+        "std": 0.005532627120813937
+      }
+    },
+    "best_fold": "4",
+    "best_fold_metrics": {
+      "fold": 4,
+      "train_metrics": {
+        "train_mse": 0.00338557967916131,
+        "train_rmse": 0.0581857343269062,
+        "train_mae": 0.042836591601371765,
+        "train_r2": 0.8192280530929565,
+        "train_head_mse": 0.002308015478774905,
+        "train_head_rmse": 0.048041809695044846,
+        "train_head_mae": 0.03694631904363632,
+        "train_head_r2": 0.7535290122032166,
+        "train_left_shoulder_mse": 0.0016934756422415376,
+        "train_left_shoulder_rmse": 0.041151860738507774,
+        "train_left_shoulder_mae": 0.03137214481830597,
+        "train_left_shoulder_r2": 0.686500072479248,
+        "train_left_elbow_mse": 0.002544453600421548,
+        "train_left_elbow_rmse": 0.05044257725792317,
+        "train_left_elbow_mae": 0.03891715779900551,
+        "train_left_elbow_r2": 0.8115512728691101,
+        "train_right_shoulder_mse": 0.002092930721119046,
+        "train_right_shoulder_rmse": 0.04574855977098127,
+        "train_right_shoulder_mae": 0.03502652421593666,
+        "train_right_shoulder_r2": 0.6637051701545715,
+        "train_right_elbow_mse": 0.003157020080834627,
+        "train_right_elbow_rmse": 0.056187365847089034,
+        "train_right_elbow_mae": 0.04393244534730911,
+        "train_right_elbow_r2": 0.7938546538352966,
+        "train_left_hand_mse": 0.0046163578517735004,
+        "train_left_hand_rmse": 0.06794378449699061,
+        "train_left_hand_mae": 0.053150150924921036,
+        "train_left_hand_r2": 0.8237113952636719,
+        "train_right_hand_mse": 0.005182340741157532,
+        "train_right_hand_rmse": 0.07198847644698096,
+        "train_right_hand_mae": 0.05553752928972244,
+        "train_right_hand_r2": 0.8099189400672913,
+        "train_left_hip_mse": 0.0036041734274476767,
+        "train_left_hip_rmse": 0.06003476848833247,
+        "train_left_hip_mae": 0.04258947819471359,
+        "train_left_hip_r2": 0.5225877165794373,
+        "train_right_hip_mse": 0.0037565177772194147,
+        "train_right_hip_rmse": 0.061290437893846174,
+        "train_right_hip_mae": 0.04313835874199867,
+        "train_right_hip_r2": 0.5214295983314514,
+        "train_left_knee_mse": 0.003299058647826314,
+        "train_left_knee_rmse": 0.05743743246199567,
+        "train_left_knee_mae": 0.041542209684848785,
+        "train_left_knee_r2": 0.6514303684234619,
+        "train_right_knee_mse": 0.0035713003017008305,
+        "train_right_knee_rmse": 0.059760357275545385,
+        "train_right_knee_mae": 0.043322280049324036,
+        "train_right_knee_r2": 0.6661621332168579,
+        "train_left_foot_mse": 0.0038703158497810364,
+        "train_left_foot_rmse": 0.062211862613018074,
+        "train_left_foot_mae": 0.044056281447410583,
+        "train_left_foot_r2": 0.10739338397979736,
+        "train_right_foot_mse": 0.004316574428230524,
+        "train_right_foot_rmse": 0.06570064252524875,
+        "train_right_foot_mae": 0.04734484851360321,
+        "train_right_foot_r2": 0.058678507804870605
+      },
+      "val_metrics": {
+        "val_mse": 0.003306726925075054,
+        "val_rmse": 0.05750414702501946,
+        "val_mae": 0.04252117872238159,
+        "val_r2": 0.8235360383987427,
+        "val_head_mse": 0.0023349327966570854,
+        "val_head_rmse": 0.04832114233601153,
+        "val_head_mae": 0.036613162606954575,
+        "val_head_r2": 0.7455886602401733,
+        "val_left_shoulder_mse": 0.0016523125814273953,
+        "val_left_shoulder_rmse": 0.04064864796555225,
+        "val_left_shoulder_mae": 0.03136732429265976,
+        "val_left_shoulder_r2": 0.6845234632492065,
+        "val_left_elbow_mse": 0.0024524519685655832,
+        "val_left_elbow_rmse": 0.049522237111883215,
+        "val_left_elbow_mae": 0.03869336470961571,
+        "val_left_elbow_r2": 0.8164341449737549,
+        "val_right_shoulder_mse": 0.0020964399445801973,
+        "val_right_shoulder_rmse": 0.04578689708399333,
+        "val_right_shoulder_mae": 0.03487198054790497,
+        "val_right_shoulder_r2": 0.658574104309082,
+        "val_right_elbow_mse": 0.0030299422796815634,
+        "val_right_elbow_rmse": 0.055044911478551437,
+        "val_right_elbow_mae": 0.042953066527843475,
+        "val_right_elbow_r2": 0.8002650141716003,
+        "val_left_hand_mse": 0.0045779673382639885,
+        "val_left_hand_rmse": 0.0676606779323411,
+        "val_left_hand_mae": 0.052861377596855164,
+        "val_left_hand_r2": 0.8249056935310364,
+        "val_right_hand_mse": 0.004963692743331194,
+        "val_right_hand_rmse": 0.0704534792847819,
+        "val_right_hand_mae": 0.05416801944375038,
+        "val_right_hand_r2": 0.8162606358528137,
+        "val_left_hip_mse": 0.003288909560069442,
+        "val_left_hip_rmse": 0.05734901533652903,
+        "val_left_hip_mae": 0.04174300655722618,
+        "val_left_hip_r2": 0.539381742477417,
+        "val_right_hip_mse": 0.003440922126173973,
+        "val_right_hip_rmse": 0.058659373728109074,
+        "val_right_hip_mae": 0.04236496612429619,
+        "val_right_hip_r2": 0.5377569198608398,
+        "val_left_knee_mse": 0.003321874886751175,
+        "val_left_knee_rmse": 0.05763570843453887,
+        "val_left_knee_mae": 0.041816290467977524,
+        "val_left_knee_r2": 0.6501542329788208,
+        "val_right_knee_mse": 0.003562789410352707,
+        "val_right_knee_rmse": 0.05968910629547662,
+        "val_right_knee_mae": 0.04347795620560646,
+        "val_right_knee_r2": 0.6667060852050781,
+        "val_left_foot_mse": 0.003974426072090864,
+        "val_left_foot_rmse": 0.06304304935590334,
+        "val_left_foot_mae": 0.04460511729121208,
+        "val_left_foot_r2": 0.1065099835395813,
+        "val_right_foot_mse": 0.004290788900107145,
+        "val_right_foot_rmse": 0.06550411361210184,
+        "val_right_foot_mae": 0.047239646315574646,
+        "val_right_foot_r2": 0.060375213623046875
+      },
+      "best_epoch": 100
+    }
+  },
+  "fold_details": [
+    {
+      "fold": 1,
+      "train_metrics": {
+        "train_mse": 0.0033288642298430204,
+        "train_rmse": 0.05769631036594126,
+        "train_mae": 0.042489372193813324,
+        "train_r2": 0.821368396282196,
+        "train_head_mse": 0.0022800802253186703,
+        "train_head_rmse": 0.047750185605070376,
+        "train_head_mae": 0.03640752285718918,
+        "train_head_r2": 0.7535367608070374,
+        "train_left_shoulder_mse": 0.001670350320637226,
+        "train_left_shoulder_rmse": 0.04086991950857288,
+        "train_left_shoulder_mae": 0.031219342723488808,
+        "train_left_shoulder_r2": 0.6867160797119141,
+        "train_left_elbow_mse": 0.00250261346809566,
+        "train_left_elbow_rmse": 0.05002612785430889,
+        "train_left_elbow_mae": 0.038575612008571625,
+        "train_left_elbow_r2": 0.8126150369644165,
+        "train_right_shoulder_mse": 0.002077667275443673,
+        "train_right_shoulder_rmse": 0.04558143564482884,
+        "train_right_shoulder_mae": 0.034738775342702866,
+        "train_right_shoulder_r2": 0.6637824177742004,
+        "train_right_elbow_mse": 0.0031010140664875507,
+        "train_right_elbow_rmse": 0.055686749469578044,
+        "train_right_elbow_mae": 0.04321889206767082,
+        "train_right_elbow_r2": 0.7953526973724365,
+        "train_left_hand_mse": 0.004548154305666685,
+        "train_left_hand_rmse": 0.06744000523181093,
+        "train_left_hand_mae": 0.052732814103364944,
+        "train_left_hand_r2": 0.8242748975753784,
+        "train_right_hand_mse": 0.005085842218250036,
+        "train_right_hand_rmse": 0.07131509109753725,
+        "train_right_hand_mae": 0.05489575117826462,
+        "train_right_hand_r2": 0.8110116720199585,
+        "train_left_hip_mse": 0.0035118439700454473,
+        "train_left_hip_rmse": 0.05926081310651624,
+        "train_left_hip_mae": 0.04224887117743492,
+        "train_left_hip_r2": 0.5260076522827148,
+        "train_right_hip_mse": 0.0036550576332956553,
+        "train_right_hip_rmse": 0.060457072649076016,
+        "train_right_hip_mae": 0.04280484840273857,
+        "train_right_hip_r2": 0.5259175300598145,
+        "train_left_knee_mse": 0.003263751044869423,
+        "train_left_knee_rmse": 0.05712924859360066,
+        "train_left_knee_mae": 0.04161122441291809,
+        "train_left_knee_r2": 0.6538857221603394,
+        "train_right_knee_mse": 0.0035294904373586178,
+        "train_right_knee_rmse": 0.059409514703948035,
+        "train_right_knee_mae": 0.04337432235479355,
+        "train_right_knee_r2": 0.6690222024917603,
+        "train_left_foot_mse": 0.0038552661426365376,
+        "train_left_foot_rmse": 0.062090789515326164,
+        "train_left_foot_mae": 0.04373478516936302,
+        "train_left_foot_r2": 0.11631584167480469,
+        "train_right_foot_mse": 0.004194105975329876,
+        "train_right_foot_rmse": 0.0647619176316597,
+        "train_right_foot_mae": 0.04679904505610466,
+        "train_right_foot_r2": 0.0674058198928833
+      },
+      "val_metrics": {
+        "val_mse": 0.003430540906265378,
+        "val_rmse": 0.05857081957993569,
+        "val_mae": 0.04299088194966316,
+        "val_r2": 0.8205012083053589,
+        "val_head_mse": 0.0022709835320711136,
+        "val_head_rmse": 0.04765483744669699,
+        "val_head_mae": 0.036439597606658936,
+        "val_head_r2": 0.7641903758049011,
+        "val_left_shoulder_mse": 0.001664926647208631,
+        "val_left_shoulder_rmse": 0.04080351268222665,
+        "val_left_shoulder_mae": 0.031092200428247452,
+        "val_left_shoulder_r2": 0.6982802152633667,
+        "val_left_elbow_mse": 0.0025384698528796434,
+        "val_left_elbow_rmse": 0.05038322987740706,
+        "val_left_elbow_mae": 0.039413563907146454,
+        "val_left_elbow_r2": 0.8180035948753357,
+        "val_right_shoulder_mse": 0.0020624881144613028,
+        "val_right_shoulder_rmse": 0.04541462445579951,
+        "val_right_shoulder_mae": 0.034483540803194046,
+        "val_right_shoulder_r2": 0.6735011339187622,
+        "val_right_elbow_mse": 0.0031821071170270443,
+        "val_right_elbow_rmse": 0.05641016856052678,
+        "val_right_elbow_mae": 0.04411982372403145,
+        "val_right_elbow_r2": 0.7988283634185791,
+        "val_left_hand_mse": 0.004683224949985743,
+        "val_left_hand_rmse": 0.0684340920154987,
+        "val_left_hand_mae": 0.05391968414187431,
+        "val_left_hand_r2": 0.8288599252700806,
+        "val_right_hand_mse": 0.005285724997520447,
+        "val_right_hand_rmse": 0.07270299166829688,
+        "val_right_hand_mae": 0.056275032460689545,
+        "val_right_hand_r2": 0.8140676021575928,
+        "val_left_hip_mse": 0.003673278959468007,
+        "val_left_hip_rmse": 0.060607581699553124,
+        "val_left_hip_mae": 0.04266956076025963,
+        "val_left_hip_r2": 0.5230990648269653,
+        "val_right_hip_mse": 0.003843999933451414,
+        "val_right_hip_rmse": 0.061999999463317856,
+        "val_right_hip_mae": 0.04303989186882973,
+        "val_right_hip_r2": 0.5197290182113647,
+        "val_left_knee_mse": 0.003301575779914856,
+        "val_left_knee_rmse": 0.057459340232157693,
+        "val_left_knee_mae": 0.041666850447654724,
+        "val_left_knee_r2": 0.6572146415710449,
+        "val_right_knee_mse": 0.003631054190918803,
+        "val_right_knee_rmse": 0.06025822923816135,
+        "val_right_knee_mae": 0.04348578304052353,
+        "val_right_knee_r2": 0.6645854711532593,
+        "val_left_foot_mse": 0.0038062280509620905,
+        "val_left_foot_rmse": 0.06169463551202884,
+        "val_left_foot_mae": 0.04370458051562309,
+        "val_left_foot_r2": 0.1221928596496582,
+        "val_right_foot_mse": 0.004652968607842922,
+        "val_right_foot_rmse": 0.06821267190077605,
+        "val_right_foot_mae": 0.04857131466269493,
+        "val_right_foot_r2": 0.054261863231658936
+      },
+      "best_epoch": 94
+    },
+    {
+      "fold": 2,
+      "train_metrics": {
+        "train_mse": 0.0033421749249100685,
+        "train_rmse": 0.05781154664001015,
+        "train_mae": 0.04259030893445015,
+        "train_r2": 0.8220480680465698,
+        "train_head_mse": 0.0022948654368519783,
+        "train_head_rmse": 0.04790475380222696,
+        "train_head_mae": 0.036706097424030304,
+        "train_head_r2": 0.7541894912719727,
+        "train_left_shoulder_mse": 0.0016657008090987802,
+        "train_left_shoulder_rmse": 0.040812998041050354,
+        "train_left_shoulder_mae": 0.03127969428896904,
+        "train_left_shoulder_r2": 0.6893564462661743,
+        "train_left_elbow_mse": 0.00247130636125803,
+        "train_left_elbow_rmse": 0.049712235528670705,
+        "train_left_elbow_mae": 0.03852050006389618,
+        "train_left_elbow_r2": 0.8172999024391174,
+        "train_right_shoulder_mse": 0.002088117180392146,
+        "train_right_shoulder_rmse": 0.04569592082880206,
+        "train_right_shoulder_mae": 0.03493313118815422,
+        "train_right_shoulder_r2": 0.6644150614738464,
+        "train_right_elbow_mse": 0.003093591658398509,
+        "train_right_elbow_rmse": 0.05562006524985843,
+        "train_right_elbow_mae": 0.043559517711400986,
+        "train_right_elbow_r2": 0.7989544868469238,
+        "train_left_hand_mse": 0.004544170573353767,
+        "train_left_hand_rmse": 0.06741046338183537,
+        "train_left_hand_mae": 0.052653077989816666,
+        "train_left_hand_r2": 0.8278173208236694,
+        "train_right_hand_mse": 0.005073940847069025,
+        "train_right_hand_rmse": 0.07123160005972788,
+        "train_right_hand_mae": 0.05505149066448212,
+        "train_right_hand_r2": 0.8154028654098511,
+        "train_left_hip_mse": 0.0035115107893943787,
+        "train_left_hip_rmse": 0.059258001901805454,
+        "train_left_hip_mae": 0.04226788505911827,
+        "train_left_hip_r2": 0.5272599458694458,
+        "train_right_hip_mse": 0.003671022830531001,
+        "train_right_hip_rmse": 0.06058896624411908,
+        "train_right_hip_mae": 0.04283449426293373,
+        "train_right_hip_r2": 0.5254803895950317,
+        "train_left_knee_mse": 0.0032811949495226145,
+        "train_left_knee_rmse": 0.05728171566497126,
+        "train_left_knee_mae": 0.04155262932181358,
+        "train_left_knee_r2": 0.6541652083396912,
+        "train_right_knee_mse": 0.003579305484890938,
+        "train_right_knee_rmse": 0.05982729715515266,
+        "train_right_knee_mae": 0.04341093450784683,
+        "train_right_knee_r2": 0.6665565967559814,
+        "train_left_foot_mse": 0.003841390134766698,
+        "train_left_foot_rmse": 0.061978949126027444,
+        "train_left_foot_mae": 0.04373142123222351,
+        "train_left_foot_r2": 0.11529737710952759,
+        "train_right_foot_mse": 0.004332158714532852,
+        "train_right_foot_rmse": 0.0658191363854985,
+        "train_right_foot_mae": 0.047173138707876205,
+        "train_right_foot_r2": 0.06369549036026001
+      },
+      "val_metrics": {
+        "val_mse": 0.0033296269830316305,
+        "val_rmse": 0.0577029200563683,
+        "val_mae": 0.042460471391677856,
+        "val_r2": 0.8202908039093018,
+        "val_head_mse": 0.0023344275541603565,
+        "val_head_rmse": 0.04831591408801407,
+        "val_head_mae": 0.037180300801992416,
+        "val_head_r2": 0.7487781047821045,
+        "val_left_shoulder_mse": 0.0017336803721264005,
+        "val_left_shoulder_rmse": 0.04163748758182223,
+        "val_left_shoulder_mae": 0.03149484843015671,
+        "val_left_shoulder_r2": 0.6787618398666382,
+        "val_left_elbow_mse": 0.0025343711022287607,
+        "val_left_elbow_rmse": 0.050342537701517996,
+        "val_left_elbow_mae": 0.03798867389559746,
+        "val_left_elbow_r2": 0.8089213967323303,
+        "val_right_shoulder_mse": 0.002106068190187216,
+        "val_right_shoulder_rmse": 0.045891918571652854,
+        "val_right_shoulder_mae": 0.03504681587219238,
+        "val_right_shoulder_r2": 0.6573251485824585,
+        "val_right_elbow_mse": 0.0030806094873696566,
+        "val_right_elbow_rmse": 0.055503238530464655,
+        "val_right_elbow_mae": 0.043089643120765686,
+        "val_right_elbow_r2": 0.7929655313491821,
+        "val_left_hand_mse": 0.0044358158484101295,
+        "val_left_hand_rmse": 0.06660192075616235,
+        "val_left_hand_mae": 0.05159301310777664,
+        "val_left_hand_r2": 0.8248593211174011,
+        "val_right_hand_mse": 0.004913744051009417,
+        "val_right_hand_rmse": 0.07009810304858054,
+        "val_right_hand_mae": 0.05434848368167877,
+        "val_right_hand_r2": 0.8119187355041504,
+        "val_left_hip_mse": 0.0036543607711791992,
+        "val_left_hip_rmse": 0.06045130909400721,
+        "val_left_hip_mae": 0.04272257536649704,
+        "val_left_hip_r2": 0.5208033323287964,
+        "val_right_hip_mse": 0.0037675891071558,
+        "val_right_hip_rmse": 0.061380690018570175,
+        "val_right_hip_mae": 0.043110284954309464,
+        "val_right_hip_r2": 0.5229098796844482,
+        "val_left_knee_mse": 0.0032244077883660793,
+        "val_left_knee_rmse": 0.056783869085912764,
+        "val_left_knee_mae": 0.04140248894691467,
+        "val_left_knee_r2": 0.6571477651596069,
+        "val_right_knee_mse": 0.0034636787604540586,
+        "val_right_knee_rmse": 0.05885302677393966,
+        "val_right_knee_mae": 0.043218523263931274,
+        "val_right_knee_r2": 0.6715198755264282,
+        "val_left_foot_mse": 0.003920021466910839,
+        "val_left_foot_rmse": 0.06261007480358763,
+        "val_left_foot_mae": 0.04397715628147125,
+        "val_left_foot_r2": 0.11414450407028198,
+        "val_right_foot_mse": 0.004116378258913755,
+        "val_right_foot_rmse": 0.0641590076210173,
+        "val_right_foot_mae": 0.046813324093818665,
+        "val_right_foot_r2": 0.06435126066207886
+      },
+      "best_epoch": 95
+    },
+    {
+      "fold": 3,
+      "train_metrics": {
+        "train_mse": 0.003369332291185856,
+        "train_rmse": 0.05804594982585655,
+        "train_mae": 0.04271283745765686,
+        "train_r2": 0.8205861449241638,
+        "train_head_mse": 0.002303315559402108,
+        "train_head_rmse": 0.04799286988086989,
+        "train_head_mae": 0.036702729761600494,
+        "train_head_r2": 0.7540707588195801,
+        "train_left_shoulder_mse": 0.0016800654120743275,
+        "train_left_shoulder_rmse": 0.04098860100167274,
+        "train_left_shoulder_mae": 0.03126674145460129,
+        "train_left_shoulder_r2": 0.6876641511917114,
+        "train_left_elbow_mse": 0.0025334381498396397,
+        "train_left_elbow_rmse": 0.05033327080410769,
+        "train_left_elbow_mae": 0.03876824676990509,
+        "train_left_elbow_r2": 0.812888503074646,
+        "train_right_shoulder_mse": 0.0020936173386871815,
+        "train_right_shoulder_rmse": 0.04575606340898637,
+        "train_right_shoulder_mae": 0.034878529608249664,
+        "train_right_shoulder_r2": 0.6632436513900757,
+        "train_right_elbow_mse": 0.003126227529719472,
+        "train_right_elbow_rmse": 0.0559126777190958,
+        "train_right_elbow_mae": 0.043678708374500275,
+        "train_right_elbow_r2": 0.7963519096374512,
+        "train_left_hand_mse": 0.004636792931705713,
+        "train_left_hand_rmse": 0.06809400070274703,
+        "train_left_hand_mae": 0.053172171115875244,
+        "train_left_hand_r2": 0.8240291476249695,
+        "train_right_hand_mse": 0.005152894649654627,
+        "train_right_hand_rmse": 0.07178366561868116,
+        "train_right_hand_mae": 0.05536463484168053,
+        "train_right_hand_r2": 0.8119027018547058,
+        "train_left_hip_mse": 0.0035180430859327316,
+        "train_left_hip_rmse": 0.059313093714058886,
+        "train_left_hip_mae": 0.04217846319079399,
+        "train_left_hip_r2": 0.5313949584960938,
+        "train_right_hip_mse": 0.003665357129648328,
+        "train_right_hip_rmse": 0.06054219297026106,
+        "train_right_hip_mae": 0.04264214262366295,
+        "train_right_hip_r2": 0.5303313732147217,
+        "train_left_knee_mse": 0.003290101420134306,
+        "train_left_knee_rmse": 0.05735940568149487,
+        "train_left_knee_mae": 0.0415622852742672,
+        "train_left_knee_r2": 0.6524262428283691,
+        "train_right_knee_mse": 0.003579295240342617,
+        "train_right_knee_rmse": 0.059827211537415124,
+        "train_right_knee_mae": 0.04339810833334923,
+        "train_right_knee_r2": 0.6651303172111511,
+        "train_left_foot_mse": 0.0038944873958826065,
+        "train_left_foot_rmse": 0.06240582822046837,
+        "train_left_foot_mae": 0.044149402529001236,
+        "train_left_foot_r2": 0.10838127136230469,
+        "train_right_foot_mse": 0.004327682312577963,
+        "train_right_foot_rmse": 0.0657851222737935,
+        "train_right_foot_mae": 0.04750477522611618,
+        "train_right_foot_r2": 0.05844360589981079
+      },
+      "val_metrics": {
+        "val_mse": 0.0033580376766622066,
+        "val_rmse": 0.057948577865744096,
+        "val_mae": 0.042630866169929504,
+        "val_r2": 0.8188227415084839,
+        "val_head_mse": 0.002260400913655758,
+        "val_head_rmse": 0.047543673750098006,
+        "val_head_mae": 0.036370255053043365,
+        "val_head_r2": 0.753574013710022,
+        "val_left_shoulder_mse": 0.0016754131065681577,
+        "val_left_shoulder_rmse": 0.040931810448209566,
+        "val_left_shoulder_mae": 0.031245199963450432,
+        "val_left_shoulder_r2": 0.6856051683425903,
+        "val_left_elbow_mse": 0.0025721408892422915,
+        "val_left_elbow_rmse": 0.05071627834573719,
+        "val_left_elbow_mae": 0.039118167012929916,
+        "val_left_elbow_r2": 0.8053022623062134,
+        "val_right_shoulder_mse": 0.0020482882391661406,
+        "val_right_shoulder_rmse": 0.04525801850684739,
+        "val_right_shoulder_mae": 0.034786585718393326,
+        "val_right_shoulder_r2": 0.6678588390350342,
+        "val_right_elbow_mse": 0.003125025425106287,
+        "val_right_elbow_rmse": 0.05590192684609616,
+        "val_right_elbow_mae": 0.04380284994840622,
+        "val_right_elbow_r2": 0.7920116782188416,
+        "val_left_hand_mse": 0.004647344350814819,
+        "val_left_hand_rmse": 0.06817143353938525,
+        "val_left_hand_mae": 0.053527966141700745,
+        "val_left_hand_r2": 0.8177134394645691,
+        "val_right_hand_mse": 0.005168290808796883,
+        "val_right_hand_rmse": 0.07189082562328021,
+        "val_right_hand_mae": 0.05551164597272873,
+        "val_right_hand_r2": 0.804916501045227,
+        "val_left_hip_mse": 0.003498736536130309,
+        "val_left_hip_rmse": 0.059150118648488854,
+        "val_left_hip_mae": 0.042049966752529144,
+        "val_left_hip_r2": 0.5212969183921814,
+        "val_right_hip_mse": 0.003667147597298026,
+        "val_right_hip_rmse": 0.06055697810573135,
+        "val_right_hip_mae": 0.042747288942337036,
+        "val_right_hip_r2": 0.5191477537155151,
+        "val_left_knee_mse": 0.0033460857812315226,
+        "val_left_knee_rmse": 0.057845360930946936,
+        "val_left_knee_mae": 0.041680075228214264,
+        "val_left_knee_r2": 0.6473819017410278,
+        "val_right_knee_mse": 0.003579974640160799,
+        "val_right_knee_rmse": 0.05983288928474705,
+        "val_right_knee_mae": 0.043237727135419846,
+        "val_right_knee_r2": 0.6662802696228027,
+        "val_left_foot_mse": 0.0038338431622833014,
+        "val_left_foot_rmse": 0.0619180358399982,
+        "val_left_foot_mae": 0.04359814524650574,
+        "val_left_foot_r2": 0.11257421970367432,
+        "val_right_foot_mse": 0.00423180079087615,
+        "val_right_foot_rmse": 0.06505229274111828,
+        "val_right_foot_mae": 0.046525388956069946,
+        "val_right_foot_r2": 0.0644795298576355
+      },
+      "best_epoch": 99
+    },
+    {
+      "fold": 4,
+      "train_metrics": {
+        "train_mse": 0.00338557967916131,
+        "train_rmse": 0.0581857343269062,
+        "train_mae": 0.042836591601371765,
+        "train_r2": 0.8192280530929565,
+        "train_head_mse": 0.002308015478774905,
+        "train_head_rmse": 0.048041809695044846,
+        "train_head_mae": 0.03694631904363632,
+        "train_head_r2": 0.7535290122032166,
+        "train_left_shoulder_mse": 0.0016934756422415376,
+        "train_left_shoulder_rmse": 0.041151860738507774,
+        "train_left_shoulder_mae": 0.03137214481830597,
+        "train_left_shoulder_r2": 0.686500072479248,
+        "train_left_elbow_mse": 0.002544453600421548,
+        "train_left_elbow_rmse": 0.05044257725792317,
+        "train_left_elbow_mae": 0.03891715779900551,
+        "train_left_elbow_r2": 0.8115512728691101,
+        "train_right_shoulder_mse": 0.002092930721119046,
+        "train_right_shoulder_rmse": 0.04574855977098127,
+        "train_right_shoulder_mae": 0.03502652421593666,
+        "train_right_shoulder_r2": 0.6637051701545715,
+        "train_right_elbow_mse": 0.003157020080834627,
+        "train_right_elbow_rmse": 0.056187365847089034,
+        "train_right_elbow_mae": 0.04393244534730911,
+        "train_right_elbow_r2": 0.7938546538352966,
+        "train_left_hand_mse": 0.0046163578517735004,
+        "train_left_hand_rmse": 0.06794378449699061,
+        "train_left_hand_mae": 0.053150150924921036,
+        "train_left_hand_r2": 0.8237113952636719,
+        "train_right_hand_mse": 0.005182340741157532,
+        "train_right_hand_rmse": 0.07198847644698096,
+        "train_right_hand_mae": 0.05553752928972244,
+        "train_right_hand_r2": 0.8099189400672913,
+        "train_left_hip_mse": 0.0036041734274476767,
+        "train_left_hip_rmse": 0.06003476848833247,
+        "train_left_hip_mae": 0.04258947819471359,
+        "train_left_hip_r2": 0.5225877165794373,
+        "train_right_hip_mse": 0.0037565177772194147,
+        "train_right_hip_rmse": 0.061290437893846174,
+        "train_right_hip_mae": 0.04313835874199867,
+        "train_right_hip_r2": 0.5214295983314514,
+        "train_left_knee_mse": 0.003299058647826314,
+        "train_left_knee_rmse": 0.05743743246199567,
+        "train_left_knee_mae": 0.041542209684848785,
+        "train_left_knee_r2": 0.6514303684234619,
+        "train_right_knee_mse": 0.0035713003017008305,
+        "train_right_knee_rmse": 0.059760357275545385,
+        "train_right_knee_mae": 0.043322280049324036,
+        "train_right_knee_r2": 0.6661621332168579,
+        "train_left_foot_mse": 0.0038703158497810364,
+        "train_left_foot_rmse": 0.062211862613018074,
+        "train_left_foot_mae": 0.044056281447410583,
+        "train_left_foot_r2": 0.10739338397979736,
+        "train_right_foot_mse": 0.004316574428230524,
+        "train_right_foot_rmse": 0.06570064252524875,
+        "train_right_foot_mae": 0.04734484851360321,
+        "train_right_foot_r2": 0.058678507804870605
+      },
+      "val_metrics": {
+        "val_mse": 0.003306726925075054,
+        "val_rmse": 0.05750414702501946,
+        "val_mae": 0.04252117872238159,
+        "val_r2": 0.8235360383987427,
+        "val_head_mse": 0.0023349327966570854,
+        "val_head_rmse": 0.04832114233601153,
+        "val_head_mae": 0.036613162606954575,
+        "val_head_r2": 0.7455886602401733,
+        "val_left_shoulder_mse": 0.0016523125814273953,
+        "val_left_shoulder_rmse": 0.04064864796555225,
+        "val_left_shoulder_mae": 0.03136732429265976,
+        "val_left_shoulder_r2": 0.6845234632492065,
+        "val_left_elbow_mse": 0.0024524519685655832,
+        "val_left_elbow_rmse": 0.049522237111883215,
+        "val_left_elbow_mae": 0.03869336470961571,
+        "val_left_elbow_r2": 0.8164341449737549,
+        "val_right_shoulder_mse": 0.0020964399445801973,
+        "val_right_shoulder_rmse": 0.04578689708399333,
+        "val_right_shoulder_mae": 0.03487198054790497,
+        "val_right_shoulder_r2": 0.658574104309082,
+        "val_right_elbow_mse": 0.0030299422796815634,
+        "val_right_elbow_rmse": 0.055044911478551437,
+        "val_right_elbow_mae": 0.042953066527843475,
+        "val_right_elbow_r2": 0.8002650141716003,
+        "val_left_hand_mse": 0.0045779673382639885,
+        "val_left_hand_rmse": 0.0676606779323411,
+        "val_left_hand_mae": 0.052861377596855164,
+        "val_left_hand_r2": 0.8249056935310364,
+        "val_right_hand_mse": 0.004963692743331194,
+        "val_right_hand_rmse": 0.0704534792847819,
+        "val_right_hand_mae": 0.05416801944375038,
+        "val_right_hand_r2": 0.8162606358528137,
+        "val_left_hip_mse": 0.003288909560069442,
+        "val_left_hip_rmse": 0.05734901533652903,
+        "val_left_hip_mae": 0.04174300655722618,
+        "val_left_hip_r2": 0.539381742477417,
+        "val_right_hip_mse": 0.003440922126173973,
+        "val_right_hip_rmse": 0.058659373728109074,
+        "val_right_hip_mae": 0.04236496612429619,
+        "val_right_hip_r2": 0.5377569198608398,
+        "val_left_knee_mse": 0.003321874886751175,
+        "val_left_knee_rmse": 0.05763570843453887,
+        "val_left_knee_mae": 0.041816290467977524,
+        "val_left_knee_r2": 0.6501542329788208,
+        "val_right_knee_mse": 0.003562789410352707,
+        "val_right_knee_rmse": 0.05968910629547662,
+        "val_right_knee_mae": 0.04347795620560646,
+        "val_right_knee_r2": 0.6667060852050781,
+        "val_left_foot_mse": 0.003974426072090864,
+        "val_left_foot_rmse": 0.06304304935590334,
+        "val_left_foot_mae": 0.04460511729121208,
+        "val_left_foot_r2": 0.1065099835395813,
+        "val_right_foot_mse": 0.004290788900107145,
+        "val_right_foot_rmse": 0.06550411361210184,
+        "val_right_foot_mae": 0.047239646315574646,
+        "val_right_foot_r2": 0.060375213623046875
+      },
+      "best_epoch": 100
+    },
+    {
+      "fold": 5,
+      "train_metrics": {
+        "train_mse": 0.0033750152215361595,
+        "train_rmse": 0.058094881199088094,
+        "train_mae": 0.04275479540228844,
+        "train_r2": 0.8197912573814392,
+        "train_head_mse": 0.002313180360943079,
+        "train_head_rmse": 0.048095533690178334,
+        "train_head_mae": 0.03665902465581894,
+        "train_head_r2": 0.7517678737640381,
+        "train_left_shoulder_mse": 0.0016815640265122056,
+        "train_left_shoulder_rmse": 0.0410068777952212,
+        "train_left_shoulder_mae": 0.03129786625504494,
+        "train_left_shoulder_r2": 0.6868886947631836,
+        "train_left_elbow_mse": 0.0025590620934963226,
+        "train_left_elbow_rmse": 0.05058717321116414,
+        "train_left_elbow_mae": 0.0391261950135231,
+        "train_left_elbow_r2": 0.809678852558136,
+        "train_right_shoulder_mse": 0.0020827187690883875,
+        "train_right_shoulder_rmse": 0.04563681374820538,
+        "train_right_shoulder_mae": 0.034769847989082336,
+        "train_right_shoulder_r2": 0.6636908054351807,
+        "train_right_elbow_mse": 0.0031523308716714382,
+        "train_right_elbow_rmse": 0.05614562201696084,
+        "train_right_elbow_mae": 0.04384330287575722,
+        "train_right_elbow_r2": 0.7929316759109497,
+        "train_left_hand_mse": 0.004639917053282261,
+        "train_left_hand_rmse": 0.06811693661111208,
+        "train_left_hand_mae": 0.05325059965252876,
+        "train_left_hand_r2": 0.8221139907836914,
+        "train_right_hand_mse": 0.005178531166166067,
+        "train_right_hand_rmse": 0.07196201196580086,
+        "train_right_hand_mae": 0.05560346692800522,
+        "train_right_hand_r2": 0.8083140850067139,
+        "train_left_hip_mse": 0.00358969927765429,
+        "train_left_hip_rmse": 0.05991409915582717,
+        "train_left_hip_mae": 0.042785149067640305,
+        "train_left_hip_r2": 0.5178183913230896,
+        "train_right_hip_mse": 0.003735553240403533,
+        "train_right_hip_rmse": 0.06111917244534266,
+        "train_right_hip_mae": 0.04324498400092125,
+        "train_right_hip_r2": 0.5175671577453613,
+        "train_left_knee_mse": 0.0032575842924416065,
+        "train_left_knee_rmse": 0.057075251137788316,
+        "train_left_knee_mae": 0.04133877158164978,
+        "train_left_knee_r2": 0.6573563814163208,
+        "train_right_knee_mse": 0.0035459590144455433,
+        "train_right_knee_rmse": 0.059547955585775934,
+        "train_right_knee_mae": 0.04328804835677147,
+        "train_right_knee_r2": 0.6685264110565186,
+        "train_left_foot_mse": 0.003844160819426179,
+        "train_left_foot_rmse": 0.06200129691729181,
+        "train_left_foot_mae": 0.04357544705271721,
+        "train_left_foot_r2": 0.12317425012588501,
+        "train_right_foot_mse": 0.00429493747651577,
+        "train_right_foot_rmse": 0.0655357724949952,
+        "train_right_foot_mae": 0.04702969267964363,
+        "train_right_foot_r2": 0.06688857078552246
+      },
+      "val_metrics": {
+        "val_mse": 0.003427845425903797,
+        "val_rmse": 0.058547804620701166,
+        "val_mae": 0.04305272549390793,
+        "val_r2": 0.81707763671875,
+        "val_head_mse": 0.002339458791539073,
+        "val_head_rmse": 0.04836795211231372,
+        "val_head_mae": 0.03702675923705101,
+        "val_head_r2": 0.7500328421592712,
+        "val_left_shoulder_mse": 0.0016875172732397914,
+        "val_left_shoulder_rmse": 0.041079402055528895,
+        "val_left_shoulder_mae": 0.03137129917740822,
+        "val_left_shoulder_r2": 0.6853364706039429,
+        "val_left_elbow_mse": 0.0025698572862893343,
+        "val_left_elbow_rmse": 0.05069375983579571,
+        "val_left_elbow_mae": 0.039150968194007874,
+        "val_left_elbow_r2": 0.8108344078063965,
+        "val_right_shoulder_mse": 0.0021531344391405582,
+        "val_right_shoulder_rmse": 0.046401879694044273,
+        "val_right_shoulder_mae": 0.03543576970696449,
+        "val_right_shoulder_r2": 0.656240701675415,
+        "val_right_elbow_mse": 0.003325940342620015,
+        "val_right_elbow_rmse": 0.05767096620154734,
+        "val_right_elbow_mae": 0.044869840145111084,
+        "val_right_elbow_r2": 0.7859042286872864,
+        "val_left_hand_mse": 0.00477806618437171,
+        "val_left_hand_rmse": 0.06912355737642349,
+        "val_left_hand_mae": 0.05390258505940437,
+        "val_left_hand_r2": 0.8200824856758118,
+        "val_right_hand_mse": 0.005573461763560772,
+        "val_right_hand_rmse": 0.07465562111161338,
+        "val_right_hand_mae": 0.057054318487644196,
+        "val_right_hand_r2": 0.8009978532791138,
+        "val_left_hip_mse": 0.003629173617810011,
+        "val_left_hip_rmse": 0.060242622932687874,
+        "val_left_hip_mae": 0.042890217155218124,
+        "val_left_hip_r2": 0.5198577642440796,
+        "val_right_hip_mse": 0.003773604054003954,
+        "val_right_hip_rmse": 0.0614296675394223,
+        "val_right_hip_mae": 0.04345984384417534,
+        "val_right_hip_r2": 0.5204321146011353,
+        "val_left_knee_mse": 0.0032086079008877277,
+        "val_left_knee_rmse": 0.056644575211468644,
+        "val_left_knee_mae": 0.04106589034199715,
+        "val_left_knee_r2": 0.6559569239616394,
+        "val_right_knee_mse": 0.003567679086700082,
+        "val_right_knee_rmse": 0.059730051788861545,
+        "val_right_knee_mae": 0.04333600029349327,
+        "val_right_knee_r2": 0.6662646532058716,
+        "val_left_foot_mse": 0.0037794802337884903,
+        "val_left_foot_rmse": 0.061477477451408906,
+        "val_left_foot_mae": 0.04341871291399002,
+        "val_left_foot_r2": 0.1120673418045044,
+        "val_right_foot_mse": 0.004176011309027672,
+        "val_right_foot_rmse": 0.06462206518695972,
+        "val_right_foot_mae": 0.04670319706201553,
+        "val_right_foot_r2": 0.07111328840255737
+      },
+      "best_epoch": 94
+    }
+  ]
+}

A9/cv_results_different_models/dense_fold1_best.h5

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

A9/cv_results_different_models/dense_fold2_best.h5

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

A9/cv_results_different_models/dense_fold3_best.h5

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

A9/cv_results_different_models/dense_fold4_best.h5

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

A9/cv_results_different_models/dense_fold5_best.h5

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

A9/cv_results_different_models/gru_cv_results.json

@@ -0,0 +1,1185 @@
+{
+  "model_name": "gru",
+  "n_splits": 5,
+  "epochs": 100,
+  "batch_size": 32,
+  "patience": 15,
+  "aggregate_results": {
+    "n_folds": 5,
+    "train_metrics": {
+      "train_mse": {
+        "mean": 0.005407693702727557,
+        "std": 0.00034858612739333255
+      },
+      "train_rmse": {
+        "mean": 0.07349830057038524,
+        "std": 0.0023861089650025077
+      },
+      "train_mae": {
+        "mean": 0.055533002316951754,
+        "std": 0.0017907277654893338
+      },
+      "train_r2": {
+        "mean": 0.7445639252662659,
+        "std": 0.017450332270188204
+      },
+      "train_head_mse": {
+        "mean": 0.003639672975987196,
+        "std": 0.00036887021069148177
+      },
+      "train_head_rmse": {
+        "mean": 0.06025070746844426,
+        "std": 0.003086296411421074
+      },
+      "train_head_mae": {
+        "mean": 0.048755767196416853,
+        "std": 0.002388071202542456
+      },
+      "train_head_r2": {
+        "mean": 0.6357398271560669,
+        "std": 0.03831699296132063
+      },
+      "train_left_shoulder_mse": {
+        "mean": 0.0025552061386406423,
+        "std": 0.0001960715879789471
+      },
+      "train_left_shoulder_rmse": {
+        "mean": 0.050511316714818934,
+        "std": 0.0019526961811547167
+      },
+      "train_left_shoulder_mae": {
+        "mean": 0.04056161493062973,
+        "std": 0.0017175429485392813
+      },
+      "train_left_shoulder_r2": {
+        "mean": 0.5424924612045288,
+        "std": 0.03733035924023864
+      },
+      "train_left_elbow_mse": {
+        "mean": 0.005528650805354118,
+        "std": 0.0005174851381653876
+      },
+      "train_left_elbow_rmse": {
+        "mean": 0.07427199859780587,
+        "std": 0.0035101324250863893
+      },
+      "train_left_elbow_mae": {
+        "mean": 0.058802831918001175,
+        "std": 0.0029215671789968225
+      },
+      "train_left_elbow_r2": {
+        "mean": 0.5977848291397094,
+        "std": 0.04015402554312586
+      },
+      "train_right_shoulder_mse": {
+        "mean": 0.003260692209005356,
+        "std": 0.000245567189183323
+      },
+      "train_right_shoulder_rmse": {
+        "mean": 0.05706132873612546,
+        "std": 0.002167249840968499
+      },
+      "train_right_shoulder_mae": {
+        "mean": 0.04581187441945076,
+        "std": 0.001869689593966545
+      },
+      "train_right_shoulder_r2": {
+        "mean": 0.502107834815979,
+        "std": 0.038190351992043715
+      },
+      "train_right_elbow_mse": {
+        "mean": 0.006750583741813898,
+        "std": 0.0006125972349087826
+      },
+      "train_right_elbow_rmse": {
+        "mean": 0.0820750485638277,
+        "std": 0.003777584553550222
+      },
+      "train_right_elbow_mae": {
+        "mean": 0.06480067297816276,
+        "std": 0.0029587520280501783
+      },
+      "train_right_elbow_r2": {
+        "mean": 0.5710707545280457,
+        "std": 0.04083656521133971
+      },
+      "train_left_hand_mse": {
+        "mean": 0.010479144752025604,
+        "std": 0.0009347572663649777
+      },
+      "train_left_hand_rmse": {
+        "mean": 0.10226442825197819,
+        "std": 0.004596897466944967
+      },
+      "train_left_hand_mae": {
+        "mean": 0.08252316415309906,
+        "std": 0.0038670526377426674
+      },
+      "train_left_hand_r2": {
+        "mean": 0.6007721185684204,
+        "std": 0.038130159563052685
+      },
+      "train_right_hand_mse": {
+        "mean": 0.011829768307507038,
+        "std": 0.001125448137534606
+      },
+      "train_right_hand_rmse": {
+        "mean": 0.10863865737058194,
+        "std": 0.005235497323496982
+      },
+      "train_right_hand_mae": {
+        "mean": 0.08576805740594864,
+        "std": 0.0040909984046320555
+      },
+      "train_right_hand_r2": {
+        "mean": 0.5758797407150269,
+        "std": 0.042345979264782486
+      },
+      "train_left_hip_mse": {
+        "mean": 0.004274861328303814,
+        "std": 0.00019655002515241862
+      },
+      "train_left_hip_rmse": {
+        "mean": 0.06536546822602732,
+        "std": 0.0014889230658518512
+      },
+      "train_left_hip_mae": {
+        "mean": 0.049389596283435824,
+        "std": 0.0016684620726999055
+      },
+      "train_left_hip_r2": {
+        "mean": 0.46563867330551145,
+        "std": 0.024648690135835545
+      },
+      "train_right_hip_mse": {
+        "mean": 0.0045171169564127926,
+        "std": 0.0002308094316479834
+      },
+      "train_right_hip_rmse": {
+        "mean": 0.06718807377637957,
+        "std": 0.0016969674812934103
+      },
+      "train_right_hip_mae": {
+        "mean": 0.050718558579683305,
+        "std": 0.001811111604416488
+      },
+      "train_right_hip_r2": {
+        "mean": 0.4573493480682373,
+        "std": 0.02722794184655313
+      },
+      "train_left_knee_mse": {
+        "mean": 0.004310890752822161,
+        "std": 0.0003215059909364965
+      },
+      "train_left_knee_rmse": {
+        "mean": 0.06561382435966372,
+        "std": 0.002390984257865752
+      },
+      "train_left_knee_mae": {
+        "mean": 0.04918176010251045,
+        "std": 0.0017410559928729534
+      },
+      "train_left_knee_r2": {
+        "mean": 0.5479459643363953,
+        "std": 0.032747594695522145
+      },
+      "train_right_knee_mse": {
+        "mean": 0.0044992403127253056,
+        "std": 0.0002501707905427025
+      },
+      "train_right_knee_rmse": {
+        "mean": 0.06705132328457544,
+        "std": 0.001833128067718345
+      },
+      "train_right_knee_mae": {
+        "mean": 0.0502285897731781,
+        "std": 0.0015331378293234393
+      },
+      "train_right_knee_r2": {
+        "mean": 0.5813280582427979,
+        "std": 0.025434494523937038
+      },
+      "train_left_foot_mse": {
+        "mean": 0.004202403221279382,
+        "std": 4.2078493237115376e-05
+      },
+      "train_left_foot_rmse": {
+        "mean": 0.06482513383253344,
+        "std": 0.0003244146629106532
+      },
+      "train_left_foot_mae": {
+        "mean": 0.04702796787023544,
+        "std": 0.0002586516748229261
+      },
+      "train_left_foot_r2": {
+        "mean": 0.05115777254104614,
+        "std": 0.005946433365054517
+      },
+      "train_right_foot_mse": {
+        "mean": 0.004451785329729318,
+        "std": 9.545716950150217e-05
+      },
+      "train_right_foot_rmse": {
+        "mean": 0.06671786480676475,
+        "std": 0.0007154336835637129
+      },
+      "train_right_foot_mae": {
+        "mean": 0.048358583450317384,
+        "std": 0.0005721187512742098
+      },
+      "train_right_foot_r2": {
+        "mean": 0.04036338329315185,
+        "std": 0.0037621134642699585
+      }
+    },
+    "val_metrics": {
+      "val_mse": {
+        "mean": 0.005478056613355875,
+        "std": 0.00027081668798800197
+      },
+      "val_rmse": {
+        "mean": 0.0739909623095735,
+        "std": 0.0018423110104289783
+      },
+      "val_mae": {
+        "mean": 0.05577969029545784,
+        "std": 0.0018564538276403394
+      },
+      "val_r2": {
+        "mean": 0.7408774375915528,
+        "std": 0.016279279828662133
+      },
+      "val_head_mse": {
+        "mean": 0.003687116177752614,
+        "std": 0.00026835237273755645
+      },
+      "val_head_rmse": {
+        "mean": 0.060681836603097936,
+        "std": 0.002197927257107851
+      },
+      "val_head_mae": {
+        "mean": 0.049092726409435274,
+        "std": 0.0025035384421660143
+      },
+      "val_head_r2": {
+        "mean": 0.6306290626525879,
+        "std": 0.02143641862947963
+      },
+      "val_left_shoulder_mse": {
+        "mean": 0.002580420905724168,
+        "std": 0.0002234334729124409
+      },
+      "val_left_shoulder_rmse": {
+        "mean": 0.05074941771214784,
+        "std": 0.0022175454002350717
+      },
+      "val_left_shoulder_mae": {
+        "mean": 0.040724245458841325,
+        "std": 0.0022838185084261874
+      },
+      "val_left_shoulder_r2": {
+        "mean": 0.5377515316009521,
+        "std": 0.0323848960685901
+      },
+      "val_left_elbow_mse": {
+        "mean": 0.005640014167875051,
+        "std": 0.0004334570469077089
+      },
+      "val_left_elbow_rmse": {
+        "mean": 0.07504546994452427,
+        "std": 0.002862098649689754
+      },
+      "val_left_elbow_mae": {
+        "mean": 0.059190322458744046,
+        "std": 0.0025991497330166975
+      },
+      "val_left_elbow_r2": {
+        "mean": 0.5889587759971618,
+        "std": 0.03282415635103076
+      },
+      "val_right_shoulder_mse": {
+        "mean": 0.003288494935259223,
+        "std": 0.0001761332472677511
+      },
+      "val_right_shoulder_rmse": {
+        "mean": 0.05732531135635566,
+        "std": 0.0015177658436347386
+      },
+      "val_right_shoulder_mae": {
+        "mean": 0.046021364629268646,
+        "std": 0.0018533935199183914
+      },
+      "val_right_shoulder_r2": {
+        "mean": 0.4966256141662598,
+        "std": 0.024582821105842343
+      },
+      "val_right_elbow_mse": {
+        "mean": 0.006871881429105997,
+        "std": 0.0004433083511971699
+      },
+      "val_right_elbow_rmse": {
+        "mean": 0.08285338339354811,
+        "std": 0.0026829627928321685
+      },
+      "val_right_elbow_mae": {
+        "mean": 0.06525605469942093,
+        "std": 0.0024083392200376494
+      },
+      "val_right_elbow_r2": {
+        "mean": 0.5620760321617126,
+        "std": 0.03216144096231467
+      },
+      "val_left_hand_mse": {
+        "mean": 0.010725713893771171,
+        "std": 0.000668782707280832
+      },
+      "val_left_hand_rmse": {
+        "mean": 0.10351605199575539,
+        "std": 0.0031844737372502038
+      },
+      "val_left_hand_mae": {
+        "mean": 0.0830638900399208,
+        "std": 0.0031902396437084785
+      },
+      "val_left_hand_r2": {
+        "mean": 0.590312659740448,
+        "std": 0.031375228573209214
+      },
+      "val_right_hand_mse": {
+        "mean": 0.012054064311087132,
+        "std": 0.0009381013811746891
+      },
+      "val_right_hand_rmse": {
+        "mean": 0.10970780925789572,
+        "std": 0.004273277304400014
+      },
+      "val_right_hand_mae": {
+        "mean": 0.08650216311216355,
+        "std": 0.00398613505747233
+      },
+      "val_right_hand_r2": {
+        "mean": 0.5669370174407959,
+        "std": 0.03531806065520998
+      },
+      "val_left_hip_mse": {
+        "mean": 0.004295512475073338,
+        "std": 0.0005391504755457043
+      },
+      "val_left_hip_rmse": {
+        "mean": 0.06541313707645799,
+        "std": 0.004078476785499173
+      },
+      "val_left_hip_mae": {
+        "mean": 0.04946355819702149,
+        "std": 0.0029649690105264977
+      },
+      "val_left_hip_r2": {
+        "mean": 0.4643921494483948,
+        "std": 0.04029116451814752
+      },
+      "val_right_hip_mse": {
+        "mean": 0.004534597042948008,
+        "std": 0.0005553497370777009
+      },
+      "val_right_hip_rmse": {
+        "mean": 0.06721450823299716,
+        "std": 0.004099625098026231
+      },
+      "val_right_hip_mae": {
+        "mean": 0.05077023580670357,
+        "std": 0.003056380287020426
+      },
+      "val_right_hip_r2": {
+        "mean": 0.4565340161323547,
+        "std": 0.03899049590320934
+      },
+      "val_left_knee_mse": {
+        "mean": 0.004340869747102261,
+        "std": 0.0002781407817722438
+      },
+      "val_left_knee_rmse": {
+        "mean": 0.06585112322793311,
+        "std": 0.0021211592872362552
+      },
+      "val_left_knee_mae": {
+        "mean": 0.049283433705568314,
+        "std": 0.002338542472369006
+      },
+      "val_left_knee_r2": {
+        "mean": 0.5444213032722474,
+        "std": 0.025439156403148724
+      },
+      "val_right_knee_mse": {
+        "mean": 0.004521301900967956,
+        "std": 0.00048010112643549177
+      },
+      "val_right_knee_rmse": {
+        "mean": 0.06714759087548117,
+        "std": 0.0035359497432731982
+      },
+      "val_right_knee_mae": {
+        "mean": 0.050302571058273314,
+        "std": 0.0029784697066226836
+      },
+      "val_right_knee_r2": {
+        "mean": 0.5794642090797424,
+        "std": 0.03850312675610406
+      },
+      "val_left_foot_mse": {
+        "mean": 0.004219178296625614,
+        "std": 0.00017746144020572974
+      },
+      "val_left_foot_rmse": {
+        "mean": 0.06494072029495798,
+        "std": 0.0013715481025635898
+      },
+      "val_left_foot_mae": {
+        "mean": 0.04712641090154648,
+        "std": 0.0014222451842950219
+      },
+      "val_left_foot_r2": {
+        "mean": 0.0473235011100769,
+        "std": 0.016249734893750565
+      },
+      "val_right_foot_mse": {
+        "mean": 0.00445556933991611,
+        "std": 0.0004237094360295466
+      },
+      "val_right_foot_rmse": {
+        "mean": 0.06667404046159596,
+        "std": 0.0031845986311582076
+      },
+      "val_right_foot_mae": {
+        "mean": 0.0483390100300312,
+        "std": 0.0018289416956339239
+      },
+      "val_right_foot_r2": {
+        "mean": 0.03934557437896728,
+        "std": 0.006818991337554979
+      }
+    },
+    "best_fold": "3",
+    "best_fold_metrics": {
+      "fold": 3,
+      "train_metrics": {
+        "train_mse": 0.005233469419181347,
+        "train_rmse": 0.07234272195032025,
+        "train_mae": 0.05435093119740486,
+        "train_r2": 0.7501266002655029,
+        "train_head_mse": 0.003414764767512679,
+        "train_head_rmse": 0.058435988632970684,
+        "train_head_mae": 0.047396495938301086,
+        "train_head_r2": 0.6567138433456421,
+        "train_left_shoulder_mse": 0.002474896376952529,
+        "train_left_shoulder_rmse": 0.04974833039361752,
+        "train_left_shoulder_mae": 0.039659593254327774,
+        "train_left_shoulder_r2": 0.557284414768219,
+        "train_left_elbow_mse": 0.005228264257311821,
+        "train_left_elbow_rmse": 0.07230673728852534,
+        "train_left_elbow_mae": 0.05692024156451225,
+        "train_left_elbow_r2": 0.6156260967254639,
+        "train_right_shoulder_mse": 0.0031457135919481516,
+        "train_right_shoulder_rmse": 0.05608666144412726,
+        "train_right_shoulder_mae": 0.04464205354452133,
+        "train_right_shoulder_r2": 0.5159428119659424,
+        "train_right_elbow_mse": 0.006523681338876486,
+        "train_right_elbow_rmse": 0.08076930938714584,
+        "train_right_elbow_mae": 0.06328018754720688,
+        "train_right_elbow_r2": 0.5805017948150635,
+        "train_left_hand_mse": 0.009903079830110073,
+        "train_left_hand_rmse": 0.09951421923579601,
+        "train_left_hand_mae": 0.08003747463226318,
+        "train_left_hand_r2": 0.6169312596321106,
+        "train_right_hand_mse": 0.011458463966846466,
+        "train_right_hand_rmse": 0.10704421500878254,
+        "train_right_hand_mae": 0.08436527103185654,
+        "train_right_hand_r2": 0.5846413373947144,
+        "train_left_hip_mse": 0.004281698260456324,
+        "train_left_hip_rmse": 0.06543468698218341,
+        "train_left_hip_mae": 0.04869461804628372,
+        "train_left_hip_r2": 0.468951940536499,
+        "train_right_hip_mse": 0.004493603482842445,
+        "train_right_hip_rmse": 0.06703434554646182,
+        "train_right_hip_mae": 0.049883730709552765,
+        "train_right_hip_r2": 0.46397966146469116,
+        "train_left_knee_mse": 0.004129727836698294,
+        "train_left_knee_rmse": 0.06426295851186976,
+        "train_left_knee_mae": 0.047328293323516846,
+        "train_left_knee_r2": 0.5719143152236938,
+        "train_right_knee_mse": 0.004376658704131842,
+        "train_right_knee_rmse": 0.06615632021305176,
+        "train_right_knee_mae": 0.04874139651656151,
+        "train_right_knee_r2": 0.5948373079299927,
+        "train_left_foot_mse": 0.004147091880440712,
+        "train_left_foot_rmse": 0.06439791829275782,
+        "train_left_foot_mae": 0.047080304473638535,
+        "train_left_foot_r2": 0.06146746873855591,
+        "train_right_foot_mse": 0.00445745512843132,
+        "train_right_foot_rmse": 0.06676417548679321,
+        "train_right_foot_mae": 0.04853244870901108,
+        "train_right_foot_r2": 0.03906047344207764
+      },
+      "val_metrics": {
+        "val_mse": 0.005024301819503307,
+        "val_rmse": 0.07088230963719584,
+        "val_mae": 0.052906859666109085,
+        "val_r2": 0.7725335359573364,
+        "val_head_mse": 0.0033587520010769367,
+        "val_head_rmse": 0.057954740971528265,
+        "val_head_mae": 0.045722596347332,
+        "val_head_r2": 0.6686442494392395,
+        "val_left_shoulder_mse": 0.0022487305104732513,
+        "val_left_shoulder_rmse": 0.04742078141989281,
+        "val_left_shoulder_mae": 0.03763116896152496,
+        "val_left_shoulder_r2": 0.5945963859558105,
+        "val_left_elbow_mse": 0.005054857116192579,
+        "val_left_elbow_rmse": 0.07109751835466958,
+        "val_left_elbow_mae": 0.05595087260007858,
+        "val_left_elbow_r2": 0.6473710536956787,
+        "val_right_shoulder_mse": 0.0031127743422985077,
+        "val_right_shoulder_rmse": 0.05579224267134731,
+        "val_right_shoulder_mae": 0.04440058767795563,
+        "val_right_shoulder_r2": 0.5350003838539124,
+        "val_right_elbow_mse": 0.006202884018421173,
+        "val_right_elbow_rmse": 0.07875839014620076,
+        "val_right_elbow_mae": 0.06287550181150436,
+        "val_right_elbow_r2": 0.6221969127655029,
+        "val_left_hand_mse": 0.009984288364648819,
+        "val_left_hand_rmse": 0.09992141094204395,
+        "val_left_hand_mae": 0.07982359826564789,
+        "val_left_hand_r2": 0.6415865421295166,
+        "val_right_hand_mse": 0.010731799528002739,
+        "val_right_hand_rmse": 0.1035943991150233,
+        "val_right_hand_mae": 0.08158831298351288,
+        "val_right_hand_r2": 0.6305816769599915,
+        "val_left_hip_mse": 0.0038004289381206036,
+        "val_left_hip_rmse": 0.06164761907909018,
+        "val_left_hip_mae": 0.047193966805934906,
+        "val_left_hip_r2": 0.5091779828071594,
+        "val_right_hip_mse": 0.004099788144230843,
+        "val_right_hip_rmse": 0.06402958803733508,
+        "val_right_hip_mae": 0.04835526645183563,
+        "val_right_hip_r2": 0.4929708242416382,
+        "val_left_knee_mse": 0.003990014083683491,
+        "val_left_knee_rmse": 0.06316655827004897,
+        "val_left_knee_mae": 0.04503210261464119,
+        "val_left_knee_r2": 0.5601710081100464,
+        "val_right_knee_mse": 0.003881337819620967,
+        "val_right_knee_rmse": 0.062300383783897885,
+        "val_right_knee_mae": 0.044968850910663605,
+        "val_right_knee_r2": 0.6313607692718506,
+        "val_left_foot_mse": 0.0043845162726938725,
+        "val_left_foot_rmse": 0.06621567996097202,
+        "val_left_foot_mae": 0.04629934951663017,
+        "val_left_foot_r2": 0.018805503845214844,
+        "val_right_foot_mse": 0.004465752746909857,
+        "val_right_foot_rmse": 0.06682628784325714,
+        "val_right_foot_mae": 0.0479469932615757,
+        "val_right_foot_r2": 0.037346065044403076
+      },
+      "best_epoch": 60
+    }
+  },
+  "fold_details": [
+    {
+      "fold": 1,
+      "train_metrics": {
+        "train_mse": 0.005406604148447514,
+        "train_rmse": 0.07352961409151766,
+        "train_mae": 0.0554690957069397,
+        "train_r2": 0.744255006313324,
+        "train_head_mse": 0.003753931727260351,
+        "train_head_rmse": 0.06126933757810958,
+        "train_head_mae": 0.04946490004658699,
+        "train_head_r2": 0.622705340385437,
+        "train_left_shoulder_mse": 0.0025851703248918056,
+        "train_left_shoulder_rmse": 0.05084457025968265,
+        "train_left_shoulder_mae": 0.040929459035396576,
+        "train_left_shoulder_r2": 0.535000741481781,
+        "train_left_elbow_mse": 0.005631607491523027,
+        "train_left_elbow_rmse": 0.07504403701509553,
+        "train_left_elbow_mae": 0.05929046496748924,
+        "train_left_elbow_r2": 0.5921444892883301,
+        "train_right_shoulder_mse": 0.0033023585565388203,
+        "train_right_shoulder_rmse": 0.05746615139835641,
+        "train_right_shoulder_mae": 0.0463193841278553,
+        "train_right_shoulder_r2": 0.49383479356765747,
+        "train_right_elbow_mse": 0.006839757785201073,
+        "train_right_elbow_rmse": 0.08270282815721039,
+        "train_right_elbow_mae": 0.06538952142000198,
+        "train_right_elbow_r2": 0.5664427280426025,
+        "train_left_hand_mse": 0.010661396197974682,
+        "train_left_hand_rmse": 0.10325403719939807,
+        "train_left_hand_mae": 0.08297795057296753,
+        "train_left_hand_r2": 0.5944560766220093,
+        "train_right_hand_mse": 0.011868588626384735,
+        "train_right_hand_rmse": 0.10894305221713194,
+        "train_right_hand_mae": 0.08615357428789139,
+        "train_right_hand_r2": 0.5738925337791443,
+        "train_left_hip_mse": 0.004100636579096317,
+        "train_left_hip_rmse": 0.06403621302900663,
+        "train_left_hip_mae": 0.048643115907907486,
+        "train_left_hip_r2": 0.47410911321640015,
+        "train_right_hip_mse": 0.004325824789702892,
+        "train_right_hip_rmse": 0.0657710026508863,
+        "train_right_hip_mae": 0.049941882491111755,
+        "train_right_hip_r2": 0.466940701007843,
+        "train_left_knee_mse": 0.004242466297000647,
+        "train_left_knee_rmse": 0.0651342175588273,
+        "train_left_knee_mae": 0.04853753000497818,
+        "train_left_knee_r2": 0.552821159362793,
+        "train_right_knee_mse": 0.004472049884498119,
+        "train_right_knee_rmse": 0.06687338696744856,
+        "train_right_knee_mae": 0.049348197877407074,
+        "train_right_knee_r2": 0.5842645764350891,
+        "train_left_foot_mse": 0.004170173313468695,
+        "train_left_foot_rmse": 0.06457687909359429,
+        "train_left_foot_mae": 0.04659763351082802,
+        "train_left_foot_r2": 0.0465083122253418,
+        "train_right_foot_mse": 0.004331889562308788,
+        "train_right_foot_rmse": 0.06581709171870775,
+        "train_right_foot_mae": 0.04750465229153633,
+        "train_right_foot_r2": 0.03402256965637207
+      },
+      "val_metrics": {
+        "val_mse": 0.005823077168315649,
+        "val_rmse": 0.07630908968344235,
+        "val_mae": 0.0576336644589901,
+        "val_r2": 0.7269455194473267,
+        "val_head_mse": 0.003901309333741665,
+        "val_head_rmse": 0.06246046216400952,
+        "val_head_mae": 0.05146763101220131,
+        "val_head_r2": 0.6170878410339355,
+        "val_left_shoulder_mse": 0.002829795703291893,
+        "val_left_shoulder_rmse": 0.05319582411516804,
+        "val_left_shoulder_mae": 0.04243249446153641,
+        "val_left_shoulder_r2": 0.5035913586616516,
+        "val_left_elbow_mse": 0.005708644632250071,
+        "val_left_elbow_rmse": 0.07555557313825415,
+        "val_left_elbow_mae": 0.05968219041824341,
+        "val_left_elbow_r2": 0.5786737203598022,
+        "val_right_shoulder_mse": 0.0036107669584453106,
+        "val_right_shoulder_rmse": 0.06008965766623497,
+        "val_right_shoulder_mae": 0.048031285405159,
+        "val_right_shoulder_r2": 0.45763373374938965,
+        "val_right_elbow_mse": 0.007306069601327181,
+        "val_right_elbow_rmse": 0.08547554972813677,
+        "val_right_elbow_mae": 0.06642314791679382,
+        "val_right_elbow_r2": 0.532703161239624,
+        "val_left_hand_mse": 0.010753454640507698,
+        "val_left_hand_rmse": 0.10369886518428106,
+        "val_left_hand_mae": 0.08406193554401398,
+        "val_left_hand_r2": 0.589398205280304,
+        "val_right_hand_mse": 0.01274879276752472,
+        "val_right_hand_rmse": 0.11291055206456445,
+        "val_right_hand_mae": 0.08850652724504471,
+        "val_right_hand_r2": 0.5469165444374084,
+        "val_left_hip_mse": 0.005038032308220863,
+        "val_left_hip_rmse": 0.07097909768531059,
+        "val_left_hip_mae": 0.05225786566734314,
+        "val_left_hip_r2": 0.42839163541793823,
+        "val_right_hip_mse": 0.005314328242093325,
+        "val_right_hip_rmse": 0.07289943924402523,
+        "val_right_hip_mae": 0.05356563255190849,
+        "val_right_hip_r2": 0.4200592637062073,
+        "val_left_knee_mse": 0.004457443952560425,
+        "val_left_knee_rmse": 0.0667640917901264,
+        "val_left_knee_mae": 0.050115931779146194,
+        "val_left_knee_r2": 0.5418477654457092,
+        "val_right_knee_mse": 0.004610919859260321,
+        "val_right_knee_rmse": 0.06790375438265782,
+        "val_right_knee_mae": 0.051631275564432144,
+        "val_right_knee_r2": 0.5700148344039917,
+        "val_left_foot_mse": 0.004406743682920933,
+        "val_left_foot_rmse": 0.06638330876749766,
+        "val_left_foot_mae": 0.04936696216464043,
+        "val_left_foot_r2": 0.05243360996246338,
+        "val_right_foot_mse": 0.005013700108975172,
+        "val_right_foot_rmse": 0.07080748624951441,
+        "val_right_foot_mae": 0.0516948364675045,
+        "val_right_foot_r2": 0.044953107833862305
+      },
+      "best_epoch": 76
+    },
+    {
+      "fold": 2,
+      "train_metrics": {
+        "train_mse": 0.005729618016630411,
+        "train_rmse": 0.07569424031345061,
+        "train_mae": 0.05749480798840523,
+        "train_r2": 0.7297403812408447,
+        "train_head_mse": 0.0038221655413508415,
+        "train_head_rmse": 0.06182366489743908,
+        "train_head_mae": 0.050370462238788605,
+        "train_head_r2": 0.621444582939148,
+        "train_left_shoulder_mse": 0.0026433279272168875,
+        "train_left_shoulder_rmse": 0.0514133049629849,
+        "train_left_shoulder_mae": 0.041557952761650085,
+        "train_left_shoulder_r2": 0.5268577337265015,
+        "train_left_elbow_mse": 0.00583728076890111,
+        "train_left_elbow_rmse": 0.07640209924407254,
+        "train_left_elbow_mae": 0.060641150921583176,
+        "train_left_elbow_r2": 0.5750073790550232,
+        "train_right_shoulder_mse": 0.0033982384484261274,
+        "train_right_shoulder_rmse": 0.05829441181130596,
+        "train_right_shoulder_mae": 0.04702089726924896,
+        "train_right_shoulder_r2": 0.48429685831069946,
+        "train_right_elbow_mse": 0.007122038397938013,
+        "train_right_elbow_rmse": 0.08439217024071613,
+        "train_right_elbow_mae": 0.06672323495149612,
+        "train_right_elbow_r2": 0.5498831272125244,
+        "train_left_hand_mse": 0.01102411188185215,
+        "train_left_hand_rmse": 0.10499577078078978,
+        "train_left_hand_mae": 0.08490493148565292,
+        "train_left_hand_r2": 0.580420732498169,
+        "train_right_hand_mse": 0.01251142006367445,
+        "train_right_hand_rmse": 0.1118544592927544,
+        "train_right_hand_mae": 0.08837948739528656,
+        "train_right_hand_r2": 0.5543529391288757,
+        "train_left_hip_mse": 0.004631414543837309,
+        "train_left_hip_rmse": 0.0680544968671234,
+        "train_left_hip_mae": 0.052654244005680084,
+        "train_left_hip_r2": 0.41861826181411743,
+        "train_right_hip_mse": 0.004943889565765858,
+        "train_right_hip_rmse": 0.07031279802259228,
+        "train_right_hip_mae": 0.05425163730978966,
+        "train_right_hip_r2": 0.40450650453567505,
+        "train_left_knee_mse": 0.004942110273987055,
+        "train_left_knee_rmse": 0.0703001441960616,
+        "train_left_knee_mae": 0.05243578925728798,
+        "train_left_knee_r2": 0.483923077583313,
+        "train_right_knee_mse": 0.004985864274203777,
+        "train_right_knee_rmse": 0.07061065269634446,
+        "train_right_knee_mae": 0.05314826965332031,
+        "train_right_knee_r2": 0.5320762395858765,
+        "train_left_foot_mse": 0.004264891613274813,
+        "train_left_foot_rmse": 0.06530613763862331,
+        "train_left_foot_mae": 0.047061510384082794,
+        "train_left_foot_r2": 0.046724677085876465,
+        "train_right_foot_mse": 0.004358281381428242,
+        "train_right_foot_rmse": 0.06601728093028553,
+        "train_right_foot_mae": 0.048282962292432785,
+        "train_right_foot_r2": 0.04465419054031372
+      },
+      "val_metrics": {
+        "val_mse": 0.005558268167078495,
+        "val_rmse": 0.07455379378058835,
+        "val_mae": 0.05685051530599594,
+        "val_r2": 0.7364064455032349,
+        "val_head_mse": 0.0035236934199929237,
+        "val_head_rmse": 0.059360706026738964,
+        "val_head_mae": 0.048691071569919586,
+        "val_head_r2": 0.6329751014709473,
+        "val_left_shoulder_mse": 0.002631589537486434,
+        "val_left_shoulder_rmse": 0.05129902082385622,
+        "val_left_shoulder_mae": 0.04142003506422043,
+        "val_left_shoulder_r2": 0.529413104057312,
+        "val_left_elbow_mse": 0.005567439831793308,
+        "val_left_elbow_rmse": 0.07461527880932503,
+        "val_left_elbow_mae": 0.059319138526916504,
+        "val_left_elbow_r2": 0.5977160334587097,
+        "val_right_shoulder_mse": 0.0032420784700661898,
+        "val_right_shoulder_rmse": 0.056939252454402574,
+        "val_right_shoulder_mae": 0.04615074396133423,
+        "val_right_shoulder_r2": 0.4931461811065674,
+        "val_right_elbow_mse": 0.006679263431578875,
+        "val_right_elbow_rmse": 0.08172676080439549,
+        "val_right_elbow_mae": 0.06460388749837875,
+        "val_right_elbow_r2": 0.5676771402359009,
+        "val_left_hand_mse": 0.010377293452620506,
+        "val_left_hand_rmse": 0.10186900143135058,
+        "val_left_hand_mae": 0.0826764777302742,
+        "val_left_hand_r2": 0.6045659184455872,
+        "val_right_hand_mse": 0.011468841694295406,
+        "val_right_hand_rmse": 0.10709267806108598,
+        "val_right_hand_mae": 0.08507442474365234,
+        "val_right_hand_r2": 0.579136848449707,
+        "val_left_hip_mse": 0.004841412417590618,
+        "val_left_hip_rmse": 0.06958025882095163,
+        "val_left_hip_mae": 0.05368497222661972,
+        "val_left_hip_r2": 0.40519070625305176,
+        "val_right_hip_mse": 0.005053681321442127,
+        "val_right_hip_rmse": 0.071089248986342,
+        "val_right_hip_mae": 0.055145956575870514,
+        "val_right_hip_r2": 0.3994207978248596,
+        "val_left_knee_mse": 0.00465551670640707,
+        "val_left_knee_rmse": 0.0682313469485036,
+        "val_left_knee_mae": 0.051686931401491165,
+        "val_left_knee_r2": 0.5031958818435669,
+        "val_right_knee_mse": 0.005340579431504011,
+        "val_right_knee_rmse": 0.07307926813744108,
+        "val_right_knee_mae": 0.05400300770998001,
+        "val_right_knee_r2": 0.5198122262954712,
+        "val_left_foot_mse": 0.004064866341650486,
+        "val_left_foot_rmse": 0.06375630432867394,
+        "val_left_foot_mae": 0.048231568187475204,
+        "val_left_foot_r2": 0.043310463428497314,
+        "val_right_foot_mse": 0.004811230581253767,
+        "val_right_foot_rmse": 0.0693630346889016,
+        "val_right_foot_mae": 0.04836846888065338,
+        "val_right_foot_r2": 0.027853727340698242
+      },
+      "best_epoch": 62
+    },
+    {
+      "fold": 3,
+      "train_metrics": {
+        "train_mse": 0.005233469419181347,
+        "train_rmse": 0.07234272195032025,
+        "train_mae": 0.05435093119740486,
+        "train_r2": 0.7501266002655029,
+        "train_head_mse": 0.003414764767512679,
+        "train_head_rmse": 0.058435988632970684,
+        "train_head_mae": 0.047396495938301086,
+        "train_head_r2": 0.6567138433456421,
+        "train_left_shoulder_mse": 0.002474896376952529,
+        "train_left_shoulder_rmse": 0.04974833039361752,
+        "train_left_shoulder_mae": 0.039659593254327774,
+        "train_left_shoulder_r2": 0.557284414768219,
+        "train_left_elbow_mse": 0.005228264257311821,
+        "train_left_elbow_rmse": 0.07230673728852534,
+        "train_left_elbow_mae": 0.05692024156451225,
+        "train_left_elbow_r2": 0.6156260967254639,
+        "train_right_shoulder_mse": 0.0031457135919481516,
+        "train_right_shoulder_rmse": 0.05608666144412726,
+        "train_right_shoulder_mae": 0.04464205354452133,
+        "train_right_shoulder_r2": 0.5159428119659424,
+        "train_right_elbow_mse": 0.006523681338876486,
+        "train_right_elbow_rmse": 0.08076930938714584,
+        "train_right_elbow_mae": 0.06328018754720688,
+        "train_right_elbow_r2": 0.5805017948150635,
+        "train_left_hand_mse": 0.009903079830110073,
+        "train_left_hand_rmse": 0.09951421923579601,
+        "train_left_hand_mae": 0.08003747463226318,
+        "train_left_hand_r2": 0.6169312596321106,
+        "train_right_hand_mse": 0.011458463966846466,
+        "train_right_hand_rmse": 0.10704421500878254,
+        "train_right_hand_mae": 0.08436527103185654,
+        "train_right_hand_r2": 0.5846413373947144,
+        "train_left_hip_mse": 0.004281698260456324,
+        "train_left_hip_rmse": 0.06543468698218341,
+        "train_left_hip_mae": 0.04869461804628372,
+        "train_left_hip_r2": 0.468951940536499,
+        "train_right_hip_mse": 0.004493603482842445,
+        "train_right_hip_rmse": 0.06703434554646182,
+        "train_right_hip_mae": 0.049883730709552765,
+        "train_right_hip_r2": 0.46397966146469116,
+        "train_left_knee_mse": 0.004129727836698294,
+        "train_left_knee_rmse": 0.06426295851186976,
+        "train_left_knee_mae": 0.047328293323516846,
+        "train_left_knee_r2": 0.5719143152236938,
+        "train_right_knee_mse": 0.004376658704131842,
+        "train_right_knee_rmse": 0.06615632021305176,
+        "train_right_knee_mae": 0.04874139651656151,
+        "train_right_knee_r2": 0.5948373079299927,
+        "train_left_foot_mse": 0.004147091880440712,
+        "train_left_foot_rmse": 0.06439791829275782,
+        "train_left_foot_mae": 0.047080304473638535,
+        "train_left_foot_r2": 0.06146746873855591,
+        "train_right_foot_mse": 0.00445745512843132,
+        "train_right_foot_rmse": 0.06676417548679321,
+        "train_right_foot_mae": 0.04853244870901108,
+        "train_right_foot_r2": 0.03906047344207764
+      },
+      "val_metrics": {
+        "val_mse": 0.005024301819503307,
+        "val_rmse": 0.07088230963719584,
+        "val_mae": 0.052906859666109085,
+        "val_r2": 0.7725335359573364,
+        "val_head_mse": 0.0033587520010769367,
+        "val_head_rmse": 0.057954740971528265,
+        "val_head_mae": 0.045722596347332,
+        "val_head_r2": 0.6686442494392395,
+        "val_left_shoulder_mse": 0.0022487305104732513,
+        "val_left_shoulder_rmse": 0.04742078141989281,
+        "val_left_shoulder_mae": 0.03763116896152496,
+        "val_left_shoulder_r2": 0.5945963859558105,
+        "val_left_elbow_mse": 0.005054857116192579,
+        "val_left_elbow_rmse": 0.07109751835466958,
+        "val_left_elbow_mae": 0.05595087260007858,
+        "val_left_elbow_r2": 0.6473710536956787,
+        "val_right_shoulder_mse": 0.0031127743422985077,
+        "val_right_shoulder_rmse": 0.05579224267134731,
+        "val_right_shoulder_mae": 0.04440058767795563,
+        "val_right_shoulder_r2": 0.5350003838539124,
+        "val_right_elbow_mse": 0.006202884018421173,
+        "val_right_elbow_rmse": 0.07875839014620076,
+        "val_right_elbow_mae": 0.06287550181150436,
+        "val_right_elbow_r2": 0.6221969127655029,
+        "val_left_hand_mse": 0.009984288364648819,
+        "val_left_hand_rmse": 0.09992141094204395,
+        "val_left_hand_mae": 0.07982359826564789,
+        "val_left_hand_r2": 0.6415865421295166,
+        "val_right_hand_mse": 0.010731799528002739,
+        "val_right_hand_rmse": 0.1035943991150233,
+        "val_right_hand_mae": 0.08158831298351288,
+        "val_right_hand_r2": 0.6305816769599915,
+        "val_left_hip_mse": 0.0038004289381206036,
+        "val_left_hip_rmse": 0.06164761907909018,
+        "val_left_hip_mae": 0.047193966805934906,
+        "val_left_hip_r2": 0.5091779828071594,
+        "val_right_hip_mse": 0.004099788144230843,
+        "val_right_hip_rmse": 0.06402958803733508,
+        "val_right_hip_mae": 0.04835526645183563,
+        "val_right_hip_r2": 0.4929708242416382,
+        "val_left_knee_mse": 0.003990014083683491,
+        "val_left_knee_rmse": 0.06316655827004897,
+        "val_left_knee_mae": 0.04503210261464119,
+        "val_left_knee_r2": 0.5601710081100464,
+        "val_right_knee_mse": 0.003881337819620967,
+        "val_right_knee_rmse": 0.062300383783897885,
+        "val_right_knee_mae": 0.044968850910663605,
+        "val_right_knee_r2": 0.6313607692718506,
+        "val_left_foot_mse": 0.0043845162726938725,
+        "val_left_foot_rmse": 0.06621567996097202,
+        "val_left_foot_mae": 0.04629934951663017,
+        "val_left_foot_r2": 0.018805503845214844,
+        "val_right_foot_mse": 0.004465752746909857,
+        "val_right_foot_rmse": 0.06682628784325714,
+        "val_right_foot_mae": 0.0479469932615757,
+        "val_right_foot_r2": 0.037346065044403076
+      },
+      "best_epoch": 60
+    },
+    {
+      "fold": 4,
+      "train_metrics": {
+        "train_mse": 0.005815441254526377,
+        "train_rmse": 0.07625904047735178,
+        "train_mae": 0.05746934935450554,
+        "train_r2": 0.7245899438858032,
+        "train_head_mse": 0.004142640624195337,
+        "train_head_rmse": 0.06436334845387814,
+        "train_head_mae": 0.05167139321565628,
+        "train_head_r2": 0.5818799734115601,
+        "train_left_shoulder_mse": 0.002833297709003091,
+        "train_left_shoulder_rmse": 0.05322873010887157,
+        "train_left_shoulder_mae": 0.04285058751702309,
+        "train_left_shoulder_r2": 0.49003076553344727,
+        "train_left_elbow_mse": 0.0062269363552331924,
+        "train_left_elbow_rmse": 0.07891093938886543,
+        "train_left_elbow_mae": 0.06278936564922333,
+        "train_left_elbow_r2": 0.5433367490768433,
+        "train_right_shoulder_mse": 0.0035929775331169367,
+        "train_right_shoulder_rmse": 0.059941450875975105,
+        "train_right_shoulder_mae": 0.04820698872208595,
+        "train_right_shoulder_r2": 0.4506111145019531,
+        "train_right_elbow_mse": 0.007542469073086977,
+        "train_right_elbow_rmse": 0.08684738955827617,
+        "train_right_elbow_mae": 0.06860202550888062,
+        "train_right_elbow_r2": 0.5174382328987122,
+        "train_left_hand_mse": 0.011762864887714386,
+        "train_left_hand_rmse": 0.1084567420113401,
+        "train_left_hand_mae": 0.0879446342587471,
+        "train_left_hand_r2": 0.5490212440490723,
+        "train_right_hand_mse": 0.013338660821318626,
+        "train_right_hand_rmse": 0.11549312023371187,
+        "train_right_hand_mae": 0.09103211015462875,
+        "train_right_hand_r2": 0.5187851190567017,
+        "train_left_hip_mse": 0.004274450242519379,
+        "train_left_hip_rmse": 0.0653792799174125,
+        "train_left_hip_mae": 0.049006521701812744,
+        "train_left_hip_r2": 0.4754042625427246,
+        "train_right_hip_mse": 0.004522082861512899,
+        "train_right_hip_rmse": 0.0672464338200391,
+        "train_right_hip_mae": 0.05037889629602432,
+        "train_right_hip_r2": 0.46831393241882324,
+        "train_left_knee_mse": 0.004183592274785042,
+        "train_left_knee_rmse": 0.06468069476114988,
+        "train_left_knee_mae": 0.0492679663002491,
+        "train_left_knee_r2": 0.5607370734214783,
+        "train_right_knee_mse": 0.004373382311314344,
+        "train_right_knee_rmse": 0.06613155306897264,
+        "train_right_knee_mae": 0.05016324296593666,
+        "train_right_knee_r2": 0.591422975063324,
+        "train_left_foot_mse": 0.00423130439594388,
+        "train_left_foot_rmse": 0.06504847727613522,
+        "train_left_foot_mae": 0.04699267819523811,
+        "train_left_foot_r2": 0.05431830883026123,
+        "train_right_foot_mse": 0.004576075356453657,
+        "train_right_foot_rmse": 0.06764669508892254,
+        "train_right_foot_mae": 0.04819516837596893,
+        "train_right_foot_r2": 0.043594181537628174
+      },
+      "val_metrics": {
+        "val_mse": 0.005624482873827219,
+        "val_rmse": 0.07499655241294241,
+        "val_mae": 0.05723588541150093,
+        "val_r2": 0.7374954223632812,
+        "val_head_mse": 0.004091551061719656,
+        "val_head_rmse": 0.0639652332264931,
+        "val_head_mae": 0.05236056447029114,
+        "val_head_r2": 0.604891836643219,
+        "val_left_shoulder_mse": 0.002788741607218981,
+        "val_left_shoulder_rmse": 0.052808537256952885,
+        "val_left_shoulder_mae": 0.043605729937553406,
+        "val_left_shoulder_r2": 0.5122315287590027,
+        "val_left_elbow_mse": 0.00638927286490798,
+        "val_left_elbow_rmse": 0.07993292728849595,
+        "val_left_elbow_mae": 0.06362991034984589,
+        "val_left_elbow_r2": 0.5511693954467773,
+        "val_right_shoulder_mse": 0.003319697454571724,
+        "val_right_shoulder_rmse": 0.057616815727456895,
+        "val_right_shoulder_mae": 0.04803778603672981,
+        "val_right_shoulder_r2": 0.49666285514831543,
+        "val_right_elbow_mse": 0.0074158115312457085,
+        "val_right_elbow_rmse": 0.08611510628946416,
+        "val_right_elbow_mae": 0.06933575123548508,
+        "val_right_elbow_r2": 0.5427825450897217,
+        "val_left_hand_mse": 0.01196402870118618,
+        "val_left_hand_rmse": 0.10938020251026316,
+        "val_left_hand_mae": 0.08860866725444794,
+        "val_left_hand_r2": 0.5571056008338928,
+        "val_right_hand_mse": 0.013399112038314342,
+        "val_right_hand_rmse": 0.11575453355404419,
+        "val_right_hand_mae": 0.09313388913869858,
+        "val_right_hand_r2": 0.53276526927948,
+        "val_left_hip_mse": 0.003750057891011238,
+        "val_left_hip_rmse": 0.06123771624588264,
+        "val_left_hip_mae": 0.0461125373840332,
+        "val_left_hip_r2": 0.4940374493598938,
+        "val_right_hip_mse": 0.003876402508467436,
+        "val_right_hip_rmse": 0.06226076218990124,
+        "val_right_hip_mae": 0.047251202166080475,
+        "val_right_hip_r2": 0.49001866579055786,
+        "val_left_knee_mse": 0.004028758499771357,
+        "val_left_knee_rmse": 0.06347250191832174,
+        "val_left_knee_mae": 0.0487222746014595,
+        "val_left_knee_r2": 0.5795191526412964,
+        "val_right_knee_mse": 0.004269528202712536,
+        "val_right_knee_rmse": 0.06534162687531231,
+        "val_right_knee_mae": 0.05000610277056694,
+        "val_right_knee_r2": 0.6095555424690247,
+        "val_left_foot_mse": 0.003959548659622669,
+        "val_left_foot_rmse": 0.06292494465331432,
+        "val_left_foot_mae": 0.045775867998600006,
+        "val_left_foot_r2": 0.06769812107086182,
+        "val_right_foot_mse": 0.003865764243528247,
+        "val_right_foot_rmse": 0.06217527035347934,
+        "val_right_foot_mae": 0.04748626798391342,
+        "val_right_foot_r2": 0.0474051833152771
+      },
+      "best_epoch": 42
+    },
+    {
+      "fold": 5,
+      "train_metrics": {
+        "train_mse": 0.004853335674852133,
+        "train_rmse": 0.06966588601928589,
+        "train_mae": 0.05288082733750343,
+        "train_r2": 0.7741076946258545,
+        "train_head_mse": 0.0030648622196167707,
+        "train_head_rmse": 0.05536119777982383,
+        "train_head_mae": 0.04487558454275131,
+        "train_head_r2": 0.6959553956985474,
+        "train_left_shoulder_mse": 0.002239338355138898,
+        "train_left_shoulder_rmse": 0.047321647848938,
+        "train_left_shoulder_mae": 0.03781048208475113,
+        "train_left_shoulder_r2": 0.6032886505126953,
+        "train_left_elbow_mse": 0.004719165153801441,
+        "train_left_elbow_rmse": 0.06869618005247047,
+        "train_left_elbow_mae": 0.054372936487197876,
+        "train_left_elbow_r2": 0.662809431552887,
+        "train_right_shoulder_mse": 0.002864172914996743,
+        "train_right_shoulder_rmse": 0.0535179681508626,
+        "train_right_shoulder_mae": 0.04287004843354225,
+        "train_right_shoulder_r2": 0.5658535957336426,
+        "train_right_elbow_mse": 0.005724972113966942,
+        "train_right_elbow_rmse": 0.07566354547579,
+        "train_right_elbow_mae": 0.060008395463228226,
+        "train_right_elbow_r2": 0.6410878896713257,
+        "train_left_hand_mse": 0.00904427096247673,
+        "train_left_hand_rmse": 0.09510137203256708,
+        "train_left_hand_mae": 0.07675082981586456,
+        "train_left_hand_r2": 0.663031280040741,
+        "train_right_hand_mse": 0.009971708059310913,
+        "train_right_hand_rmse": 0.09985844010052887,
+        "train_right_hand_mae": 0.07890984416007996,
+        "train_right_hand_r2": 0.6477267742156982,
+        "train_left_hip_mse": 0.004086107015609741,
+        "train_left_hip_rmse": 0.0639226643344107,
+        "train_left_hip_mae": 0.04794948175549507,
+        "train_left_hip_r2": 0.49110978841781616,
+        "train_right_hip_mse": 0.004300184082239866,
+        "train_right_hip_rmse": 0.06557578884191838,
+        "train_right_hip_mae": 0.04913664609193802,
+        "train_right_hip_r2": 0.48300594091415405,
+        "train_left_knee_mse": 0.004056557081639767,
+        "train_left_knee_rmse": 0.06369110677041,
+        "train_left_knee_mae": 0.04833922162652016,
+        "train_left_knee_r2": 0.5703341960906982,
+        "train_right_knee_mse": 0.004288246389478445,
+        "train_right_knee_rmse": 0.06548470347705977,
+        "train_right_knee_mae": 0.04974184185266495,
+        "train_right_knee_r2": 0.604039192199707,
+        "train_left_foot_mse": 0.004198554903268814,
+        "train_left_foot_rmse": 0.06479625686155654,
+        "train_left_foot_mae": 0.047407712787389755,
+        "train_left_foot_r2": 0.04677009582519531,
+        "train_right_foot_mse": 0.004535225220024586,
+        "train_right_foot_rmse": 0.06734408080911482,
+        "train_right_foot_mae": 0.04927768558263779,
+        "train_right_foot_r2": 0.040485501289367676
+      },
+      "val_metrics": {
+        "val_mse": 0.005360153038054705,
+        "val_rmse": 0.07321306603369855,
+        "val_mae": 0.054271526634693146,
+        "val_r2": 0.7310062646865845,
+        "val_head_mse": 0.0035602750722318888,
+        "val_head_rmse": 0.059668040626719834,
+        "val_head_mae": 0.04722176864743233,
+        "val_head_r2": 0.6295462846755981,
+        "val_left_shoulder_mse": 0.00240324717015028,
+        "val_left_shoulder_rmse": 0.04902292494486921,
+        "val_left_shoulder_mae": 0.038531798869371414,
+        "val_left_shoulder_r2": 0.5489252805709839,
+        "val_left_elbow_mse": 0.005479856394231319,
+        "val_left_elbow_rmse": 0.07402605213187664,
+        "val_left_elbow_mae": 0.057369500398635864,
+        "val_left_elbow_r2": 0.5698636770248413,
+        "val_right_shoulder_mse": 0.003157157450914383,
+        "val_right_shoulder_rmse": 0.05618858826233653,
+        "val_right_shoulder_mae": 0.043486420065164566,
+        "val_right_shoulder_r2": 0.500684916973114,
+        "val_right_elbow_mse": 0.006755378562957048,
+        "val_right_elbow_rmse": 0.08219110999954343,
+        "val_right_elbow_mae": 0.06304198503494263,
+        "val_right_elbow_r2": 0.5450204014778137,
+        "val_left_hand_mse": 0.010549504309892654,
+        "val_left_hand_rmse": 0.10271077991083825,
+        "val_left_hand_mae": 0.08014877140522003,
+        "val_left_hand_r2": 0.5589070320129395,
+        "val_right_hand_mse": 0.01192177552729845,
+        "val_right_hand_rmse": 0.10918688349476072,
+        "val_right_hand_mae": 0.08420766144990921,
+        "val_right_hand_r2": 0.5452847480773926,
+        "val_left_hip_mse": 0.004047630820423365,
+        "val_left_hip_rmse": 0.06362099355105487,
+        "val_left_hip_mae": 0.04806844890117645,
+        "val_left_hip_r2": 0.48516297340393066,
+        "val_right_hip_mse": 0.004328784998506308,
+        "val_right_hip_rmse": 0.06579350270738218,
+        "val_right_hip_mae": 0.04953312128782272,
+        "val_right_hip_r2": 0.48020052909851074,
+        "val_left_knee_mse": 0.004572615493088961,
+        "val_left_knee_rmse": 0.06762111721266487,
+        "val_left_knee_mae": 0.050859928131103516,
+        "val_left_knee_r2": 0.5373727083206177,
+        "val_right_knee_mse": 0.004504144191741943,
+        "val_right_knee_rmse": 0.06711292119809674,
+        "val_right_knee_mae": 0.05090361833572388,
+        "val_right_knee_r2": 0.566577672958374,
+        "val_left_foot_mse": 0.00428021652624011,
+        "val_left_foot_rmse": 0.06542336376433201,
+        "val_left_foot_mae": 0.04595830664038658,
+        "val_left_foot_r2": 0.05436980724334717,
+        "val_right_foot_mse": 0.0041213990189135075,
+        "val_right_foot_rmse": 0.06419812317282732,
+        "val_right_foot_mae": 0.04619848355650902,
+        "val_right_foot_r2": 0.0391697883605957
+      },
+      "best_epoch": 71
+    }
+  ]
+}

A9/cv_results_different_models/gru_fold1_best.h5

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

A9/cv_results_different_models/gru_fold2_best.h5

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

A9/cv_results_different_models/gru_fold3_best.h5

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