diff --git a/A9/all_models_config.py b/A9/all_models_config.py new file mode 100644 index 0000000000000000000000000000000000000000..d252e730dc169c297611876ff602c3d047b260b7 --- /dev/null +++ b/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', + # }, +} diff --git a/A9/all_optimizery.py b/A9/all_optimizery.py new file mode 100644 index 0000000000000000000000000000000000000000..a5a9552d3430d1e19a9ff06332a8481c99d6edb5 --- /dev/null +++ b/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' + }, +} diff --git a/A9/cv_results_conv1D_variants/conv1d_cv_results.json b/A9/cv_results_conv1D_variants/conv1d_cv_results.json new file mode 100644 index 0000000000000000000000000000000000000000..8b445ee1943e170b6e8534d3fe595e39cf0b7510 --- /dev/null +++ b/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 + } +} \ No newline at end of file diff --git a/A9/cv_results_conv1D_variants/conv1d_fold1_best.h5 b/A9/cv_results_conv1D_variants/conv1d_fold1_best.h5 new file mode 100644 index 0000000000000000000000000000000000000000..3e8413ef5346aca2c029e7c335d4694e98170812 --- /dev/null +++ b/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 diff --git a/A9/cv_results_conv1D_variants/conv1d_fold2_best.h5 b/A9/cv_results_conv1D_variants/conv1d_fold2_best.h5 new file mode 100644 index 0000000000000000000000000000000000000000..68d1a7ad8ccb6959a384aa07934fb9dc38262cdf --- /dev/null +++ b/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 diff --git a/A9/cv_results_conv1D_variants/conv1d_fold3_best.h5 b/A9/cv_results_conv1D_variants/conv1d_fold3_best.h5 new file mode 100644 index 0000000000000000000000000000000000000000..4f99ab8c3238b06d9908ddd188c4ca532ee5f1d4 --- /dev/null +++ b/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 diff --git a/A9/cv_results_conv1D_variants/conv1d_fold4_best.h5 b/A9/cv_results_conv1D_variants/conv1d_fold4_best.h5 new file mode 100644 index 0000000000000000000000000000000000000000..3d0eeeeb5d80ac209ede971b84eb81012590cb30 --- /dev/null +++ b/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 diff --git a/A9/cv_results_conv1D_variants/conv1d_fold5_best.h5 b/A9/cv_results_conv1D_variants/conv1d_fold5_best.h5 new file mode 100644 index 0000000000000000000000000000000000000000..5349456ba721fa1ef00f6c627e27fd5785bbfcd7 --- /dev/null +++ b/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 diff --git a/A9/cv_results_conv1D_variants/conv1d_predictions.png b/A9/cv_results_conv1D_variants/conv1d_predictions.png new file mode 100644 index 0000000000000000000000000000000000000000..0e8ed075fde0d07fd62ab6988d8cd0bfe8dd7117 --- /dev/null +++ b/A9/cv_results_conv1D_variants/conv1d_predictions.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:d927ce5ec6446ad0a4927eb1d4aa4a088b0f1b723914174a49500746d21a5ab2 +size 608778 diff --git a/A9/cv_results_conv1D_variants/conv1d_test_predictions.png b/A9/cv_results_conv1D_variants/conv1d_test_predictions.png new file mode 100644 index 0000000000000000000000000000000000000000..b63a37baca7d279d3d247c313ab07906973f28b2 --- /dev/null +++ b/A9/cv_results_conv1D_variants/conv1d_test_predictions.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:35c234c0b8feea3e0f69f0dc121802de5f883dd699c24fc62fe8e3afb2ea4078 +size 486673 diff --git a/A9/cv_results_conv1D_variants/conv1d_training_history.png b/A9/cv_results_conv1D_variants/conv1d_training_history.png new file mode 100644 index 0000000000000000000000000000000000000000..02302c28a5e588591d501e77e3722d984c224fff --- /dev/null +++ b/A9/cv_results_conv1D_variants/conv1d_training_history.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:497d140d24a692dc9b3631b2873c63468a469455c944fff496081ac40b06f4fc +size 369122 diff --git a/A9/cv_results_conv1D_variants/conv1d_v2_cv_results.json b/A9/cv_results_conv1D_variants/conv1d_v2_cv_results.json new file mode 100644 index 0000000000000000000000000000000000000000..c080e9da70564358321b178f4892c7fefe86ef45 --- /dev/null +++ b/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 + } +} \ No newline at end of file diff --git a/A9/cv_results_conv1D_variants/conv1d_v2_fold1_best.h5 b/A9/cv_results_conv1D_variants/conv1d_v2_fold1_best.h5 new file mode 100644 index 0000000000000000000000000000000000000000..141e4553f2b023116cbb411eb253cb61bc6eb04f --- /dev/null +++ b/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 diff --git a/A9/cv_results_conv1D_variants/conv1d_v2_fold2_best.h5 b/A9/cv_results_conv1D_variants/conv1d_v2_fold2_best.h5 new file mode 100644 index 0000000000000000000000000000000000000000..0022ae13fac0f00b5084f9fe8509baea5f480081 --- /dev/null +++ b/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 diff --git a/A9/cv_results_conv1D_variants/conv1d_v2_fold3_best.h5 b/A9/cv_results_conv1D_variants/conv1d_v2_fold3_best.h5 new file mode 100644 index 0000000000000000000000000000000000000000..b4b56292da556f43d6a736cde3cd60c1e2b924b9 --- /dev/null +++ b/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 diff --git a/A9/cv_results_conv1D_variants/conv1d_v2_fold4_best.h5 b/A9/cv_results_conv1D_variants/conv1d_v2_fold4_best.h5 new file mode 100644 index 0000000000000000000000000000000000000000..8a81393de0f604f38a21c8e144313a9ed260eda4 --- /dev/null +++ b/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 diff --git a/A9/cv_results_conv1D_variants/conv1d_v2_fold5_best.h5 b/A9/cv_results_conv1D_variants/conv1d_v2_fold5_best.h5 new file mode 100644 index 0000000000000000000000000000000000000000..a081100cd71310ee28a55de8b76a10f5b918239e --- /dev/null +++ b/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 diff --git a/A9/cv_results_conv1D_variants/conv1d_v2_predictions.png b/A9/cv_results_conv1D_variants/conv1d_v2_predictions.png new file mode 100644 index 0000000000000000000000000000000000000000..4d65afc584931093e0e0152e0e6474ffbce20f39 --- /dev/null +++ b/A9/cv_results_conv1D_variants/conv1d_v2_predictions.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c87ece9ba6b7ba116ed495ab6478bf0b5c3acd22499f8722c115c7a076f7403d +size 671567 diff --git a/A9/cv_results_conv1D_variants/conv1d_v2_test_predictions.png b/A9/cv_results_conv1D_variants/conv1d_v2_test_predictions.png new file mode 100644 index 0000000000000000000000000000000000000000..82bee7c6dc19353947a7f9db638699d555bb4da6 --- /dev/null +++ b/A9/cv_results_conv1D_variants/conv1d_v2_test_predictions.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:bfe0411c2c173beed666f25de2c0f94d9d20ed50c52ceadacd4d38fec8fd4542 +size 503786 diff --git a/A9/cv_results_conv1D_variants/conv1d_v2_training_history.png b/A9/cv_results_conv1D_variants/conv1d_v2_training_history.png new file mode 100644 index 0000000000000000000000000000000000000000..886afb9e0cf19d7d66c3ad0cf00e467d4d53682a --- /dev/null +++ b/A9/cv_results_conv1D_variants/conv1d_v2_training_history.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:cc6bd005c51652d12c560a4d54736b1997ce91fb7dc40e96af87d43d02571061 +size 347310 diff --git a/A9/cv_results_conv1D_variants/conv1d_v3_cv_results.json b/A9/cv_results_conv1D_variants/conv1d_v3_cv_results.json new file mode 100644 index 0000000000000000000000000000000000000000..1da756350640a692c2a14dde2b7f54a60fd79b86 --- /dev/null +++ b/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 + } +} \ No newline at end of file diff --git a/A9/cv_results_conv1D_variants/conv1d_v3_fold1_best.h5 b/A9/cv_results_conv1D_variants/conv1d_v3_fold1_best.h5 new file mode 100644 index 0000000000000000000000000000000000000000..1331bf07834fba624d5d50f124f7e55b96f650d9 --- /dev/null +++ b/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 diff --git a/A9/cv_results_conv1D_variants/conv1d_v3_fold2_best.h5 b/A9/cv_results_conv1D_variants/conv1d_v3_fold2_best.h5 new file mode 100644 index 0000000000000000000000000000000000000000..9c28b8c76daebbf11a38ea777568c65fa0afaf18 --- /dev/null +++ b/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 diff --git a/A9/cv_results_conv1D_variants/conv1d_v3_fold3_best.h5 b/A9/cv_results_conv1D_variants/conv1d_v3_fold3_best.h5 new file mode 100644 index 0000000000000000000000000000000000000000..699d1f7b2844827453d93497c4a4a5b0a6bf5483 --- /dev/null +++ b/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 diff --git a/A9/cv_results_conv1D_variants/conv1d_v3_fold4_best.h5 b/A9/cv_results_conv1D_variants/conv1d_v3_fold4_best.h5 new file mode 100644 index 0000000000000000000000000000000000000000..a020a1dcb66a6db50b4703c6cf18ec9680773268 --- /dev/null +++ b/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 diff --git a/A9/cv_results_conv1D_variants/conv1d_v3_fold5_best.h5 b/A9/cv_results_conv1D_variants/conv1d_v3_fold5_best.h5 new file mode 100644 index 0000000000000000000000000000000000000000..3c93d12584ea9602cf49511f9335adf6e75ef1a8 --- /dev/null +++ b/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 diff --git a/A9/cv_results_conv1D_variants/conv1d_v3_predictions.png b/A9/cv_results_conv1D_variants/conv1d_v3_predictions.png new file mode 100644 index 0000000000000000000000000000000000000000..7db8cc3be7d28ae646272d6cd9a985e53944781c --- /dev/null +++ b/A9/cv_results_conv1D_variants/conv1d_v3_predictions.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:3ceb771ed31cc30686071be4557b9c81f81205ec1c48a367a38ecf161ea2cf2d +size 543588 diff --git a/A9/cv_results_conv1D_variants/conv1d_v3_test_predictions.png b/A9/cv_results_conv1D_variants/conv1d_v3_test_predictions.png new file mode 100644 index 0000000000000000000000000000000000000000..7395b40e0e5717b7fb7be7a2191e4b331dfd5b3b --- /dev/null +++ b/A9/cv_results_conv1D_variants/conv1d_v3_test_predictions.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:a9b325bae657578fb2dd973a4be5b814358762bff4b72e465ed59b2cb4add66e +size 461704 diff --git a/A9/cv_results_conv1D_variants/conv1d_v3_training_history.png b/A9/cv_results_conv1D_variants/conv1d_v3_training_history.png new file mode 100644 index 0000000000000000000000000000000000000000..762857023dacac6572da80156afa1d07468ad861 --- /dev/null +++ b/A9/cv_results_conv1D_variants/conv1d_v3_training_history.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:cef7f1b366748e4ca02fa5ecc46c3a4c219b1dbbe78afdef8903f367f325e99b +size 396332 diff --git a/A9/cv_results_conv1D_variants/summary.txt b/A9/cv_results_conv1D_variants/summary.txt new file mode 100644 index 0000000000000000000000000000000000000000..446591097137de97d51a0d1a1c7230a2c3c412b9 --- /dev/null +++ b/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 diff --git a/A9/cv_results_different_models/conv1d_cv_results.json b/A9/cv_results_different_models/conv1d_cv_results.json new file mode 100644 index 0000000000000000000000000000000000000000..837a26903208a07a993ae5cb6737ee8ac648b525 --- /dev/null +++ b/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 + } + ] +} \ No newline at end of file diff --git a/A9/cv_results_different_models/conv1d_fold1_best.h5 b/A9/cv_results_different_models/conv1d_fold1_best.h5 new file mode 100644 index 0000000000000000000000000000000000000000..e31e53736a88ad258b7868f9d7249c0d1535babe --- /dev/null +++ b/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 diff --git a/A9/cv_results_different_models/conv1d_fold2_best.h5 b/A9/cv_results_different_models/conv1d_fold2_best.h5 new file mode 100644 index 0000000000000000000000000000000000000000..e6b81b2856504b1618e08ae0cc4bdbd7a45e4617 --- /dev/null +++ b/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 diff --git a/A9/cv_results_different_models/conv1d_fold3_best.h5 b/A9/cv_results_different_models/conv1d_fold3_best.h5 new file mode 100644 index 0000000000000000000000000000000000000000..d38dea77f5a86e215c3e560aa179ba7dcd83d325 --- /dev/null +++ b/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 diff --git a/A9/cv_results_different_models/conv1d_fold4_best.h5 b/A9/cv_results_different_models/conv1d_fold4_best.h5 new file mode 100644 index 0000000000000000000000000000000000000000..ae490d78651cb5d903a242d9f4a8642440daf032 --- /dev/null +++ b/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 diff --git a/A9/cv_results_different_models/conv1d_fold5_best.h5 b/A9/cv_results_different_models/conv1d_fold5_best.h5 new file mode 100644 index 0000000000000000000000000000000000000000..9dd73c3d869c3ec550fda93d7b06923e6b6dd65a --- /dev/null +++ b/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 diff --git a/A9/cv_results_different_models/conv1d_predictions.png b/A9/cv_results_different_models/conv1d_predictions.png new file mode 100644 index 0000000000000000000000000000000000000000..d5f1f12da9bed18e57dca211d33ac6b4e4258192 --- /dev/null +++ b/A9/cv_results_different_models/conv1d_predictions.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c2624cdc056a476b5c615897cc3264928b1762c77d230d6aa0ac2d6519bee142 +size 589078 diff --git a/A9/cv_results_different_models/conv1d_training_history.png b/A9/cv_results_different_models/conv1d_training_history.png new file mode 100644 index 0000000000000000000000000000000000000000..6776fcc55ce942855debfdc9f70f309adee48b1d --- /dev/null +++ b/A9/cv_results_different_models/conv1d_training_history.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e8ec51d8f26b732b8f4a9a5eddb20474f64fcb5b1153cde05687fff605d9ae99 +size 376184 diff --git a/A9/cv_results_different_models/dense_cv_results.json b/A9/cv_results_different_models/dense_cv_results.json new file mode 100644 index 0000000000000000000000000000000000000000..8be953d5932abbdfa3d16e6d498597a6b7eb24b8 --- /dev/null +++ b/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 + } + ] +} \ No newline at end of file diff --git a/A9/cv_results_different_models/dense_fold1_best.h5 b/A9/cv_results_different_models/dense_fold1_best.h5 new file mode 100644 index 0000000000000000000000000000000000000000..ab5f53063345e113917b7c6febcc445f7664a67e --- /dev/null +++ b/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 diff --git a/A9/cv_results_different_models/dense_fold2_best.h5 b/A9/cv_results_different_models/dense_fold2_best.h5 new file mode 100644 index 0000000000000000000000000000000000000000..da7aa3e90d58c325c36e18c190f81fd24e412691 --- /dev/null +++ b/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 diff --git a/A9/cv_results_different_models/dense_fold3_best.h5 b/A9/cv_results_different_models/dense_fold3_best.h5 new file mode 100644 index 0000000000000000000000000000000000000000..5ee4584c3c55bd9cf150cbea5e3b8b6728c848b3 --- /dev/null +++ b/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 diff --git a/A9/cv_results_different_models/dense_fold4_best.h5 b/A9/cv_results_different_models/dense_fold4_best.h5 new file mode 100644 index 0000000000000000000000000000000000000000..f8668310e39fc3e10fbc4fac355a277d1545a6bc --- /dev/null +++ b/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 diff --git a/A9/cv_results_different_models/dense_fold5_best.h5 b/A9/cv_results_different_models/dense_fold5_best.h5 new file mode 100644 index 0000000000000000000000000000000000000000..2a7273c9a0fd165575f6c5d6be9c20e9e90312c1 --- /dev/null +++ b/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 diff --git a/A9/cv_results_different_models/dense_predictions.png b/A9/cv_results_different_models/dense_predictions.png new file mode 100644 index 0000000000000000000000000000000000000000..0b49b1bd97004f67be2a4c0fb7a964df8f220a7e --- /dev/null +++ b/A9/cv_results_different_models/dense_predictions.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e5fe71eb697ac97041815f40256d85ac763e8b1ebabedaca5063f1340c951b4b +size 724794 diff --git a/A9/cv_results_different_models/dense_training_history.png b/A9/cv_results_different_models/dense_training_history.png new file mode 100644 index 0000000000000000000000000000000000000000..ad0814ce47abab2a6c843e8c880fca0ebd7b6197 --- /dev/null +++ b/A9/cv_results_different_models/dense_training_history.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:9543ce5258e9d6d4f0066a9f13cfb2f49c052f60c807670cc4228926dd86de63 +size 343734 diff --git a/A9/cv_results_different_models/gru_cv_results.json b/A9/cv_results_different_models/gru_cv_results.json new file mode 100644 index 0000000000000000000000000000000000000000..3c4f3fc7f335a5032eab89de01fb0028be181e12 --- /dev/null +++ b/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 + } + ] +} \ No newline at end of file diff --git a/A9/cv_results_different_models/gru_fold1_best.h5 b/A9/cv_results_different_models/gru_fold1_best.h5 new file mode 100644 index 0000000000000000000000000000000000000000..f1a747581a8da8ad85c19798815f9741c9d13dc7 --- /dev/null +++ b/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 diff --git a/A9/cv_results_different_models/gru_fold2_best.h5 b/A9/cv_results_different_models/gru_fold2_best.h5 new file mode 100644 index 0000000000000000000000000000000000000000..09b1814bd0d46b2a060b982258516f47589d6e11 --- /dev/null +++ b/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 diff --git a/A9/cv_results_different_models/gru_fold3_best.h5 b/A9/cv_results_different_models/gru_fold3_best.h5 new file mode 100644 index 0000000000000000000000000000000000000000..b18155f0e4bb4d684c610a87fc1ce67ac858c824 --- /dev/null +++ b/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 diff --git a/A9/cv_results_different_models/gru_fold4_best.h5 b/A9/cv_results_different_models/gru_fold4_best.h5 new file mode 100644 index 0000000000000000000000000000000000000000..6717ec41c4a9dfcd80c0ff1be4177c4a5f1131bb --- /dev/null +++ b/A9/cv_results_different_models/gru_fold4_best.h5 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:77b8843155eb0fbe5b1310349956113ee38830231374787194f6294eaf13b1a7 +size 383816 diff --git a/A9/cv_results_different_models/gru_fold5_best.h5 b/A9/cv_results_different_models/gru_fold5_best.h5 new file mode 100644 index 0000000000000000000000000000000000000000..5deff60860d7fda4ac97714a4d72e044fdbc71e1 --- /dev/null +++ b/A9/cv_results_different_models/gru_fold5_best.h5 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:fc0d8e431bdc7ac8bc10ba55ac2230532436b28b1e9176a80873fd00d4b7ba14 +size 383816 diff --git a/A9/cv_results_different_models/gru_predictions.png b/A9/cv_results_different_models/gru_predictions.png new file mode 100644 index 0000000000000000000000000000000000000000..f6f2778a400ca87ecdb0f3bc6d706016871b4dbb --- /dev/null +++ b/A9/cv_results_different_models/gru_predictions.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:36253a1441977ce94598eedca2bef1256db0f31fd95372612e6bec37f9d08460 +size 737718 diff --git a/A9/cv_results_different_models/gru_training_history.png b/A9/cv_results_different_models/gru_training_history.png new file mode 100644 index 0000000000000000000000000000000000000000..3f23de2133abcb5ad004266512b9fb6e8705bdc4 --- /dev/null +++ b/A9/cv_results_different_models/gru_training_history.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:de27ea33c1b2b0ab4b9fff95a6c72c0bbef57b94425e9d7f70d6836a57e5b51f +size 373757 diff --git a/A9/cv_results_different_models/lstm_cv_results.json b/A9/cv_results_different_models/lstm_cv_results.json new file mode 100644 index 0000000000000000000000000000000000000000..57741dac80b1bda43bb169dc7d1b361d1f574c1d --- /dev/null +++ b/A9/cv_results_different_models/lstm_cv_results.json @@ -0,0 +1,1185 @@ +{ + "model_name": "lstm", + "n_splits": 5, + "epochs": 100, + "batch_size": 32, + "patience": 15, + "aggregate_results": { + "n_folds": 5, + "train_metrics": { + "train_mse": { + "mean": 0.004811993613839149, + "std": 0.0002131097957991066 + }, + "train_rmse": { + "mean": 0.06935143616584538, + "std": 0.001540102455625147 + }, + "train_mae": { + "mean": 0.05187585130333901, + "std": 0.0012661690660198592 + }, + "train_r2": { + "mean": 0.7727830886840821, + "std": 0.009728731445154976 + }, + "train_head_mse": { + "mean": 0.003271667240187526, + "std": 0.00020290710468368352 + }, + "train_head_rmse": { + "mean": 0.057170760973923505, + "std": 0.0017808228014125557 + }, + "train_head_mae": { + "mean": 0.04582602828741074, + "std": 0.0014678009347576986 + }, + "train_head_r2": { + "mean": 0.6727981686592102, + "std": 0.018754256572428223 + }, + "train_left_shoulder_mse": { + "mean": 0.002229378232732415, + "std": 0.00010874577207164447 + }, + "train_left_shoulder_rmse": { + "mean": 0.047202083883603266, + "std": 0.0011582356313357069 + }, + "train_left_shoulder_mae": { + "mean": 0.03726118803024292, + "std": 0.001010821144529126 + }, + "train_left_shoulder_r2": { + "mean": 0.6010156393051147, + "std": 0.019080986789245677 + }, + "train_left_elbow_mse": { + "mean": 0.00461384691298008, + "std": 0.000304282574186518 + }, + "train_left_elbow_rmse": { + "mean": 0.06788790998765781, + "std": 0.0022535728272571093 + }, + "train_left_elbow_mae": { + "mean": 0.053285499662160875, + "std": 0.0019497447046475265 + }, + "train_left_elbow_r2": { + "mean": 0.6646448969841003, + "std": 0.02084243627920227 + }, + "train_right_shoulder_mse": { + "mean": 0.002878066524863243, + "std": 0.00011818537600344773 + }, + "train_right_shoulder_rmse": { + "mean": 0.05363633162792254, + "std": 0.0011002065091427064 + }, + "train_right_shoulder_mae": { + "mean": 0.04229972958564758, + "std": 0.0010288763513419995 + }, + "train_right_shoulder_r2": { + "mean": 0.5606404066085815, + "std": 0.016701753768302015 + }, + "train_right_elbow_mse": { + "mean": 0.005865630507469177, + "std": 0.0002807402937432626 + }, + "train_right_elbow_rmse": { + "mean": 0.07656529943282234, + "std": 0.0018399538666587418 + }, + "train_right_elbow_mae": { + "mean": 0.059941603988409045, + "std": 0.001835144265114935 + }, + "train_right_elbow_r2": { + "mean": 0.6275461673736572, + "std": 0.016255766772795374 + }, + "train_left_hand_mse": { + "mean": 0.009051253460347653, + "std": 0.0005877893887642696 + }, + "train_left_hand_rmse": { + "mean": 0.09508782484528355, + "std": 0.0030917675430614296 + }, + "train_left_hand_mae": { + "mean": 0.07584438920021057, + "std": 0.002458797028863905 + }, + "train_left_hand_r2": { + "mean": 0.655476176738739, + "std": 0.021182577860682957 + }, + "train_right_hand_mse": { + "mean": 0.010388927161693573, + "std": 0.000548523389040045 + }, + "train_right_hand_rmse": { + "mean": 0.10189014811360209, + "std": 0.0027064513817548802 + }, + "train_right_hand_mae": { + "mean": 0.0797205463051796, + "std": 0.002407860846049173 + }, + "train_right_hand_r2": { + "mean": 0.6277915835380554, + "std": 0.01850585806941287 + }, + "train_left_hip_mse": { + "mean": 0.0039705156814306974, + "std": 0.00016057163559198432 + }, + "train_left_hip_rmse": { + "mean": 0.06299929179772432, + "std": 0.0012668521681249503 + }, + "train_left_hip_mae": { + "mean": 0.04646013602614403, + "std": 0.001936593451455194 + }, + "train_left_hip_r2": { + "mean": 0.5033331513404846, + "std": 0.027553907715825717 + }, + "train_right_hip_mse": { + "mean": 0.004132610373198986, + "std": 0.00017648508347408294 + }, + "train_right_hip_rmse": { + "mean": 0.06427087720692025, + "std": 0.0013655461368863352 + }, + "train_right_hip_mae": { + "mean": 0.047165734320878984, + "std": 0.002074278416348818 + }, + "train_right_hip_r2": { + "mean": 0.503155517578125, + "std": 0.028534095205183897 + }, + "train_left_knee_mse": { + "mean": 0.0037466858979314567, + "std": 0.00019619452148506155 + }, + "train_left_knee_rmse": { + "mean": 0.061189135174471866, + "std": 0.0016048783541646656 + }, + "train_left_knee_mae": { + "mean": 0.04554715156555176, + "std": 0.0010775964249682142 + }, + "train_left_knee_r2": { + "mean": 0.6069826006889343, + "std": 0.02189539226934561 + }, + "train_right_knee_mse": { + "mean": 0.0038714648224413397, + "std": 0.00019323724090642745 + }, + "train_right_knee_rmse": { + "mean": 0.062201609040776554, + "std": 0.0015571304311858209 + }, + "train_right_knee_mae": { + "mean": 0.045899265259504315, + "std": 0.0009115145463423917 + }, + "train_right_knee_r2": { + "mean": 0.6398303627967834, + "std": 0.018329815460619604 + }, + "train_left_foot_mse": { + "mean": 0.004146361071616411, + "std": 3.531320596230505e-05 + }, + "train_left_foot_rmse": { + "mean": 0.06439166018454112, + "std": 0.0002741720171587229 + }, + "train_left_foot_mae": { + "mean": 0.046790570765733716, + "std": 0.0003340319112892813 + }, + "train_left_foot_r2": { + "mean": 0.06377550363540649, + "std": 0.008785930659807565 + }, + "train_right_foot_mse": { + "mean": 0.0043895099312067035, + "std": 7.47805225202785e-05 + }, + "train_right_foot_rmse": { + "mean": 0.06625097638910672, + "std": 0.0005639669287529224 + }, + "train_right_foot_mae": { + "mean": 0.048344234377145766, + "std": 0.0004848134352129206 + }, + "train_right_foot_r2": { + "mean": 0.05367337465286255, + "std": 0.008791824422669953 + } + }, + "val_metrics": { + "val_mse": { + "mean": 0.004886408895254135, + "std": 0.0004338486514685952 + }, + "val_rmse": { + "mean": 0.06983400017995575, + "std": 0.0031018243212783177 + }, + "val_mae": { + "mean": 0.052217034995555876, + "std": 0.002198656746161434 + }, + "val_r2": { + "mean": 0.7685302019119262, + "std": 0.02583644020853613 + }, + "val_head_mse": { + "mean": 0.0033083776477724315, + "std": 0.00029724928458275165 + }, + "val_head_rmse": { + "mean": 0.057461230195601024, + "std": 0.0025660616088825933 + }, + "val_head_mae": { + "mean": 0.04619952812790871, + "std": 0.0019301102031651085 + }, + "val_head_r2": { + "mean": 0.6676291823387146, + "std": 0.03645821728754075 + }, + "val_left_shoulder_mse": { + "mean": 0.002256858209148049, + "std": 0.0002172621612961369 + }, + "val_left_shoulder_rmse": { + "mean": 0.04745097048303108, + "std": 0.00229425573259881 + }, + "val_left_shoulder_mae": { + "mean": 0.0374816469848156, + "std": 0.001754447318524106 + }, + "val_left_shoulder_r2": { + "mean": 0.594955050945282, + "std": 0.0412744158665287 + }, + "val_left_elbow_mse": { + "mean": 0.004725911188870668, + "std": 0.0003869573130244922 + }, + "val_left_elbow_rmse": { + "mean": 0.06868872239333598, + "std": 0.0027875804637519 + }, + "val_left_elbow_mae": { + "mean": 0.05384641215205192, + "std": 0.0019663407361249297 + }, + "val_left_elbow_r2": { + "mean": 0.6542348504066468, + "std": 0.04360785852929914 + }, + "val_right_shoulder_mse": { + "mean": 0.002915846323594451, + "std": 0.00027740351683731965 + }, + "val_right_shoulder_rmse": { + "mean": 0.05393724196463658, + "std": 0.0025729852006371933 + }, + "val_right_shoulder_mae": { + "mean": 0.04262771084904671, + "std": 0.00154498734102719 + }, + "val_right_shoulder_r2": { + "mean": 0.5532354950904846, + "std": 0.04579652745719623 + }, + "val_right_elbow_mse": { + "mean": 0.005983852874487638, + "std": 0.0005818281094070517 + }, + "val_right_elbow_rmse": { + "mean": 0.07726532667493534, + "std": 0.0037312421930457396 + }, + "val_right_elbow_mae": { + "mean": 0.06046148985624313, + "std": 0.0019967134499930075 + }, + "val_right_elbow_r2": { + "mean": 0.6176719069480896, + "std": 0.04891835495488435 + }, + "val_left_hand_mse": { + "mean": 0.009270233847200871, + "std": 0.0007680954554143014 + }, + "val_left_hand_rmse": { + "mean": 0.09620102380218841, + "std": 0.003949286848489095 + }, + "val_left_hand_mae": { + "mean": 0.07645980417728424, + "std": 0.002616737299022061 + }, + "val_left_hand_r2": { + "mean": 0.6445365309715271, + "std": 0.047369304769214304 + }, + "val_right_hand_mse": { + "mean": 0.010614957846701144, + "std": 0.0010608202168161608 + }, + "val_right_hand_rmse": { + "mean": 0.10290251625158817, + "std": 0.005101959995214896 + }, + "val_right_hand_mae": { + "mean": 0.08046282976865768, + "std": 0.002931328569792433 + }, + "val_right_hand_r2": { + "mean": 0.617678451538086, + "std": 0.048775871489845656 + }, + "val_left_hip_mse": { + "mean": 0.003988596191629767, + "std": 0.0007058671211842805 + }, + "val_left_hip_rmse": { + "mean": 0.06291429539714563, + "std": 0.005512497284395533 + }, + "val_left_hip_mae": { + "mean": 0.04667298644781113, + "std": 0.004191350844901293 + }, + "val_left_hip_r2": { + "mean": 0.5043633580207825, + "std": 0.05821224225688641 + }, + "val_right_hip_mse": { + "mean": 0.004152466962113977, + "std": 0.0007451240407493119 + }, + "val_right_hip_rmse": { + "mean": 0.0641850026598835, + "std": 0.005722970877500989 + }, + "val_right_hip_mae": { + "mean": 0.047387586534023286, + "std": 0.00443244464433604 + }, + "val_right_hip_r2": { + "mean": 0.5043165206909179, + "std": 0.058559126938994066 + }, + "val_left_knee_mse": { + "mean": 0.003811284340918064, + "std": 0.00038825898778595025 + }, + "val_left_knee_rmse": { + "mean": 0.06165592098746644, + "std": 0.0031355618484347402 + }, + "val_left_knee_mae": { + "mean": 0.04581769332289696, + "std": 0.0025675927555527792 + }, + "val_left_knee_r2": { + "mean": 0.6004354119300842, + "std": 0.03330259698453738 + }, + "val_right_knee_mse": { + "mean": 0.003929925430566072, + "std": 0.0004675380021491685 + }, + "val_right_knee_rmse": { + "mean": 0.06257456715805199, + "std": 0.003788004164268308 + }, + "val_right_knee_mae": { + "mean": 0.04614148139953613, + "std": 0.002917056600735037 + }, + "val_right_knee_r2": { + "mean": 0.634096622467041, + "std": 0.04270730709518828 + }, + "val_left_foot_mse": { + "mean": 0.00416032737120986, + "std": 0.00020977022863971975 + }, + "val_left_foot_rmse": { + "mean": 0.06447998430602217, + "std": 0.001630642543598489 + }, + "val_left_foot_mae": { + "mean": 0.04687633514404297, + "std": 0.0013549831974382473 + }, + "val_left_foot_r2": { + "mean": 0.06079589128494263, + "std": 0.02445561121174172 + }, + "val_right_foot_mse": { + "mean": 0.004404677962884307, + "std": 0.00043151900975307844 + }, + "val_right_foot_rmse": { + "mean": 0.0662876133582093, + "std": 0.0032604107343796288 + }, + "val_right_foot_mae": { + "mean": 0.04838596805930138, + "std": 0.0018809785459679516 + }, + "val_right_foot_r2": { + "mean": 0.050543367862701416, + "std": 0.010651224475362453 + } + }, + "best_fold": "4", + "best_fold_metrics": { + "fold": 4, + "train_metrics": { + "train_mse": 0.004498118069022894, + "train_rmse": 0.06706801077281847, + "train_mae": 0.04993272200226784, + "train_r2": 0.7869763374328613, + "train_head_mse": 0.00296544237062335, + "train_head_rmse": 0.05445587544630377, + "train_head_mae": 0.043518174439668655, + "train_head_r2": 0.7006955146789551, + "train_left_shoulder_mse": 0.002048891270533204, + "train_left_shoulder_rmse": 0.04526468016603237, + "train_left_shoulder_mae": 0.03560788184404373, + "train_left_shoulder_r2": 0.6312171816825867, + "train_left_elbow_mse": 0.0041388217359781265, + "train_left_elbow_rmse": 0.06433367497647034, + "train_left_elbow_mae": 0.05006039887666702, + "train_left_elbow_r2": 0.6964722871780396, + "train_right_shoulder_mse": 0.0027137314900755882, + "train_right_shoulder_rmse": 0.052093487981470274, + "train_right_shoulder_mae": 0.04076620563864708, + "train_right_shoulder_r2": 0.5850533843040466, + "train_right_elbow_mse": 0.0054282862693071365, + "train_right_elbow_rmse": 0.07367690458554252, + "train_right_elbow_mae": 0.056952714920043945, + "train_right_elbow_r2": 0.6527020931243896, + "train_left_hand_mse": 0.008239386603236198, + "train_left_hand_rmse": 0.09077106699403835, + "train_left_hand_mae": 0.07196556031703949, + "train_left_hand_r2": 0.6841086149215698, + "train_right_hand_mse": 0.009507128037512302, + "train_right_hand_rmse": 0.09750450265250474, + "train_right_hand_mae": 0.0755743756890297, + "train_right_hand_r2": 0.6570142507553101, + "train_left_hip_mse": 0.003784741507843137, + "train_left_hip_rmse": 0.061520252826554093, + "train_left_hip_mae": 0.04455612599849701, + "train_left_hip_r2": 0.5355053544044495, + "train_right_hip_mse": 0.003950424958020449, + "train_right_hip_rmse": 0.06285240614344409, + "train_right_hip_mae": 0.04521644860506058, + "train_right_hip_r2": 0.5355268716812134, + "train_left_knee_mse": 0.003496406599879265, + "train_left_knee_rmse": 0.0591304202579287, + "train_left_knee_mae": 0.04464443400502205, + "train_left_knee_r2": 0.6328892707824707, + "train_right_knee_mse": 0.003604211611673236, + "train_right_knee_rmse": 0.06003508650508664, + "train_right_knee_mae": 0.045035526156425476, + "train_right_knee_r2": 0.6632816791534424, + "train_left_foot_mse": 0.004139684606343508, + "train_left_foot_rmse": 0.06434038083772514, + "train_left_foot_mae": 0.046888403594493866, + "train_left_foot_r2": 0.07479500770568848, + "train_right_foot_mse": 0.004458376672118902, + "train_right_foot_rmse": 0.0667710766134477, + "train_right_foot_mae": 0.04833918809890747, + "train_right_foot_r2": 0.06819337606430054 + }, + "val_metrics": { + "val_mse": 0.004327595233917236, + "val_rmse": 0.06578446042886751, + "val_mae": 0.04989447444677353, + "val_r2": 0.7980234622955322, + "val_head_mse": 0.0029733891133219004, + "val_head_rmse": 0.05452879159968521, + "val_head_mae": 0.04494099318981171, + "val_head_r2": 0.7128691673278809, + "val_left_shoulder_mse": 0.002064111875370145, + "val_left_shoulder_rmse": 0.04543249800935609, + "val_left_shoulder_mae": 0.03673253580927849, + "val_left_shoulder_r2": 0.6389738321304321, + "val_left_elbow_mse": 0.004366486798971891, + "val_left_elbow_rmse": 0.0660793976892336, + "val_left_elbow_mae": 0.05158422514796257, + "val_left_elbow_r2": 0.6932650804519653, + "val_right_shoulder_mse": 0.0025226348079741, + "val_right_shoulder_rmse": 0.05022583805148601, + "val_right_shoulder_mae": 0.04109163209795952, + "val_right_shoulder_r2": 0.6175146102905273, + "val_right_elbow_mse": 0.005406757351011038, + "val_right_elbow_rmse": 0.07353065585870316, + "val_right_elbow_mae": 0.05781988427042961, + "val_right_elbow_r2": 0.666649580001831, + "val_left_hand_mse": 0.008402742445468903, + "val_left_hand_rmse": 0.0916664739447793, + "val_left_hand_mae": 0.0731738731265068, + "val_left_hand_r2": 0.6889402866363525, + "val_right_hand_mse": 0.009680505841970444, + "val_right_hand_rmse": 0.0983895616514803, + "val_right_hand_mae": 0.07751485705375671, + "val_right_hand_r2": 0.6624351739883423, + "val_left_hip_mse": 0.0031526698730885983, + "val_left_hip_rmse": 0.05614864088371684, + "val_left_hip_mae": 0.04147360846400261, + "val_left_hip_r2": 0.574637770652771, + "val_right_hip_mse": 0.0031944497022777796, + "val_right_hip_rmse": 0.056519463039538684, + "val_right_hip_mae": 0.041883185505867004, + "val_right_hip_r2": 0.5797367095947266, + "val_left_knee_mse": 0.0033285561949014664, + "val_left_knee_rmse": 0.0576936408532298, + "val_left_knee_mae": 0.04376426339149475, + "val_left_knee_r2": 0.6525990962982178, + "val_right_knee_mse": 0.003496857825666666, + "val_right_knee_rmse": 0.0591342356479448, + "val_right_knee_mae": 0.04514757916331291, + "val_right_knee_r2": 0.6802155375480652, + "val_left_foot_mse": 0.0038629467599093914, + "val_left_foot_rmse": 0.06215260863318121, + "val_left_foot_mae": 0.045711204409599304, + "val_left_foot_r2": 0.0904436707496643, + "val_right_foot_mse": 0.0038066308479756117, + "val_right_foot_rmse": 0.06169789986681566, + "val_right_foot_mae": 0.04779032990336418, + "val_right_foot_r2": 0.061976730823516846 + }, + "best_epoch": 62 + } + }, + "fold_details": [ + { + "fold": 1, + "train_metrics": { + "train_mse": 0.005062841810286045, + "train_rmse": 0.07115364931109328, + "train_mae": 0.053328368812799454, + "train_r2": 0.7605157494544983, + "train_head_mse": 0.0034392497036606073, + "train_head_rmse": 0.05864511662244869, + "train_head_mae": 0.046843022108078, + "train_head_r2": 0.6543329358100891, + "train_left_shoulder_mse": 0.002292995573952794, + "train_left_shoulder_rmse": 0.04788523336011629, + "train_left_shoulder_mae": 0.037935927510261536, + "train_left_shoulder_r2": 0.5875546932220459, + "train_left_elbow_mse": 0.004977520555257797, + "train_left_elbow_rmse": 0.07055154537823957, + "train_left_elbow_mae": 0.05514269322156906, + "train_left_elbow_r2": 0.6395151019096375, + "train_right_shoulder_mse": 0.0029234718531370163, + "train_right_shoulder_rmse": 0.054069139563497924, + "train_right_shoulder_mae": 0.04291445389389992, + "train_right_shoulder_r2": 0.5519082546234131, + "train_right_elbow_mse": 0.006128083448857069, + "train_right_elbow_rmse": 0.07828207616598495, + "train_right_elbow_mae": 0.06117560714483261, + "train_right_elbow_r2": 0.6115542650222778, + "train_left_hand_mse": 0.00987115316092968, + "train_left_hand_rmse": 0.09935367713844154, + "train_left_hand_mae": 0.07861798256635666, + "train_left_hand_r2": 0.6245157718658447, + "train_right_hand_mse": 0.01098676398396492, + "train_right_hand_rmse": 0.10481776559326629, + "train_right_hand_mae": 0.08116581290960312, + "train_right_hand_r2": 0.6055518984794617, + "train_left_hip_mse": 0.004240549635142088, + "train_left_hip_rmse": 0.0651195027249294, + "train_left_hip_mae": 0.05000155791640282, + "train_left_hip_r2": 0.45616579055786133, + "train_right_hip_mse": 0.004414218012243509, + "train_right_hip_rmse": 0.06643958166818564, + "train_right_hip_mae": 0.05087462440133095, + "train_right_hip_r2": 0.45604825019836426, + "train_left_knee_mse": 0.00400224793702364, + "train_left_knee_rmse": 0.06326332220982107, + "train_left_knee_mae": 0.04737363010644913, + "train_left_knee_r2": 0.578141450881958, + "train_right_knee_mse": 0.004114257637411356, + "train_right_knee_rmse": 0.06414247919601608, + "train_right_knee_mae": 0.0473116971552372, + "train_right_knee_r2": 0.6175259947776794, + "train_left_foot_mse": 0.004128873813897371, + "train_left_foot_rmse": 0.06425631341663923, + "train_left_foot_mae": 0.046350762248039246, + "train_left_foot_r2": 0.05595117807388306, + "train_right_foot_mse": 0.00429756473749876, + "train_right_foot_rmse": 0.06555581391073381, + "train_right_foot_mae": 0.04756103828549385, + "train_right_foot_r2": 0.04167675971984863 + }, + "val_metrics": { + "val_mse": 0.005467141047120094, + "val_rmse": 0.07394011798151322, + "val_mae": 0.05568956211209297, + "val_r2": 0.7436360120773315, + "val_head_mse": 0.0035457804333418608, + "val_head_rmse": 0.05954645609389245, + "val_head_mae": 0.04855097085237503, + "val_head_r2": 0.6519829034805298, + "val_left_shoulder_mse": 0.002557316329330206, + "val_left_shoulder_rmse": 0.05056991525927452, + "val_left_shoulder_mae": 0.03986159339547157, + "val_left_shoulder_r2": 0.5513902902603149, + "val_left_elbow_mse": 0.004995889961719513, + "val_left_elbow_rmse": 0.07068160978443766, + "val_left_elbow_mae": 0.05558169260621071, + "val_left_elbow_r2": 0.6312785744667053, + "val_right_shoulder_mse": 0.0032826608512550592, + "val_right_shoulder_rmse": 0.05729450978283224, + "val_right_shoulder_mae": 0.045232612639665604, + "val_right_shoulder_r2": 0.5069179534912109, + "val_right_elbow_mse": 0.00661214254796505, + "val_right_elbow_rmse": 0.08131508192189842, + "val_right_elbow_mae": 0.06292656809091568, + "val_right_elbow_r2": 0.5770868062973022, + "val_left_hand_mse": 0.00976537261158228, + "val_left_hand_rmse": 0.09881989987640283, + "val_left_hand_mae": 0.07913925498723984, + "val_left_hand_r2": 0.6271263957023621, + "val_right_hand_mse": 0.01178454328328371, + "val_right_hand_rmse": 0.10855663629315211, + "val_right_hand_mae": 0.08412878960371017, + "val_right_hand_r2": 0.5811853408813477, + "val_left_hip_mse": 0.0051516941748559475, + "val_left_hip_rmse": 0.07177530337696907, + "val_left_hip_mae": 0.053675465285778046, + "val_left_hip_r2": 0.41549569368362427, + "val_right_hip_mse": 0.005384501069784164, + "val_right_hip_rmse": 0.07337915964212294, + "val_right_hip_mae": 0.054696910083293915, + "val_right_hip_r2": 0.4124014973640442, + "val_left_knee_mse": 0.004303486552089453, + "val_left_knee_rmse": 0.06560096456676115, + "val_left_knee_mae": 0.04939848929643631, + "val_left_knee_r2": 0.5576720237731934, + "val_right_knee_mse": 0.004343558568507433, + "val_right_knee_rmse": 0.0659056793342382, + "val_right_knee_mae": 0.04995642974972725, + "val_right_knee_r2": 0.5949472188949585, + "val_left_foot_mse": 0.0043831681832671165, + "val_left_foot_rmse": 0.06620549964517386, + "val_left_foot_mae": 0.04912933334708214, + "val_left_foot_r2": 0.05750298500061035, + "val_right_foot_mse": 0.004962716717272997, + "val_right_foot_rmse": 0.07044655220287929, + "val_right_foot_mae": 0.05168623477220535, + "val_right_foot_r2": 0.0546647310256958 + }, + "best_epoch": 43 + }, + { + "fold": 2, + "train_metrics": { + "train_mse": 0.005019244272261858, + "train_rmse": 0.07084662498850498, + "train_mae": 0.05304316431283951, + "train_r2": 0.7632479071617126, + "train_head_mse": 0.0035323468036949635, + "train_head_rmse": 0.05943354947918695, + "train_head_mae": 0.04774189367890358, + "train_head_r2": 0.650148868560791, + "train_left_shoulder_mse": 0.0023738546296954155, + "train_left_shoulder_rmse": 0.04872221905553374, + "train_left_shoulder_mae": 0.03859010338783264, + "train_left_shoulder_r2": 0.575092077255249, + "train_left_elbow_mse": 0.004889137111604214, + "train_left_elbow_rmse": 0.06992236488852628, + "train_left_elbow_mae": 0.05539742112159729, + "train_left_elbow_r2": 0.6440385580062866, + "train_right_shoulder_mse": 0.0030617977026849985, + "train_right_shoulder_rmse": 0.05533351337738277, + "train_right_shoulder_mae": 0.0436348095536232, + "train_right_shoulder_r2": 0.5353537797927856, + "train_right_elbow_mse": 0.006201236043125391, + "train_right_elbow_rmse": 0.07874792723066043, + "train_right_elbow_mae": 0.06233873963356018, + "train_right_elbow_r2": 0.6080783605575562, + "train_left_hand_mse": 0.009487093426287174, + "train_left_hand_rmse": 0.09740171161887852, + "train_left_hand_mae": 0.07808903604745865, + "train_left_hand_r2": 0.6389198303222656, + "train_right_hand_mse": 0.010941960848867893, + "train_right_hand_rmse": 0.1046038280794154, + "train_right_hand_mae": 0.08279334753751755, + "train_right_hand_r2": 0.6102558374404907, + "train_left_hip_mse": 0.004054211545735598, + "train_left_hip_rmse": 0.06367269073736084, + "train_left_hip_mae": 0.04673575237393379, + "train_left_hip_r2": 0.4910745620727539, + "train_right_hip_mse": 0.004256293177604675, + "train_right_hip_rmse": 0.06524027266654145, + "train_right_hip_mae": 0.047607965767383575, + "train_right_hip_r2": 0.4873277544975281, + "train_left_knee_mse": 0.00390450656414032, + "train_left_knee_rmse": 0.06248605095651605, + "train_left_knee_mae": 0.04573553428053856, + "train_left_knee_r2": 0.5922742486000061, + "train_right_knee_mse": 0.0040232646279037, + "train_right_knee_rmse": 0.06342920957968576, + "train_right_knee_mae": 0.046190809458494186, + "train_right_knee_r2": 0.6224162578582764, + "train_left_foot_mse": 0.00419847946614027, + "train_left_foot_rmse": 0.06479567474870734, + "train_left_foot_mae": 0.04651668295264244, + "train_left_foot_r2": 0.06156891584396362, + "train_right_foot_mse": 0.004325994290411472, + "train_right_foot_rmse": 0.06577229120542687, + "train_right_foot_mae": 0.04818904399871826, + "train_right_foot_r2": 0.05173158645629883 + }, + "val_metrics": { + "val_mse": 0.004829944111406803, + "val_rmse": 0.06949779932779744, + "val_mae": 0.05238183215260506, + "val_r2": 0.7709462642669678, + "val_head_mse": 0.0032049177680164576, + "val_head_rmse": 0.05661199314647434, + "val_head_mae": 0.04593202844262123, + "val_head_r2": 0.6661785244941711, + "val_left_shoulder_mse": 0.0023045754060149193, + "val_left_shoulder_rmse": 0.04800599343847515, + "val_left_shoulder_mae": 0.03797100484371185, + "val_left_shoulder_r2": 0.5878905057907104, + "val_left_elbow_mse": 0.0045684087090194225, + "val_left_elbow_rmse": 0.06759000450524784, + "val_left_elbow_mae": 0.05406608432531357, + "val_left_elbow_r2": 0.6699025630950928, + "val_right_shoulder_mse": 0.002840488450601697, + "val_right_shoulder_rmse": 0.053296232986972884, + "val_right_shoulder_mae": 0.04235820099711418, + "val_right_shoulder_r2": 0.5559291839599609, + "val_right_elbow_mse": 0.005636692047119141, + "val_right_elbow_rmse": 0.07507790651795733, + "val_right_elbow_mae": 0.06010153144598007, + "val_right_elbow_r2": 0.6351587176322937, + "val_left_hand_mse": 0.008810432627797127, + "val_left_hand_rmse": 0.09386390481861026, + "val_left_hand_mae": 0.07560791075229645, + "val_left_hand_r2": 0.6642723083496094, + "val_right_hand_mse": 0.009845725260674953, + "val_right_hand_rmse": 0.09922562804374158, + "val_right_hand_mae": 0.07931085675954819, + "val_right_hand_r2": 0.6386991143226624, + "val_left_hip_mse": 0.004323530942201614, + "val_left_hip_rmse": 0.06575356220161471, + "val_left_hip_mae": 0.04832253232598305, + "val_left_hip_r2": 0.468816876411438, + "val_right_hip_mse": 0.004421530291438103, + "val_right_hip_rmse": 0.06649458843724129, + "val_right_hip_mae": 0.04907643795013428, + "val_right_hip_r2": 0.47454559803009033, + "val_left_knee_mse": 0.0036711525171995163, + "val_left_knee_rmse": 0.060590036451544704, + "val_left_knee_mae": 0.045058127492666245, + "val_left_knee_r2": 0.6082403063774109, + "val_right_knee_mse": 0.0043763164430856705, + "val_right_knee_rmse": 0.0661537334024745, + "val_right_knee_mae": 0.04719344899058342, + "val_right_knee_r2": 0.6065120100975037, + "val_left_foot_mse": 0.003988622222095728, + "val_left_foot_rmse": 0.06315553991611289, + "val_left_foot_mae": 0.047573257237672806, + "val_left_foot_r2": 0.06125497817993164, + "val_right_foot_mse": 0.004796884022653103, + "val_right_foot_rmse": 0.06925954102254146, + "val_right_foot_mae": 0.04839243367314339, + "val_right_foot_r2": 0.030752599239349365 + }, + "best_epoch": 57 + }, + { + "fold": 3, + "train_metrics": { + "train_mse": 0.004661513492465019, + "train_rmse": 0.0682752773151821, + "train_mae": 0.051018472760915756, + "train_r2": 0.777434766292572, + "train_head_mse": 0.0031450798269361258, + "train_head_rmse": 0.056081011286674615, + "train_head_mae": 0.04499109089374542, + "train_head_r2": 0.6838252544403076, + "train_left_shoulder_mse": 0.0021905647590756416, + "train_left_shoulder_rmse": 0.046803469519637556, + "train_left_shoulder_mae": 0.03687070682644844, + "train_left_shoulder_r2": 0.6081463694572449, + "train_left_elbow_mse": 0.004442969337105751, + "train_left_elbow_rmse": 0.0666556024434987, + "train_left_elbow_mae": 0.05245518684387207, + "train_left_elbow_r2": 0.673359751701355, + "train_right_shoulder_mse": 0.002796095097437501, + "train_right_shoulder_rmse": 0.05287811548682026, + "train_right_shoulder_mae": 0.04150240868330002, + "train_right_shoulder_r2": 0.5697415471076965, + "train_right_elbow_mse": 0.005720549263060093, + "train_right_elbow_rmse": 0.07563431273608621, + "train_right_elbow_mae": 0.05915677174925804, + "train_right_elbow_r2": 0.6321463584899902, + "train_left_hand_mse": 0.008597314357757568, + "train_left_hand_rmse": 0.0927217038117698, + "train_left_hand_mae": 0.07432528585195541, + "train_left_hand_r2": 0.6674405932426453, + "train_right_hand_mse": 0.01013741921633482, + "train_right_hand_rmse": 0.10068475165751177, + "train_right_hand_mae": 0.07921981066465378, + "train_right_hand_r2": 0.6325279474258423, + "train_left_hip_mse": 0.0038728206418454647, + "train_left_hip_rmse": 0.062231990502035726, + "train_left_hip_mae": 0.04492085054516792, + "train_left_hip_r2": 0.51966392993927, + "train_right_hip_mse": 0.003981973510235548, + "train_right_hip_rmse": 0.06310288036401784, + "train_right_hip_mae": 0.045246902853250504, + "train_right_hip_r2": 0.5250095129013062, + "train_left_knee_mse": 0.0035504584666341543, + "train_left_knee_rmse": 0.05958572368138323, + "train_left_knee_mae": 0.044283539056777954, + "train_left_knee_r2": 0.6319611072540283, + "train_right_knee_mse": 0.00369559065438807, + "train_right_knee_rmse": 0.060791369900571164, + "train_right_knee_mae": 0.044780831784009933, + "train_right_knee_r2": 0.6578861474990845, + "train_left_foot_mse": 0.004095116630196571, + "train_left_foot_rmse": 0.06399309830127442, + "train_left_foot_mae": 0.046886105090379715, + "train_left_foot_r2": 0.07322996854782104, + "train_right_foot_mse": 0.004373728297650814, + "train_right_foot_rmse": 0.06613416891177218, + "train_right_foot_mae": 0.048600681126117706, + "train_right_foot_r2": 0.05711030960083008 + }, + "val_metrics": { + "val_mse": 0.0045243301428854465, + "val_rmse": 0.06726314104236768, + "val_mae": 0.04985286295413971, + "val_r2": 0.7951688766479492, + "val_head_mse": 0.003060838207602501, + "val_head_rmse": 0.05532484258994779, + "val_head_mae": 0.04343084990978241, + "val_head_r2": 0.6980347633361816, + "val_left_shoulder_mse": 0.001962081529200077, + "val_left_shoulder_rmse": 0.04429538948017138, + "val_left_shoulder_mae": 0.034594252705574036, + "val_left_shoulder_r2": 0.6462737917900085, + "val_left_elbow_mse": 0.004353750962764025, + "val_left_elbow_rmse": 0.06598295963931919, + "val_left_elbow_mae": 0.05164330080151558, + "val_left_elbow_r2": 0.6962804794311523, + "val_right_shoulder_mse": 0.0027608845848590136, + "val_right_shoulder_rmse": 0.05254412036430921, + "val_right_shoulder_mae": 0.04113311320543289, + "val_right_shoulder_r2": 0.5875672101974487, + "val_right_elbow_mse": 0.005499177612364292, + "val_right_elbow_rmse": 0.07415644012737055, + "val_right_elbow_mae": 0.05891229584813118, + "val_right_elbow_r2": 0.6650580167770386, + "val_left_hand_mse": 0.00885129813104868, + "val_left_hand_rmse": 0.09408133784682635, + "val_left_hand_mae": 0.07449913769960403, + "val_left_hand_r2": 0.6822583675384521, + "val_right_hand_mse": 0.009730656631290913, + "val_right_hand_rmse": 0.0986440907063921, + "val_right_hand_mae": 0.07756434381008148, + "val_right_hand_r2": 0.665043830871582, + "val_left_hip_mse": 0.0034248430747538805, + "val_left_hip_rmse": 0.05852215883538372, + "val_left_hip_mae": 0.043704189360141754, + "val_left_hip_r2": 0.5576845407485962, + "val_right_hip_mse": 0.0036354875192046165, + "val_right_hip_rmse": 0.060295004098222076, + "val_right_hip_mae": 0.04397445544600487, + "val_right_hip_r2": 0.5503918528556824, + "val_left_knee_mse": 0.003522873856127262, + "val_left_knee_rmse": 0.05935380237295048, + "val_left_knee_mae": 0.04269041121006012, + "val_left_knee_r2": 0.6116650104522705, + "val_right_knee_mse": 0.003245327156037092, + "val_right_knee_rmse": 0.05696777296013152, + "val_right_knee_mae": 0.04117201641201973, + "val_right_knee_r2": 0.6917673349380493, + "val_left_foot_mse": 0.004388624802231789, + "val_left_foot_rmse": 0.06624669653825607, + "val_left_foot_mae": 0.0465618371963501, + "val_left_foot_r2": 0.017885982990264893, + "val_right_foot_mse": 0.004380448255687952, + "val_right_foot_rmse": 0.06618495490432816, + "val_right_foot_mae": 0.04820702597498894, + "val_right_foot_r2": 0.05573457479476929 + }, + "best_epoch": 61 + }, + { + "fold": 4, + "train_metrics": { + "train_mse": 0.004498118069022894, + "train_rmse": 0.06706801077281847, + "train_mae": 0.04993272200226784, + "train_r2": 0.7869763374328613, + "train_head_mse": 0.00296544237062335, + "train_head_rmse": 0.05445587544630377, + "train_head_mae": 0.043518174439668655, + "train_head_r2": 0.7006955146789551, + "train_left_shoulder_mse": 0.002048891270533204, + "train_left_shoulder_rmse": 0.04526468016603237, + "train_left_shoulder_mae": 0.03560788184404373, + "train_left_shoulder_r2": 0.6312171816825867, + "train_left_elbow_mse": 0.0041388217359781265, + "train_left_elbow_rmse": 0.06433367497647034, + "train_left_elbow_mae": 0.05006039887666702, + "train_left_elbow_r2": 0.6964722871780396, + "train_right_shoulder_mse": 0.0027137314900755882, + "train_right_shoulder_rmse": 0.052093487981470274, + "train_right_shoulder_mae": 0.04076620563864708, + "train_right_shoulder_r2": 0.5850533843040466, + "train_right_elbow_mse": 0.0054282862693071365, + "train_right_elbow_rmse": 0.07367690458554252, + "train_right_elbow_mae": 0.056952714920043945, + "train_right_elbow_r2": 0.6527020931243896, + "train_left_hand_mse": 0.008239386603236198, + "train_left_hand_rmse": 0.09077106699403835, + "train_left_hand_mae": 0.07196556031703949, + "train_left_hand_r2": 0.6841086149215698, + "train_right_hand_mse": 0.009507128037512302, + "train_right_hand_rmse": 0.09750450265250474, + "train_right_hand_mae": 0.0755743756890297, + "train_right_hand_r2": 0.6570142507553101, + "train_left_hip_mse": 0.003784741507843137, + "train_left_hip_rmse": 0.061520252826554093, + "train_left_hip_mae": 0.04455612599849701, + "train_left_hip_r2": 0.5355053544044495, + "train_right_hip_mse": 0.003950424958020449, + "train_right_hip_rmse": 0.06285240614344409, + "train_right_hip_mae": 0.04521644860506058, + "train_right_hip_r2": 0.5355268716812134, + "train_left_knee_mse": 0.003496406599879265, + "train_left_knee_rmse": 0.0591304202579287, + "train_left_knee_mae": 0.04464443400502205, + "train_left_knee_r2": 0.6328892707824707, + "train_right_knee_mse": 0.003604211611673236, + "train_right_knee_rmse": 0.06003508650508664, + "train_right_knee_mae": 0.045035526156425476, + "train_right_knee_r2": 0.6632816791534424, + "train_left_foot_mse": 0.004139684606343508, + "train_left_foot_rmse": 0.06434038083772514, + "train_left_foot_mae": 0.046888403594493866, + "train_left_foot_r2": 0.07479500770568848, + "train_right_foot_mse": 0.004458376672118902, + "train_right_foot_rmse": 0.0667710766134477, + "train_right_foot_mae": 0.04833918809890747, + "train_right_foot_r2": 0.06819337606430054 + }, + "val_metrics": { + "val_mse": 0.004327595233917236, + "val_rmse": 0.06578446042886751, + "val_mae": 0.04989447444677353, + "val_r2": 0.7980234622955322, + "val_head_mse": 0.0029733891133219004, + "val_head_rmse": 0.05452879159968521, + "val_head_mae": 0.04494099318981171, + "val_head_r2": 0.7128691673278809, + "val_left_shoulder_mse": 0.002064111875370145, + "val_left_shoulder_rmse": 0.04543249800935609, + "val_left_shoulder_mae": 0.03673253580927849, + "val_left_shoulder_r2": 0.6389738321304321, + "val_left_elbow_mse": 0.004366486798971891, + "val_left_elbow_rmse": 0.0660793976892336, + "val_left_elbow_mae": 0.05158422514796257, + "val_left_elbow_r2": 0.6932650804519653, + "val_right_shoulder_mse": 0.0025226348079741, + "val_right_shoulder_rmse": 0.05022583805148601, + "val_right_shoulder_mae": 0.04109163209795952, + "val_right_shoulder_r2": 0.6175146102905273, + "val_right_elbow_mse": 0.005406757351011038, + "val_right_elbow_rmse": 0.07353065585870316, + "val_right_elbow_mae": 0.05781988427042961, + "val_right_elbow_r2": 0.666649580001831, + "val_left_hand_mse": 0.008402742445468903, + "val_left_hand_rmse": 0.0916664739447793, + "val_left_hand_mae": 0.0731738731265068, + "val_left_hand_r2": 0.6889402866363525, + "val_right_hand_mse": 0.009680505841970444, + "val_right_hand_rmse": 0.0983895616514803, + "val_right_hand_mae": 0.07751485705375671, + "val_right_hand_r2": 0.6624351739883423, + "val_left_hip_mse": 0.0031526698730885983, + "val_left_hip_rmse": 0.05614864088371684, + "val_left_hip_mae": 0.04147360846400261, + "val_left_hip_r2": 0.574637770652771, + "val_right_hip_mse": 0.0031944497022777796, + "val_right_hip_rmse": 0.056519463039538684, + "val_right_hip_mae": 0.041883185505867004, + "val_right_hip_r2": 0.5797367095947266, + "val_left_knee_mse": 0.0033285561949014664, + "val_left_knee_rmse": 0.0576936408532298, + "val_left_knee_mae": 0.04376426339149475, + "val_left_knee_r2": 0.6525990962982178, + "val_right_knee_mse": 0.003496857825666666, + "val_right_knee_rmse": 0.0591342356479448, + "val_right_knee_mae": 0.04514757916331291, + "val_right_knee_r2": 0.6802155375480652, + "val_left_foot_mse": 0.0038629467599093914, + "val_left_foot_rmse": 0.06215260863318121, + "val_left_foot_mae": 0.045711204409599304, + "val_left_foot_r2": 0.0904436707496643, + "val_right_foot_mse": 0.0038066308479756117, + "val_right_foot_rmse": 0.06169789986681566, + "val_right_foot_mae": 0.04779032990336418, + "val_right_foot_r2": 0.061976730823516846 + }, + "best_epoch": 62 + }, + { + "fold": 5, + "train_metrics": { + "train_mse": 0.004818250425159931, + "train_rmse": 0.06941361844162809, + "train_mae": 0.05205652862787247, + "train_r2": 0.7757406830787659, + "train_head_mse": 0.003276217496022582, + "train_head_rmse": 0.057238252035003497, + "train_head_mae": 0.04603596031665802, + "train_head_r2": 0.6749882698059082, + "train_left_shoulder_mse": 0.0022405849304050207, + "train_left_shoulder_rmse": 0.04733481731669639, + "train_left_shoulder_mae": 0.03730132058262825, + "train_left_shoulder_r2": 0.6030678749084473, + "train_left_elbow_mse": 0.00462078582495451, + "train_left_elbow_rmse": 0.06797636225155411, + "train_left_elbow_mae": 0.05337179824709892, + "train_left_elbow_r2": 0.6698387861251831, + "train_right_shoulder_mse": 0.0028952364809811115, + "train_right_shoulder_rmse": 0.053807401730441434, + "train_right_shoulder_mae": 0.0426807701587677, + "train_right_shoulder_r2": 0.5611450672149658, + "train_right_elbow_mse": 0.005849997512996197, + "train_right_elbow_rmse": 0.07648527644583758, + "train_right_elbow_mae": 0.06008418649435043, + "train_right_elbow_r2": 0.6332497596740723, + "train_left_hand_mse": 0.009061319753527641, + "train_left_hand_rmse": 0.09519096466328955, + "train_left_hand_mae": 0.07622408121824265, + "train_left_hand_r2": 0.6623960733413696, + "train_right_hand_mse": 0.01037136372178793, + "train_right_hand_rmse": 0.10183989258531222, + "train_right_hand_mae": 0.07984938472509384, + "train_right_hand_r2": 0.6336079835891724, + "train_left_hip_mse": 0.0039002550765872, + "train_left_hip_rmse": 0.06245202219774153, + "train_left_hip_mae": 0.0460863932967186, + "train_left_hip_r2": 0.5142561197280884, + "train_right_hip_mse": 0.004060142207890749, + "train_right_hip_rmse": 0.06371924519241223, + "train_right_hip_mae": 0.04688272997736931, + "train_right_hip_r2": 0.5118651986122131, + "train_left_knee_mse": 0.003779809921979904, + "train_left_knee_rmse": 0.06148015876671029, + "train_left_knee_mae": 0.0456986203789711, + "train_left_knee_r2": 0.5996469259262085, + "train_right_knee_mse": 0.003919999580830336, + "train_right_knee_rmse": 0.06260990002252308, + "train_right_knee_mae": 0.0461774617433548, + "train_right_knee_r2": 0.6380417346954346, + "train_left_foot_mse": 0.004169650841504335, + "train_left_foot_rmse": 0.06457283361835947, + "train_left_foot_mae": 0.04731089994311333, + "train_left_foot_r2": 0.05333244800567627, + "train_right_foot_mse": 0.004491885658353567, + "train_right_foot_rmse": 0.06702153130415305, + "train_right_foot_mae": 0.04903122037649155, + "train_right_foot_r2": 0.04965484142303467 + }, + "val_metrics": { + "val_mse": 0.005283033940941095, + "val_rmse": 0.07268448211923295, + "val_mae": 0.05326644331216812, + "val_r2": 0.7348763942718506, + "val_head_mse": 0.0037569627165794373, + "val_head_rmse": 0.06129406754800531, + "val_head_mae": 0.048142798244953156, + "val_head_r2": 0.6090805530548096, + "val_left_shoulder_mse": 0.0023962059058248997, + "val_left_shoulder_rmse": 0.048951056227878265, + "val_left_shoulder_mae": 0.03824884817004204, + "val_left_shoulder_r2": 0.5502468347549438, + "val_left_elbow_mse": 0.0053450195118784904, + "val_left_elbow_rmse": 0.07310964034844167, + "val_left_elbow_mae": 0.0563567578792572, + "val_left_elbow_r2": 0.5804475545883179, + "val_right_shoulder_mse": 0.003172562923282385, + "val_right_shoulder_rmse": 0.05632550863758254, + "val_right_shoulder_mae": 0.04332299530506134, + "val_right_shoulder_r2": 0.49824851751327515, + "val_right_elbow_mse": 0.006764494813978672, + "val_right_elbow_rmse": 0.08224654894874722, + "val_right_elbow_mae": 0.06254716962575912, + "val_right_elbow_r2": 0.5444064140319824, + "val_left_hand_mse": 0.010521323420107365, + "val_left_hand_rmse": 0.10257350252432333, + "val_left_hand_mae": 0.07987884432077408, + "val_left_hand_r2": 0.5600852966308594, + "val_right_hand_mse": 0.012033358216285706, + "val_right_hand_rmse": 0.10969666456317487, + "val_right_hand_mae": 0.08379530161619186, + "val_right_hand_r2": 0.5410287976264954, + "val_left_hip_mse": 0.0038902428932487965, + "val_left_hip_rmse": 0.062371811688043795, + "val_left_hip_mae": 0.04618913680315018, + "val_left_hip_r2": 0.5051819086074829, + "val_right_hip_mse": 0.004126366227865219, + "val_right_hip_rmse": 0.06423679808229252, + "val_right_hip_mae": 0.04730694368481636, + "val_right_hip_r2": 0.5045069456100464, + "val_left_knee_mse": 0.004230352584272623, + "val_left_knee_rmse": 0.06504116069284606, + "val_left_knee_mae": 0.04817717522382736, + "val_left_knee_r2": 0.5720006227493286, + "val_right_knee_mse": 0.004187567159533501, + "val_right_knee_rmse": 0.0647114144454709, + "val_right_knee_mae": 0.047237932682037354, + "val_right_knee_r2": 0.5970410108566284, + "val_left_foot_mse": 0.004178274888545275, + "val_left_foot_rmse": 0.06463957679738687, + "val_left_foot_mae": 0.0454060435295105, + "val_left_foot_r2": 0.07689183950424194, + "val_right_foot_mse": 0.004076709970831871, + "val_right_foot_rmse": 0.06384911879448198, + "val_right_foot_mae": 0.04585381597280502, + "val_right_foot_r2": 0.04958820343017578 + }, + "best_epoch": 50 + } + ] +} \ No newline at end of file diff --git a/A9/cv_results_different_models/lstm_fold1_best.h5 b/A9/cv_results_different_models/lstm_fold1_best.h5 new file mode 100644 index 0000000000000000000000000000000000000000..bae0098710b6c5945609059726598c5df9977134 --- /dev/null +++ b/A9/cv_results_different_models/lstm_fold1_best.h5 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:29850b036538fae1c47d1f0f983d277f69711965ed1c6d8ec7c2eb329738a13a +size 487544 diff --git a/A9/cv_results_different_models/lstm_fold2_best.h5 b/A9/cv_results_different_models/lstm_fold2_best.h5 new file mode 100644 index 0000000000000000000000000000000000000000..820ef3a318aa9b7edf9f42c4da8589e83b6f68bf --- /dev/null +++ b/A9/cv_results_different_models/lstm_fold2_best.h5 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:0648436461861e0632cf932a4dde1849697c96855464b455c6ca004eab9971cf +size 487544 diff --git a/A9/cv_results_different_models/lstm_fold3_best.h5 b/A9/cv_results_different_models/lstm_fold3_best.h5 new file mode 100644 index 0000000000000000000000000000000000000000..6d93e8f4814848cdb4e21e68c6d1ebfd3ae3c981 --- /dev/null +++ b/A9/cv_results_different_models/lstm_fold3_best.h5 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:ba70e1eebd04da0ad3515ff8c3e3003ab4af084bb707cd8f88f259a1491e1a2e +size 487544 diff --git a/A9/cv_results_different_models/lstm_fold4_best.h5 b/A9/cv_results_different_models/lstm_fold4_best.h5 new file mode 100644 index 0000000000000000000000000000000000000000..587fed3a095878344d0109094c165456dc52e5d6 --- /dev/null +++ b/A9/cv_results_different_models/lstm_fold4_best.h5 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:8c6085d79e36d02ece908a9daa34407d1d2862821d9f3cbd3868e9093f902883 +size 487544 diff --git a/A9/cv_results_different_models/lstm_fold5_best.h5 b/A9/cv_results_different_models/lstm_fold5_best.h5 new file mode 100644 index 0000000000000000000000000000000000000000..dabce59f276bcbade05cb811376e1ade1652fd82 --- /dev/null +++ b/A9/cv_results_different_models/lstm_fold5_best.h5 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:9f4175aa67de8d794d1090f2edbb19c58e53cc6277eb51aa2a6d81352e5b9c7c +size 487544 diff --git a/A9/cv_results_different_models/lstm_predictions.png b/A9/cv_results_different_models/lstm_predictions.png new file mode 100644 index 0000000000000000000000000000000000000000..4fda9a72733a7a3a4cb16a15daf2965143856c0b --- /dev/null +++ b/A9/cv_results_different_models/lstm_predictions.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:71e6ef1552b6abcf1a212d21b73d2b90ac547ccd5786f28c6d758f53640386b3 +size 728524 diff --git a/A9/cv_results_different_models/lstm_training_history.png b/A9/cv_results_different_models/lstm_training_history.png new file mode 100644 index 0000000000000000000000000000000000000000..98516bd34079db0be80d31d66f9ee6190eb8a260 --- /dev/null +++ b/A9/cv_results_different_models/lstm_training_history.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:860753478d70ebaea286b19464e0b89d7edc0157f50dea5d032fb4304c7cc2c0 +size 403395 diff --git a/A9/cv_results_different_models/summary.txt b/A9/cv_results_different_models/summary.txt new file mode 100644 index 0000000000000000000000000000000000000000..6470476ef31a4a2faf1b09ab15335e7f0e20513a --- /dev/null +++ b/A9/cv_results_different_models/summary.txt @@ -0,0 +1,274 @@ +Cross-Validation Training Summary +================================================== + +DENSE: + Best Fold: 4 + Val RMSE: 0.058055 ± 0.000435 + Val MAE: 0.042731 ± 0.000244 + Val R²: 0.820046 ± 0.002133 + +CONV1D: + Best Fold: 2 + Val RMSE: 0.037704 ± 0.001386 + Val MAE: 0.027598 ± 0.000905 + Val R²: 0.932681 ± 0.005506 + +LSTM: + Best Fold: 4 + Val RMSE: 0.069834 ± 0.003102 + Val MAE: 0.052217 ± 0.002199 + Val R²: 0.768530 ± 0.025836 + +GRU: + Best Fold: 3 + Val RMSE: 0.073991 ± 0.001842 + Val MAE: 0.055780 ± 0.001856 + Val R²: 0.740877 ± 0.016279 + +DENSE +------------------------------ +Best Fold: 4 + +Validation Metrics (mean ± std): + val_mse: 0.003371 ± 0.000051 + val_rmse: 0.058055 ± 0.000435 + val_mae: 0.042731 ± 0.000244 + val_r2: 0.820046 ± 0.002133 + val_head_mse: 0.002308 ± 0.000035 + val_head_rmse: 0.048041 ± 0.000363 + val_head_mae: 0.036726 ± 0.000322 + val_head_r2: 0.752433 ± 0.006411 + val_left_shoulder_mse: 0.001683 ± 0.000028 + val_left_shoulder_rmse: 0.041020 ± 0.000340 + val_left_shoulder_mae: 0.031314 ± 0.000136 + val_left_shoulder_r2: 0.686501 ± 0.006399 + val_left_elbow_mse: 0.002533 ± 0.000043 + val_left_elbow_rmse: 0.050332 ± 0.000433 + val_left_elbow_mae: 0.038873 ± 0.000499 + val_left_elbow_r2: 0.811899 ± 0.004719 + val_right_shoulder_mse: 0.002093 ± 0.000037 + val_right_shoulder_rmse: 0.045751 ± 0.000400 + val_right_shoulder_mae: 0.034925 ± 0.000314 + val_right_shoulder_r2: 0.662700 ± 0.006796 + val_right_elbow_mse: 0.003149 ± 0.000102 + val_right_elbow_rmse: 0.056106 ± 0.000902 + val_right_elbow_mae: 0.043767 ± 0.000702 + val_right_elbow_r2: 0.793995 ± 0.005160 + val_left_hand_mse: 0.004624 ± 0.000114 + val_left_hand_rmse: 0.067998 ± 0.000843 + val_left_hand_mae: 0.053161 ± 0.000873 + val_left_hand_r2: 0.823284 ± 0.003937 + val_right_hand_mse: 0.005181 ± 0.000238 + val_right_hand_rmse: 0.071960 ± 0.001647 + val_right_hand_mae: 0.055472 ± 0.001106 + val_right_hand_r2: 0.809632 ± 0.005755 + val_left_hip_mse: 0.003549 ± 0.000144 + val_left_hip_rmse: 0.059560 ± 0.001218 + val_left_hip_mae: 0.042415 ± 0.000440 + val_left_hip_r2: 0.524888 ± 0.007323 + val_right_hip_mse: 0.003699 ± 0.000141 + val_right_hip_rmse: 0.060805 ± 0.001167 + val_right_hip_mae: 0.042944 ± 0.000368 + val_right_hip_r2: 0.523995 ± 0.006999 + val_left_knee_mse: 0.003281 ± 0.000054 + val_left_knee_rmse: 0.057274 ± 0.000475 + val_left_knee_mae: 0.041526 ± 0.000266 + val_left_knee_r2: 0.653571 ± 0.004043 + val_right_knee_mse: 0.003561 ± 0.000054 + val_right_knee_rmse: 0.059673 ± 0.000457 + val_right_knee_mae: 0.043351 ± 0.000114 + val_right_knee_r2: 0.667071 ± 0.002340 + val_left_foot_mse: 0.003863 ± 0.000073 + val_left_foot_rmse: 0.062149 ± 0.000587 + val_left_foot_mae: 0.043861 ± 0.000414 + val_left_foot_r2: 0.113498 ± 0.005055 + val_right_foot_mse: 0.004294 ± 0.000189 + val_right_foot_rmse: 0.065510 ± 0.001423 + val_right_foot_mae: 0.047171 ± 0.000739 + val_right_foot_r2: 0.062916 ± 0.005533 + +CONV1D +------------------------------ +Best Fold: 2 + +Validation Metrics (mean ± std): + val_mse: 0.001424 ± 0.000105 + val_rmse: 0.037704 ± 0.001386 + val_mae: 0.027598 ± 0.000905 + val_r2: 0.932681 ± 0.005506 + val_head_mse: 0.001379 ± 0.000079 + val_head_rmse: 0.037122 ± 0.001048 + val_head_mae: 0.027979 ± 0.000676 + val_head_r2: 0.861529 ± 0.010834 + val_left_shoulder_mse: 0.000857 ± 0.000045 + val_left_shoulder_rmse: 0.029272 ± 0.000766 + val_left_shoulder_mae: 0.022188 ± 0.000318 + val_left_shoulder_r2: 0.846117 ± 0.009572 + val_left_elbow_mse: 0.001437 ± 0.000061 + val_left_elbow_rmse: 0.037902 ± 0.000820 + val_left_elbow_mae: 0.029248 ± 0.000801 + val_left_elbow_r2: 0.895202 ± 0.006488 + val_right_shoulder_mse: 0.000854 ± 0.000050 + val_right_shoulder_rmse: 0.029215 ± 0.000836 + val_right_shoulder_mae: 0.022053 ± 0.000665 + val_right_shoulder_r2: 0.869218 ± 0.007467 + val_right_elbow_mse: 0.001476 ± 0.000201 + val_right_elbow_rmse: 0.038338 ± 0.002502 + val_right_elbow_mae: 0.029134 ± 0.001471 + val_right_elbow_r2: 0.905963 ± 0.013118 + val_left_hand_mse: 0.002437 ± 0.000200 + val_left_hand_rmse: 0.049329 ± 0.002031 + val_left_hand_mae: 0.038094 ± 0.001428 + val_left_hand_r2: 0.906929 ± 0.008496 + val_right_hand_mse: 0.002583 ± 0.000285 + val_right_hand_rmse: 0.050748 ± 0.002751 + val_right_hand_mae: 0.038414 ± 0.002125 + val_right_hand_r2: 0.907321 ± 0.009518 + val_left_hip_mse: 0.001121 ± 0.000102 + val_left_hip_rmse: 0.033447 ± 0.001520 + val_left_hip_mae: 0.025383 ± 0.001205 + val_left_hip_r2: 0.859985 ± 0.005885 + val_right_hip_mse: 0.001135 ± 0.000095 + val_right_hip_rmse: 0.033656 ± 0.001396 + val_right_hip_mae: 0.025120 ± 0.001117 + val_right_hip_r2: 0.863767 ± 0.003510 + val_left_knee_mse: 0.001167 ± 0.000125 + val_left_knee_rmse: 0.034116 ± 0.001831 + val_left_knee_mae: 0.025196 ± 0.001110 + val_left_knee_r2: 0.877695 ± 0.010287 + val_right_knee_mse: 0.001187 ± 0.000078 + val_right_knee_rmse: 0.034428 ± 0.001139 + val_right_knee_mae: 0.025879 ± 0.001005 + val_right_knee_r2: 0.889287 ± 0.009750 + val_left_foot_mse: 0.001416 ± 0.000238 + val_left_foot_rmse: 0.037493 ± 0.003229 + val_left_foot_mae: 0.024317 ± 0.001835 + val_left_foot_r2: 0.681305 ± 0.045592 + val_right_foot_mse: 0.001455 ± 0.000267 + val_right_foot_rmse: 0.037982 ± 0.003561 + val_right_foot_mae: 0.025772 ± 0.001390 + val_right_foot_r2: 0.687791 ± 0.039643 + +LSTM +------------------------------ +Best Fold: 4 + +Validation Metrics (mean ± std): + val_mse: 0.004886 ± 0.000434 + val_rmse: 0.069834 ± 0.003102 + val_mae: 0.052217 ± 0.002199 + val_r2: 0.768530 ± 0.025836 + val_head_mse: 0.003308 ± 0.000297 + val_head_rmse: 0.057461 ± 0.002566 + val_head_mae: 0.046200 ± 0.001930 + val_head_r2: 0.667629 ± 0.036458 + val_left_shoulder_mse: 0.002257 ± 0.000217 + val_left_shoulder_rmse: 0.047451 ± 0.002294 + val_left_shoulder_mae: 0.037482 ± 0.001754 + val_left_shoulder_r2: 0.594955 ± 0.041274 + val_left_elbow_mse: 0.004726 ± 0.000387 + val_left_elbow_rmse: 0.068689 ± 0.002788 + val_left_elbow_mae: 0.053846 ± 0.001966 + val_left_elbow_r2: 0.654235 ± 0.043608 + val_right_shoulder_mse: 0.002916 ± 0.000277 + val_right_shoulder_rmse: 0.053937 ± 0.002573 + val_right_shoulder_mae: 0.042628 ± 0.001545 + val_right_shoulder_r2: 0.553235 ± 0.045797 + val_right_elbow_mse: 0.005984 ± 0.000582 + val_right_elbow_rmse: 0.077265 ± 0.003731 + val_right_elbow_mae: 0.060461 ± 0.001997 + val_right_elbow_r2: 0.617672 ± 0.048918 + val_left_hand_mse: 0.009270 ± 0.000768 + val_left_hand_rmse: 0.096201 ± 0.003949 + val_left_hand_mae: 0.076460 ± 0.002617 + val_left_hand_r2: 0.644537 ± 0.047369 + val_right_hand_mse: 0.010615 ± 0.001061 + val_right_hand_rmse: 0.102903 ± 0.005102 + val_right_hand_mae: 0.080463 ± 0.002931 + val_right_hand_r2: 0.617678 ± 0.048776 + val_left_hip_mse: 0.003989 ± 0.000706 + val_left_hip_rmse: 0.062914 ± 0.005512 + val_left_hip_mae: 0.046673 ± 0.004191 + val_left_hip_r2: 0.504363 ± 0.058212 + val_right_hip_mse: 0.004152 ± 0.000745 + val_right_hip_rmse: 0.064185 ± 0.005723 + val_right_hip_mae: 0.047388 ± 0.004432 + val_right_hip_r2: 0.504317 ± 0.058559 + val_left_knee_mse: 0.003811 ± 0.000388 + val_left_knee_rmse: 0.061656 ± 0.003136 + val_left_knee_mae: 0.045818 ± 0.002568 + val_left_knee_r2: 0.600435 ± 0.033303 + val_right_knee_mse: 0.003930 ± 0.000468 + val_right_knee_rmse: 0.062575 ± 0.003788 + val_right_knee_mae: 0.046141 ± 0.002917 + val_right_knee_r2: 0.634097 ± 0.042707 + val_left_foot_mse: 0.004160 ± 0.000210 + val_left_foot_rmse: 0.064480 ± 0.001631 + val_left_foot_mae: 0.046876 ± 0.001355 + val_left_foot_r2: 0.060796 ± 0.024456 + val_right_foot_mse: 0.004405 ± 0.000432 + val_right_foot_rmse: 0.066288 ± 0.003260 + val_right_foot_mae: 0.048386 ± 0.001881 + val_right_foot_r2: 0.050543 ± 0.010651 + +GRU +------------------------------ +Best Fold: 3 + +Validation Metrics (mean ± std): + val_mse: 0.005478 ± 0.000271 + val_rmse: 0.073991 ± 0.001842 + val_mae: 0.055780 ± 0.001856 + val_r2: 0.740877 ± 0.016279 + val_head_mse: 0.003687 ± 0.000268 + val_head_rmse: 0.060682 ± 0.002198 + val_head_mae: 0.049093 ± 0.002504 + val_head_r2: 0.630629 ± 0.021436 + val_left_shoulder_mse: 0.002580 ± 0.000223 + val_left_shoulder_rmse: 0.050749 ± 0.002218 + val_left_shoulder_mae: 0.040724 ± 0.002284 + val_left_shoulder_r2: 0.537752 ± 0.032385 + val_left_elbow_mse: 0.005640 ± 0.000433 + val_left_elbow_rmse: 0.075045 ± 0.002862 + val_left_elbow_mae: 0.059190 ± 0.002599 + val_left_elbow_r2: 0.588959 ± 0.032824 + val_right_shoulder_mse: 0.003288 ± 0.000176 + val_right_shoulder_rmse: 0.057325 ± 0.001518 + val_right_shoulder_mae: 0.046021 ± 0.001853 + val_right_shoulder_r2: 0.496626 ± 0.024583 + val_right_elbow_mse: 0.006872 ± 0.000443 + val_right_elbow_rmse: 0.082853 ± 0.002683 + val_right_elbow_mae: 0.065256 ± 0.002408 + val_right_elbow_r2: 0.562076 ± 0.032161 + val_left_hand_mse: 0.010726 ± 0.000669 + val_left_hand_rmse: 0.103516 ± 0.003184 + val_left_hand_mae: 0.083064 ± 0.003190 + val_left_hand_r2: 0.590313 ± 0.031375 + val_right_hand_mse: 0.012054 ± 0.000938 + val_right_hand_rmse: 0.109708 ± 0.004273 + val_right_hand_mae: 0.086502 ± 0.003986 + val_right_hand_r2: 0.566937 ± 0.035318 + val_left_hip_mse: 0.004296 ± 0.000539 + val_left_hip_rmse: 0.065413 ± 0.004078 + val_left_hip_mae: 0.049464 ± 0.002965 + val_left_hip_r2: 0.464392 ± 0.040291 + val_right_hip_mse: 0.004535 ± 0.000555 + val_right_hip_rmse: 0.067215 ± 0.004100 + val_right_hip_mae: 0.050770 ± 0.003056 + val_right_hip_r2: 0.456534 ± 0.038990 + val_left_knee_mse: 0.004341 ± 0.000278 + val_left_knee_rmse: 0.065851 ± 0.002121 + val_left_knee_mae: 0.049283 ± 0.002339 + val_left_knee_r2: 0.544421 ± 0.025439 + val_right_knee_mse: 0.004521 ± 0.000480 + val_right_knee_rmse: 0.067148 ± 0.003536 + val_right_knee_mae: 0.050303 ± 0.002978 + val_right_knee_r2: 0.579464 ± 0.038503 + val_left_foot_mse: 0.004219 ± 0.000177 + val_left_foot_rmse: 0.064941 ± 0.001372 + val_left_foot_mae: 0.047126 ± 0.001422 + val_left_foot_r2: 0.047324 ± 0.016250 + val_right_foot_mse: 0.004456 ± 0.000424 + val_right_foot_rmse: 0.066674 ± 0.003185 + val_right_foot_mae: 0.048339 ± 0.001829 + val_right_foot_r2: 0.039346 ± 0.006819 diff --git a/A9/cv_results_loss_optimizer/conv1d_v3_adam_mae_cv_results.json b/A9/cv_results_loss_optimizer/conv1d_v3_adam_mae_cv_results.json new file mode 100644 index 0000000000000000000000000000000000000000..1a86567b2521d29b1283d4f80cc2d94c05f1e11e --- /dev/null +++ b/A9/cv_results_loss_optimizer/conv1d_v3_adam_mae_cv_results.json @@ -0,0 +1,1243 @@ +{ + "model_name": "conv1d_v3_adam_mae", + "n_splits": 5, + "epochs": 100, + "batch_size": 32, + "patience": 10, + "aggregate_results": { + "n_folds": 5, + "train_metrics": { + "train_mse": { + "mean": 0.0009795495541766285, + "std": 6.903378643329666e-05 + }, + "train_rmse": { + "mean": 0.031278718487376886, + "std": 0.0010914778806991648 + }, + "train_mae": { + "mean": 0.02144474908709526, + "std": 0.0010089178682808834 + }, + "train_r2": { + "mean": 0.9536787390708923, + "std": 0.003024775451351158 + }, + "train_head_mse": { + "mean": 0.0008719879202544689, + "std": 5.196556035458068e-05 + }, + "train_head_rmse": { + "mean": 0.02951611519451796, + "std": 0.0008870535938796401 + }, + "train_head_mae": { + "mean": 0.021689416095614433, + "std": 0.0003945047822960699 + }, + "train_head_r2": { + "mean": 0.9123600006103516, + "std": 0.005049402494098235 + }, + "train_left_shoulder_mse": { + "mean": 0.0005542863160371781, + "std": 6.1003378512109384e-05 + }, + "train_left_shoulder_rmse": { + "mean": 0.02350868848302187, + "std": 0.0012758847304568286 + }, + "train_left_shoulder_mae": { + "mean": 0.0172866303473711, + "std": 0.0008353776644682164 + }, + "train_left_shoulder_r2": { + "mean": 0.900395131111145, + "std": 0.009876068477306998 + }, + "train_left_elbow_mse": { + "mean": 0.000897819281090051, + "std": 8.053466407679178e-05 + }, + "train_left_elbow_rmse": { + "mean": 0.02993456309106519, + "std": 0.001319550543589941 + }, + "train_left_elbow_mae": { + "mean": 0.02234143055975437, + "std": 0.001016374779984171 + }, + "train_left_elbow_r2": { + "mean": 0.93497793674469, + "std": 0.005771621599482448 + }, + "train_right_shoulder_mse": { + "mean": 0.0005198540107812733, + "std": 7.334051359755378e-05 + }, + "train_right_shoulder_rmse": { + "mean": 0.022746805828804884, + "std": 0.0015610366324725789 + }, + "train_right_shoulder_mae": { + "mean": 0.01680699847638607, + "std": 0.001157980956513988 + }, + "train_right_shoulder_r2": { + "mean": 0.9203736543655395, + "std": 0.010701310440897657 + }, + "train_right_elbow_mse": { + "mean": 0.0010081957909278572, + "std": 9.464714525369803e-05 + }, + "train_right_elbow_rmse": { + "mean": 0.031718394478393155, + "std": 0.001462615000912964 + }, + "train_right_elbow_mae": { + "mean": 0.022643600404262543, + "std": 0.001518941539278204 + }, + "train_right_elbow_r2": { + "mean": 0.9361241698265076, + "std": 0.00553721605440987 + }, + "train_left_hand_mse": { + "mean": 0.0014447469962760805, + "std": 0.0001683242618756633 + }, + "train_left_hand_rmse": { + "mean": 0.03794804256970115, + "std": 0.0021663474800295806 + }, + "train_left_hand_mae": { + "mean": 0.028974155336618422, + "std": 0.0016133951869642068 + }, + "train_left_hand_r2": { + "mean": 0.9450257539749145, + "std": 0.006556289191658132 + }, + "train_right_hand_mse": { + "mean": 0.0019198063062503933, + "std": 0.00010870125348876824 + }, + "train_right_hand_rmse": { + "mean": 0.04379846719756828, + "std": 0.0012249805687962923 + }, + "train_right_hand_mae": { + "mean": 0.030997533723711966, + "std": 0.00175020387682319 + }, + "train_right_hand_r2": { + "mean": 0.9310840606689453, + "std": 0.003567275045671363 + }, + "train_left_hip_mse": { + "mean": 0.0007112229708582163, + "std": 6.9657037009084e-05 + }, + "train_left_hip_rmse": { + "mean": 0.026638022817103257, + "std": 0.0012801215777035896 + }, + "train_left_hip_mae": { + "mean": 0.019725799188017847, + "std": 0.0010150432278669987 + }, + "train_left_hip_r2": { + "mean": 0.9088081002235413, + "std": 0.008423994097630552 + }, + "train_right_hip_mse": { + "mean": 0.0007044269470497966, + "std": 7.93924372190348e-05 + }, + "train_right_hip_rmse": { + "mean": 0.02650025318744185, + "std": 0.0014708936233713424 + }, + "train_right_hip_mae": { + "mean": 0.019503872469067575, + "std": 0.0009727329132850842 + }, + "train_right_hip_r2": { + "mean": 0.9131611585617065, + "std": 0.00984657721769651 + }, + "train_left_knee_mse": { + "mean": 0.0008385802619159221, + "std": 8.762654506989687e-05 + }, + "train_left_knee_rmse": { + "mean": 0.02891896605870698, + "std": 0.0015078673719117917 + }, + "train_left_knee_mae": { + "mean": 0.01983027495443821, + "std": 0.0009664489315703164 + }, + "train_left_knee_r2": { + "mean": 0.9115729331970215, + "std": 0.009437715477968539 + }, + "train_right_knee_mse": { + "mean": 0.0008977711200714112, + "std": 8.172278752011631e-05 + }, + "train_right_knee_rmse": { + "mean": 0.029929993959104312, + "std": 0.0014023486297603373 + }, + "train_right_knee_mae": { + "mean": 0.01984255723655224, + "std": 0.0009396402582316761 + }, + "train_right_knee_r2": { + "mean": 0.9165327072143554, + "std": 0.007581274123050004 + }, + "train_left_foot_mse": { + "mean": 0.0011755786137655378, + "std": 9.30680488687264e-05 + }, + "train_left_foot_rmse": { + "mean": 0.034258771066470545, + "std": 0.0013839142967318283 + }, + "train_left_foot_mae": { + "mean": 0.019504211843013763, + "std": 0.0005895322414672522 + }, + "train_left_foot_r2": { + "mean": 0.731277585029602, + "std": 0.018579969879145796 + }, + "train_right_foot_mse": { + "mean": 0.001189867639914155, + "std": 8.343761992774535e-05 + }, + "train_right_foot_rmse": { + "mean": 0.03447259903237353, + "std": 0.001227825666515661 + }, + "train_right_foot_mae": { + "mean": 0.01963524967432022, + "std": 0.001195289082528765 + }, + "train_right_foot_r2": { + "mean": 0.7345233917236328, + "std": 0.01972943520301782 + } + }, + "val_metrics": { + "val_mse": { + "mean": 0.0012682204134762288, + "std": 6.44337256465927e-05 + }, + "val_rmse": { + "mean": 0.03560098284681343, + "std": 0.0008890634494369338 + }, + "val_mae": { + "mean": 0.024090223386883734, + "std": 0.0005202208397450156 + }, + "val_r2": { + "mean": 0.9399824142456055, + "std": 0.0025257740290760295 + }, + "val_head_mse": { + "mean": 0.0010863309958949684, + "std": 0.00010637004504412988 + }, + "val_head_rmse": { + "mean": 0.032922085676166224, + "std": 0.0015707547950394281 + }, + "val_head_mae": { + "mean": 0.024147843196988105, + "std": 0.0008009967237753239 + }, + "val_head_r2": { + "mean": 0.8907873630523682, + "std": 0.007882024252117743 + }, + "val_left_shoulder_mse": { + "mean": 0.0007004949380643666, + "std": 4.475260491353417e-05 + }, + "val_left_shoulder_rmse": { + "mean": 0.02645347177786908, + "std": 0.0008418841736548858 + }, + "val_left_shoulder_mae": { + "mean": 0.01939507611095905, + "std": 0.0003270353738252629 + }, + "val_left_shoulder_r2": { + "mean": 0.873916232585907, + "std": 0.003247511413455061 + }, + "val_left_elbow_mse": { + "mean": 0.0011581871891394257, + "std": 6.325369801352384e-05 + }, + "val_left_elbow_rmse": { + "mean": 0.03401959946129547, + "std": 0.0009241437293251133 + }, + "val_left_elbow_mae": { + "mean": 0.025217337906360625, + "std": 7.774352698583276e-05 + }, + "val_left_elbow_r2": { + "mean": 0.9159456133842468, + "std": 0.005490444779497361 + }, + "val_right_shoulder_mse": { + "mean": 0.0006856702384538949, + "std": 8.542469841736924e-05 + }, + "val_right_shoulder_rmse": { + "mean": 0.02613577625820133, + "std": 0.0016097943456027165 + }, + "val_right_shoulder_mae": { + "mean": 0.019131946191191674, + "std": 0.000863904979342797 + }, + "val_right_shoulder_r2": { + "mean": 0.8946998596191407, + "std": 0.012625397954558056 + }, + "val_right_elbow_mse": { + "mean": 0.0013539781561121345, + "std": 0.0001153007404454513 + }, + "val_right_elbow_rmse": { + "mean": 0.03676274167916847, + "std": 0.0015744778000542816 + }, + "val_right_elbow_mae": { + "mean": 0.026058284938335417, + "std": 0.001299665685813914 + }, + "val_right_elbow_r2": { + "mean": 0.9138714790344238, + "std": 0.008335446113133346 + }, + "val_left_hand_mse": { + "mean": 0.0019447145285084843, + "std": 0.00023608513246016748 + }, + "val_left_hand_rmse": { + "mean": 0.04402108528943727, + "std": 0.0026188887812531934 + }, + "val_left_hand_mae": { + "mean": 0.03326229378581047, + "std": 0.0010291454959134947 + }, + "val_left_hand_r2": { + "mean": 0.9257623434066773, + "std": 0.010007308197493589 + }, + "val_right_hand_mse": { + "mean": 0.0025529030710458754, + "std": 0.0002601374171787325 + }, + "val_right_hand_rmse": { + "mean": 0.050462237500943696, + "std": 0.0025427657076946713 + }, + "val_right_hand_mae": { + "mean": 0.03561046496033669, + "std": 0.0012382269086748578 + }, + "val_right_hand_r2": { + "mean": 0.9080064773559571, + "std": 0.010368732802588492 + }, + "val_left_hip_mse": { + "mean": 0.0008940620231442153, + "std": 0.0001133453286664282 + }, + "val_left_hip_rmse": { + "mean": 0.029841257306646423, + "std": 0.0018871633481860098 + }, + "val_left_hip_mae": { + "mean": 0.021671008318662643, + "std": 0.000832922273005169 + }, + "val_left_hip_r2": { + "mean": 0.8856171011924744, + "std": 0.008022667343321372 + }, + "val_right_hip_mse": { + "mean": 0.0009040305158123374, + "std": 0.00014878872523135046 + }, + "val_right_hip_rmse": { + "mean": 0.029972528082960852, + "std": 0.002382871404096391 + }, + "val_right_hip_mae": { + "mean": 0.021577286720275878, + "std": 0.0010212858661694032 + }, + "val_right_hip_r2": { + "mean": 0.8891724824905396, + "std": 0.009689909714866045 + }, + "val_left_knee_mse": { + "mean": 0.0010744167608208953, + "std": 0.00016599135022702467 + }, + "val_left_knee_rmse": { + "mean": 0.03268340947346925, + "std": 0.0024922893111426173 + }, + "val_left_knee_mae": { + "mean": 0.02207806333899498, + "std": 0.0009310872405752417 + }, + "val_left_knee_r2": { + "mean": 0.886946702003479, + "std": 0.013709637775234108 + }, + "val_right_knee_mse": { + "mean": 0.0011682481970638036, + "std": 0.0003137764537681749 + }, + "val_right_knee_rmse": { + "mean": 0.033880939140249436, + "std": 0.00450889787403945 + }, + "val_right_knee_mae": { + "mean": 0.022206682339310647, + "std": 0.0015918373188222377 + }, + "val_right_knee_r2": { + "mean": 0.8922719955444336, + "std": 0.024400300693713555 + }, + "val_left_foot_mse": { + "mean": 0.001454160234425217, + "std": 0.0003811823046504875 + }, + "val_left_foot_rmse": { + "mean": 0.037750343658941504, + "std": 0.005391826041058093 + }, + "val_left_foot_mae": { + "mean": 0.021218356862664222, + "std": 0.0018519924278663222 + }, + "val_left_foot_r2": { + "mean": 0.6734158873558045, + "std": 0.053395745350954135 + }, + "val_right_foot_mse": { + "mean": 0.001509668305516243, + "std": 0.00026329977097187704 + }, + "val_right_foot_rmse": { + "mean": 0.038712506338551604, + "std": 0.003318155889020989 + }, + "val_right_foot_mae": { + "mean": 0.02159826159477234, + "std": 0.0010848730905192487 + }, + "val_right_foot_r2": { + "mean": 0.6643227577209473, + "std": 0.04629943432772424 + } + }, + "best_fold": "1", + "best_fold_metrics": { + "fold": 1, + "train_metrics": { + "train_mse": 0.0011018142104148865, + "train_rmse": 0.033193586886850396, + "train_mae": 0.023304687812924385, + "train_r2": 0.9482403993606567, + "train_head_mse": 0.0009279437945224345, + "train_head_rmse": 0.030462169891891065, + "train_head_mae": 0.022327352315187454, + "train_head_r2": 0.9080937504768372, + "train_left_shoulder_mse": 0.0006621021893806756, + "train_left_shoulder_rmse": 0.02573134643544087, + "train_left_shoulder_mae": 0.018719777464866638, + "train_left_shoulder_r2": 0.8828005194664001, + "train_left_elbow_mse": 0.0010456827003508806, + "train_left_elbow_rmse": 0.03233701749312822, + "train_left_elbow_mae": 0.0241067036986351, + "train_left_elbow_r2": 0.924235999584198, + "train_right_shoulder_mse": 0.0006559866014868021, + "train_right_shoulder_rmse": 0.02561223538636958, + "train_right_shoulder_mae": 0.018930697813630104, + "train_right_shoulder_r2": 0.9007279276847839, + "train_right_elbow_mse": 0.0011797392508015037, + "train_right_elbow_rmse": 0.03434733251362475, + "train_right_elbow_mae": 0.02548491396009922, + "train_right_elbow_r2": 0.9258905053138733, + "train_left_hand_mse": 0.001753255375660956, + "train_left_hand_rmse": 0.041871892429898076, + "train_left_hand_mae": 0.03192637488245964, + "train_left_hand_r2": 0.932714581489563, + "train_right_hand_mse": 0.0021199367474764585, + "train_right_hand_rmse": 0.04604277084924906, + "train_right_hand_mae": 0.03416356444358826, + "train_right_hand_r2": 0.9241409301757812, + "train_left_hip_mse": 0.0008367355912923813, + "train_left_hip_rmse": 0.028926382271075333, + "train_left_hip_mae": 0.02161344885826111, + "train_left_hip_r2": 0.8953300714492798, + "train_right_hip_mse": 0.0008361928048543632, + "train_right_hip_rmse": 0.028916998545048953, + "train_right_hip_mae": 0.021306367591023445, + "train_right_hip_r2": 0.899240255355835, + "train_left_knee_mse": 0.0009710715967230499, + "train_left_knee_rmse": 0.031162021704681644, + "train_left_knee_mae": 0.021543342620134354, + "train_left_knee_r2": 0.8984268307685852, + "train_right_knee_mse": 0.0009539542952552438, + "train_right_knee_rmse": 0.030886150541225493, + "train_right_knee_mae": 0.02151007018983364, + "train_right_knee_r2": 0.911297082901001, + "train_left_foot_mse": 0.0011605967301875353, + "train_left_foot_rmse": 0.03406753190631125, + "train_left_foot_mae": 0.01974382996559143, + "train_left_foot_r2": 0.7365503311157227, + "train_right_foot_mse": 0.0012203857768326998, + "train_right_foot_rmse": 0.03493402033595188, + "train_right_foot_mae": 0.021584484726190567, + "train_right_foot_r2": 0.7250706553459167 + }, + "val_metrics": { + "val_mse": 0.0012262249365448952, + "val_rmse": 0.035017494721137535, + "val_mae": 0.02452075108885765, + "val_r2": 0.9403130412101746, + "val_head_mse": 0.0010262727737426758, + "val_head_rmse": 0.03203549240674593, + "val_head_mae": 0.023517325520515442, + "val_head_r2": 0.8900337815284729, + "val_left_shoulder_mse": 0.0006540872855111957, + "val_left_shoulder_rmse": 0.02557513021494115, + "val_left_shoulder_mae": 0.019159896299242973, + "val_left_shoulder_r2": 0.8738552331924438, + "val_left_elbow_mse": 0.0010994977783411741, + "val_left_elbow_rmse": 0.033158675762780004, + "val_left_elbow_mae": 0.02516893297433853, + "val_left_elbow_r2": 0.9203133583068848, + "val_right_shoulder_mse": 0.0007205271976999938, + "val_right_shoulder_rmse": 0.0268426376814946, + "val_right_shoulder_mae": 0.019788872450590134, + "val_right_shoulder_r2": 0.8834189772605896, + "val_right_elbow_mse": 0.0013501467183232307, + "val_right_elbow_rmse": 0.03674434267099128, + "val_right_elbow_mae": 0.02697659656405449, + "val_right_elbow_r2": 0.9111173748970032, + "val_left_hand_mse": 0.001990390243008733, + "val_left_hand_rmse": 0.0446137898301493, + "val_left_hand_mae": 0.03459678217768669, + "val_left_hand_r2": 0.9266937971115112, + "val_right_hand_mse": 0.0023561131674796343, + "val_right_hand_rmse": 0.04853981013023881, + "val_right_hand_mae": 0.03641336411237717, + "val_right_hand_r2": 0.9142040610313416, + "val_left_hip_mse": 0.0007877653115428984, + "val_left_hip_rmse": 0.028067157168885103, + "val_left_hip_mae": 0.021093299612402916, + "val_left_hip_r2": 0.8876473903656006, + "val_right_hip_mse": 0.0008082646527327597, + "val_right_hip_rmse": 0.028429995651296884, + "val_right_hip_mae": 0.021135902032256126, + "val_right_hip_r2": 0.8904798626899719, + "val_left_knee_mse": 0.001120245666243136, + "val_left_knee_rmse": 0.033470071201644254, + "val_left_knee_mae": 0.022890888154506683, + "val_left_knee_r2": 0.8780137896537781, + "val_right_knee_mse": 0.0011903524864464998, + "val_right_knee_rmse": 0.034501485278847055, + "val_right_knee_mae": 0.02372620441019535, + "val_right_knee_r2": 0.8893001675605774, + "val_left_foot_mse": 0.0013415212742984295, + "val_left_foot_rmse": 0.03662678356474166, + "val_left_foot_mae": 0.021327316761016846, + "val_left_foot_r2": 0.6840490698814392, + "val_right_foot_mse": 0.0014957404928281903, + "val_right_foot_rmse": 0.038674804367031905, + "val_right_foot_mae": 0.02297438122332096, + "val_right_foot_r2": 0.6794781684875488 + }, + "best_epoch": 100 + } + }, + "fold_details": [ + { + "fold": 1, + "train_metrics": { + "train_mse": 0.0011018142104148865, + "train_rmse": 0.033193586886850396, + "train_mae": 0.023304687812924385, + "train_r2": 0.9482403993606567, + "train_head_mse": 0.0009279437945224345, + "train_head_rmse": 0.030462169891891065, + "train_head_mae": 0.022327352315187454, + "train_head_r2": 0.9080937504768372, + "train_left_shoulder_mse": 0.0006621021893806756, + "train_left_shoulder_rmse": 0.02573134643544087, + "train_left_shoulder_mae": 0.018719777464866638, + "train_left_shoulder_r2": 0.8828005194664001, + "train_left_elbow_mse": 0.0010456827003508806, + "train_left_elbow_rmse": 0.03233701749312822, + "train_left_elbow_mae": 0.0241067036986351, + "train_left_elbow_r2": 0.924235999584198, + "train_right_shoulder_mse": 0.0006559866014868021, + "train_right_shoulder_rmse": 0.02561223538636958, + "train_right_shoulder_mae": 0.018930697813630104, + "train_right_shoulder_r2": 0.9007279276847839, + "train_right_elbow_mse": 0.0011797392508015037, + "train_right_elbow_rmse": 0.03434733251362475, + "train_right_elbow_mae": 0.02548491396009922, + "train_right_elbow_r2": 0.9258905053138733, + "train_left_hand_mse": 0.001753255375660956, + "train_left_hand_rmse": 0.041871892429898076, + "train_left_hand_mae": 0.03192637488245964, + "train_left_hand_r2": 0.932714581489563, + "train_right_hand_mse": 0.0021199367474764585, + "train_right_hand_rmse": 0.04604277084924906, + "train_right_hand_mae": 0.03416356444358826, + "train_right_hand_r2": 0.9241409301757812, + "train_left_hip_mse": 0.0008367355912923813, + "train_left_hip_rmse": 0.028926382271075333, + "train_left_hip_mae": 0.02161344885826111, + "train_left_hip_r2": 0.8953300714492798, + "train_right_hip_mse": 0.0008361928048543632, + "train_right_hip_rmse": 0.028916998545048953, + "train_right_hip_mae": 0.021306367591023445, + "train_right_hip_r2": 0.899240255355835, + "train_left_knee_mse": 0.0009710715967230499, + "train_left_knee_rmse": 0.031162021704681644, + "train_left_knee_mae": 0.021543342620134354, + "train_left_knee_r2": 0.8984268307685852, + "train_right_knee_mse": 0.0009539542952552438, + "train_right_knee_rmse": 0.030886150541225493, + "train_right_knee_mae": 0.02151007018983364, + "train_right_knee_r2": 0.911297082901001, + "train_left_foot_mse": 0.0011605967301875353, + "train_left_foot_rmse": 0.03406753190631125, + "train_left_foot_mae": 0.01974382996559143, + "train_left_foot_r2": 0.7365503311157227, + "train_right_foot_mse": 0.0012203857768326998, + "train_right_foot_rmse": 0.03493402033595188, + "train_right_foot_mae": 0.021584484726190567, + "train_right_foot_r2": 0.7250706553459167 + }, + "val_metrics": { + "val_mse": 0.0012262249365448952, + "val_rmse": 0.035017494721137535, + "val_mae": 0.02452075108885765, + "val_r2": 0.9403130412101746, + "val_head_mse": 0.0010262727737426758, + "val_head_rmse": 0.03203549240674593, + "val_head_mae": 0.023517325520515442, + "val_head_r2": 0.8900337815284729, + "val_left_shoulder_mse": 0.0006540872855111957, + "val_left_shoulder_rmse": 0.02557513021494115, + "val_left_shoulder_mae": 0.019159896299242973, + "val_left_shoulder_r2": 0.8738552331924438, + "val_left_elbow_mse": 0.0010994977783411741, + "val_left_elbow_rmse": 0.033158675762780004, + "val_left_elbow_mae": 0.02516893297433853, + "val_left_elbow_r2": 0.9203133583068848, + "val_right_shoulder_mse": 0.0007205271976999938, + "val_right_shoulder_rmse": 0.0268426376814946, + "val_right_shoulder_mae": 0.019788872450590134, + "val_right_shoulder_r2": 0.8834189772605896, + "val_right_elbow_mse": 0.0013501467183232307, + "val_right_elbow_rmse": 0.03674434267099128, + "val_right_elbow_mae": 0.02697659656405449, + "val_right_elbow_r2": 0.9111173748970032, + "val_left_hand_mse": 0.001990390243008733, + "val_left_hand_rmse": 0.0446137898301493, + "val_left_hand_mae": 0.03459678217768669, + "val_left_hand_r2": 0.9266937971115112, + "val_right_hand_mse": 0.0023561131674796343, + "val_right_hand_rmse": 0.04853981013023881, + "val_right_hand_mae": 0.03641336411237717, + "val_right_hand_r2": 0.9142040610313416, + "val_left_hip_mse": 0.0007877653115428984, + "val_left_hip_rmse": 0.028067157168885103, + "val_left_hip_mae": 0.021093299612402916, + "val_left_hip_r2": 0.8876473903656006, + "val_right_hip_mse": 0.0008082646527327597, + "val_right_hip_rmse": 0.028429995651296884, + "val_right_hip_mae": 0.021135902032256126, + "val_right_hip_r2": 0.8904798626899719, + "val_left_knee_mse": 0.001120245666243136, + "val_left_knee_rmse": 0.033470071201644254, + "val_left_knee_mae": 0.022890888154506683, + "val_left_knee_r2": 0.8780137896537781, + "val_right_knee_mse": 0.0011903524864464998, + "val_right_knee_rmse": 0.034501485278847055, + "val_right_knee_mae": 0.02372620441019535, + "val_right_knee_r2": 0.8893001675605774, + "val_left_foot_mse": 0.0013415212742984295, + "val_left_foot_rmse": 0.03662678356474166, + "val_left_foot_mae": 0.021327316761016846, + "val_left_foot_r2": 0.6840490698814392, + "val_right_foot_mse": 0.0014957404928281903, + "val_right_foot_rmse": 0.038674804367031905, + "val_right_foot_mae": 0.02297438122332096, + "val_right_foot_r2": 0.6794781684875488 + }, + "best_epoch": 100 + }, + { + "fold": 2, + "train_metrics": { + "train_mse": 0.0009609388071112335, + "train_rmse": 0.030999013002210786, + "train_mae": 0.02107030339539051, + "train_r2": 0.9541847705841064, + "train_head_mse": 0.0009207567782141268, + "train_head_rmse": 0.030343974331226403, + "train_head_mae": 0.021782469004392624, + "train_head_r2": 0.9058244228363037, + "train_left_shoulder_mse": 0.0005358402850106359, + "train_left_shoulder_rmse": 0.023148224230178777, + "train_left_shoulder_mae": 0.016778161749243736, + "train_left_shoulder_r2": 0.9031292796134949, + "train_left_elbow_mse": 0.0008505755686201155, + "train_left_elbow_rmse": 0.029164628724194577, + "train_left_elbow_mae": 0.021592743694782257, + "train_left_elbow_r2": 0.9388150572776794, + "train_right_shoulder_mse": 0.0005230805254541337, + "train_right_shoulder_rmse": 0.02287095375042619, + "train_right_shoulder_mae": 0.016792556270956993, + "train_right_shoulder_r2": 0.918494701385498, + "train_right_elbow_mse": 0.0009833888616412878, + "train_right_elbow_rmse": 0.03135903158009328, + "train_right_elbow_mae": 0.02204115316271782, + "train_right_elbow_r2": 0.9371633529663086, + "train_left_hand_mse": 0.0013915594900026917, + "train_left_hand_rmse": 0.03730361229160913, + "train_left_hand_mae": 0.028157079592347145, + "train_left_hand_r2": 0.9473135471343994, + "train_right_hand_mse": 0.0018575697904452682, + "train_right_hand_rmse": 0.04309953352932336, + "train_right_hand_mae": 0.02979215979576111, + "train_right_hand_r2": 0.9329650402069092, + "train_left_hip_mse": 0.0007351770764216781, + "train_left_hip_rmse": 0.027114149007882914, + "train_left_hip_mae": 0.019951578229665756, + "train_left_hip_r2": 0.9026778340339661, + "train_right_hip_mse": 0.0007566532003693283, + "train_right_hip_rmse": 0.027507329938933154, + "train_right_hip_mae": 0.019738411530852318, + "train_right_hip_r2": 0.9031779766082764, + "train_left_knee_mse": 0.0008943179273046553, + "train_left_knee_rmse": 0.02990514884271027, + "train_left_knee_mae": 0.019483238458633423, + "train_left_knee_r2": 0.9037702083587646, + "train_right_knee_mse": 0.0007448363467119634, + "train_right_knee_rmse": 0.027291690066977594, + "train_right_knee_mae": 0.018714983016252518, + "train_right_knee_r2": 0.930830717086792, + "train_left_foot_mse": 0.0012588155223056674, + "train_left_foot_rmse": 0.03547979033626985, + "train_left_foot_mae": 0.01970815658569336, + "train_left_foot_r2": 0.708513081073761, + "train_right_foot_mse": 0.0010396336438134313, + "train_right_foot_rmse": 0.03224335038133338, + "train_right_foot_mae": 0.019381258636713028, + "train_right_foot_r2": 0.7724143266677856 + }, + "val_metrics": { + "val_mse": 0.0013959154020994902, + "val_rmse": 0.0373619512619388, + "val_mae": 0.02480027824640274, + "val_r2": 0.9359145164489746, + "val_head_mse": 0.0012911149533465505, + "val_head_rmse": 0.03593208807384495, + "val_head_mae": 0.02544349804520607, + "val_head_r2": 0.8784104585647583, + "val_left_shoulder_mse": 0.000720703974366188, + "val_left_shoulder_rmse": 0.02684593031292058, + "val_left_shoulder_mae": 0.019341595470905304, + "val_left_shoulder_r2": 0.8724914193153381, + "val_left_elbow_mse": 0.0012632880825549364, + "val_left_elbow_rmse": 0.03554276413779514, + "val_left_elbow_mae": 0.025299735367298126, + "val_left_elbow_r2": 0.9056519865989685, + "val_right_shoulder_mse": 0.0008292924612760544, + "val_right_shoulder_rmse": 0.02879743844990478, + "val_right_shoulder_mae": 0.020210696384310722, + "val_right_shoulder_r2": 0.8803569674491882, + "val_right_elbow_mse": 0.0015267902053892612, + "val_right_elbow_rmse": 0.03907416288788874, + "val_right_elbow_mae": 0.02776600606739521, + "val_right_elbow_r2": 0.9059062600135803, + "val_left_hand_mse": 0.0023726315703243017, + "val_left_hand_rmse": 0.04870966608717721, + "val_left_hand_mae": 0.03435362130403519, + "val_left_hand_r2": 0.9077266454696655, + "val_right_hand_mse": 0.0029963389970362186, + "val_right_hand_rmse": 0.05473882531655405, + "val_right_hand_mae": 0.03698313236236572, + "val_right_hand_r2": 0.8942573666572571, + "val_left_hip_mse": 0.001055872649885714, + "val_left_hip_rmse": 0.03249419409503356, + "val_left_hip_mae": 0.023282887414097786, + "val_left_hip_r2": 0.8794505596160889, + "val_right_hip_mse": 0.001184001099318266, + "val_right_hip_rmse": 0.03440931704231088, + "val_right_hip_mae": 0.02361701987683773, + "val_right_hip_r2": 0.8727319240570068, + "val_left_knee_mse": 0.0013590819435194135, + "val_left_knee_rmse": 0.03686572857708652, + "val_left_knee_mae": 0.02286830171942711, + "val_left_knee_r2": 0.867103099822998, + "val_right_knee_mse": 0.000840215478092432, + "val_right_knee_rmse": 0.02898647060427385, + "val_right_knee_mae": 0.021065374836325645, + "val_right_knee_r2": 0.9214866161346436, + "val_left_foot_mse": 0.0015386140439659357, + "val_left_foot_rmse": 0.03922517105081807, + "val_left_foot_mae": 0.02175767906010151, + "val_left_foot_r2": 0.6640238761901855, + "val_right_foot_mse": 0.0011689545353874564, + "val_right_foot_rmse": 0.03418997711884956, + "val_right_foot_mae": 0.02041410095989704, + "val_right_foot_r2": 0.7179993391036987 + }, + "best_epoch": 100 + }, + { + "fold": 3, + "train_metrics": { + "train_mse": 0.0008912301273085177, + "train_rmse": 0.029853477641784344, + "train_mae": 0.02028517983853817, + "train_r2": 0.9573683142662048, + "train_head_mse": 0.0007871467969380319, + "train_head_rmse": 0.028056136529073846, + "train_head_mae": 0.02109845168888569, + "train_head_r2": 0.9200119376182556, + "train_left_shoulder_mse": 0.0004777564899995923, + "train_left_shoulder_rmse": 0.021857641455554903, + "train_left_shoulder_mae": 0.016251051798462868, + "train_left_shoulder_r2": 0.9123229384422302, + "train_left_elbow_mse": 0.0008085868903435767, + "train_left_elbow_rmse": 0.028435662298310842, + "train_left_elbow_mae": 0.02117946930229664, + "train_left_elbow_r2": 0.9408701658248901, + "train_right_shoulder_mse": 0.0004489571147132665, + "train_right_shoulder_rmse": 0.021188608135346374, + "train_right_shoulder_mae": 0.015605945140123367, + "train_right_shoulder_r2": 0.9302448034286499, + "train_right_elbow_mse": 0.0009258670033887029, + "train_right_elbow_rmse": 0.03042806276102215, + "train_right_elbow_mae": 0.021198999136686325, + "train_right_elbow_r2": 0.9403999447822571, + "train_left_hand_mse": 0.0012487918138504028, + "train_left_hand_rmse": 0.03533824859625053, + "train_left_hand_mae": 0.02710305154323578, + "train_left_hand_r2": 0.9520196914672852, + "train_right_hand_mse": 0.001802817452698946, + "train_right_hand_rmse": 0.04245959788668454, + "train_right_hand_mae": 0.02908281609416008, + "train_right_hand_r2": 0.9341380000114441, + "train_left_hip_mse": 0.0006500851595774293, + "train_left_hip_rmse": 0.025496767629984575, + "train_left_hip_mae": 0.01897484064102173, + "train_left_hip_r2": 0.9162177443504333, + "train_right_hip_mse": 0.0006333807250484824, + "train_right_hip_rmse": 0.02516705634452473, + "train_right_hip_mae": 0.018675269559025764, + "train_right_hip_r2": 0.9218783378601074, + "train_left_knee_mse": 0.0007288604392670095, + "train_left_knee_rmse": 0.026997415418276792, + "train_left_knee_mae": 0.01898268423974514, + "train_left_knee_r2": 0.9225853681564331, + "train_right_knee_mse": 0.0008806405821815133, + "train_right_knee_rmse": 0.02967558899468574, + "train_right_knee_mae": 0.019359257072210312, + "train_right_knee_r2": 0.917660653591156, + "train_left_foot_mse": 0.0010069888085126877, + "train_left_foot_rmse": 0.031733086967906034, + "train_left_foot_mae": 0.018339399248361588, + "train_left_foot_r2": 0.7636898756027222, + "train_right_foot_mse": 0.0011861121747642756, + "train_right_foot_rmse": 0.034439979308418224, + "train_right_foot_mae": 0.017856091260910034, + "train_right_foot_r2": 0.733180820941925 + }, + "val_metrics": { + "val_mse": 0.001239411416463554, + "val_rmse": 0.03520527540672781, + "val_mae": 0.023660846054553986, + "val_r2": 0.9438684582710266, + "val_head_mse": 0.0010609752498567104, + "val_head_rmse": 0.03257261502944936, + "val_head_mae": 0.0243411622941494, + "val_head_r2": 0.8978170156478882, + "val_left_shoulder_mse": 0.0007713373051956296, + "val_left_shoulder_rmse": 0.027772959964606395, + "val_left_shoulder_mae": 0.019654646515846252, + "val_left_shoulder_r2": 0.8714439868927002, + "val_left_elbow_mse": 0.0011691275285556912, + "val_left_elbow_rmse": 0.03419250690656787, + "val_left_elbow_mae": 0.025177545845508575, + "val_left_elbow_r2": 0.9183889627456665, + "val_right_shoulder_mse": 0.0006158151081763208, + "val_right_shoulder_rmse": 0.024815622260510026, + "val_right_shoulder_mae": 0.01835784688591957, + "val_right_shoulder_r2": 0.9104443192481995, + "val_right_elbow_mse": 0.001170895411632955, + "val_right_elbow_rmse": 0.03421834904890876, + "val_right_elbow_mae": 0.02547324448823929, + "val_right_elbow_r2": 0.9300365447998047, + "val_left_hand_mse": 0.001675296574831009, + "val_left_hand_rmse": 0.04093038693722561, + "val_left_hand_mae": 0.032117705792188644, + "val_left_hand_r2": 0.9386305212974548, + "val_right_hand_mse": 0.0022483645007014275, + "val_right_hand_rmse": 0.04741692209223863, + "val_right_hand_mae": 0.0351017601788044, + "val_right_hand_r2": 0.9244474172592163, + "val_left_hip_mse": 0.0009901782032102346, + "val_left_hip_rmse": 0.03146709715258519, + "val_left_hip_mae": 0.02152426727116108, + "val_left_hip_r2": 0.8754059076309204, + "val_right_hip_mse": 0.0009166168747469783, + "val_right_hip_rmse": 0.030275681243317684, + "val_right_hip_mae": 0.021019887179136276, + "val_right_hip_r2": 0.8874129056930542, + "val_left_knee_mse": 0.001062354538589716, + "val_left_knee_rmse": 0.03259378067346155, + "val_left_knee_mae": 0.02164739929139614, + "val_left_knee_r2": 0.8913215398788452, + "val_right_knee_mse": 0.0012604023795574903, + "val_right_knee_rmse": 0.03550214612607934, + "val_right_knee_mae": 0.021472150459885597, + "val_right_knee_r2": 0.8854038119316101, + "val_left_foot_mse": 0.0017460761591792107, + "val_left_foot_rmse": 0.041786076140015956, + "val_left_foot_mae": 0.021384555846452713, + "val_left_foot_r2": 0.6363295316696167, + "val_right_foot_mse": 0.0014249086380004883, + "val_right_foot_rmse": 0.03774796203771123, + "val_right_foot_mae": 0.020318830385804176, + "val_right_foot_r2": 0.6926189661026001 + }, + "best_epoch": 100 + }, + { + "fold": 4, + "train_metrics": { + "train_mse": 0.0009543406777083874, + "train_rmse": 0.030892404854727436, + "train_mae": 0.02106698416173458, + "train_r2": 0.9551746249198914, + "train_head_mse": 0.0008447698201052845, + "train_head_rmse": 0.029064924223284747, + "train_head_mae": 0.021661274135112762, + "train_head_r2": 0.9151707887649536, + "train_left_shoulder_mse": 0.0005291140405461192, + "train_left_shoulder_rmse": 0.023002479008709456, + "train_left_shoulder_mae": 0.01712043210864067, + "train_left_shoulder_r2": 0.905330240726471, + "train_left_elbow_mse": 0.0008797493064776063, + "train_left_elbow_rmse": 0.02966056820894715, + "train_left_elbow_mae": 0.022161485627293587, + "train_left_elbow_r2": 0.9362634420394897, + "train_right_shoulder_mse": 0.0004634671495296061, + "train_right_shoulder_rmse": 0.021528287194517035, + "train_right_shoulder_mae": 0.015940982848405838, + "train_right_shoulder_r2": 0.9291806221008301, + "train_right_elbow_mse": 0.0009216979960910976, + "train_right_elbow_rmse": 0.030359479509555126, + "train_right_elbow_mae": 0.021655870601534843, + "train_right_elbow_r2": 0.9415090084075928, + "train_left_hand_mse": 0.0013732408406212926, + "train_left_hand_rmse": 0.03705726434346298, + "train_left_hand_mae": 0.028622910380363464, + "train_left_hand_r2": 0.9480606317520142, + "train_right_hand_mse": 0.0018843604484573007, + "train_right_hand_rmse": 0.04340922077689602, + "train_right_hand_mae": 0.031267985701560974, + "train_right_hand_r2": 0.9326073527336121, + "train_left_hip_mse": 0.000654924544505775, + "train_left_hip_rmse": 0.02559149359661868, + "train_left_hip_mae": 0.018894122913479805, + "train_left_hip_r2": 0.9166786670684814, + "train_right_hip_mse": 0.0006478539435192943, + "train_right_hip_rmse": 0.025452975140821833, + "train_right_hip_mae": 0.01882784627377987, + "train_right_hip_r2": 0.9208241701126099, + "train_left_knee_mse": 0.0007631930639036, + "train_left_knee_rmse": 0.027625949104123103, + "train_left_knee_mae": 0.018953688442707062, + "train_left_knee_r2": 0.9210693836212158, + "train_right_knee_mse": 0.0009606049861758947, + "train_right_knee_rmse": 0.030993628154443207, + "train_right_knee_mae": 0.01956183649599552, + "train_right_knee_r2": 0.9125192165374756, + "train_left_foot_mse": 0.001189007656648755, + "train_left_foot_rmse": 0.0344819903231927, + "train_left_foot_mae": 0.019760333001613617, + "train_left_foot_r2": 0.7208020091056824, + "train_right_foot_mse": 0.0012944453628733754, + "train_right_foot_rmse": 0.03597840133848884, + "train_right_foot_mae": 0.01944201998412609, + "train_right_foot_r2": 0.715862512588501 + }, + "val_metrics": { + "val_mse": 0.001228805398568511, + "val_rmse": 0.035054320683312504, + "val_mae": 0.023398209363222122, + "val_r2": 0.9401611685752869, + "val_head_mse": 0.0009862605948001146, + "val_head_rmse": 0.031404786176634204, + "val_head_mae": 0.023116178810596466, + "val_head_r2": 0.9005029797554016, + "val_left_shoulder_mse": 0.0006513707339763641, + "val_left_shoulder_rmse": 0.025521965715366912, + "val_left_shoulder_mae": 0.01895751990377903, + "val_left_shoulder_r2": 0.8801820278167725, + "val_left_elbow_mse": 0.0011728137033060193, + "val_left_elbow_rmse": 0.03424636773887151, + "val_left_elbow_mae": 0.025318948552012444, + "val_left_elbow_r2": 0.9150739312171936, + "val_right_shoulder_mse": 0.000587396789342165, + "val_right_shoulder_rmse": 0.02423627012025912, + "val_right_shoulder_mae": 0.017918061465024948, + "val_right_shoulder_r2": 0.9087865948677063, + "val_right_elbow_mse": 0.0013995901681482792, + "val_right_elbow_rmse": 0.037411096858395894, + "val_right_elbow_mae": 0.023974550887942314, + "val_right_elbow_r2": 0.9114885330200195, + "val_left_hand_mse": 0.0018477248959243298, + "val_left_hand_rmse": 0.04298517065133428, + "val_left_hand_mae": 0.03231104090809822, + "val_left_hand_r2": 0.927966833114624, + "val_right_hand_mse": 0.0026473328471183777, + "val_right_hand_rmse": 0.05145223850444583, + "val_right_hand_mae": 0.03345620632171631, + "val_right_hand_r2": 0.9031811952590942, + "val_left_hip_mse": 0.0007633835775777698, + "val_left_hip_rmse": 0.02762939698179766, + "val_left_hip_mae": 0.020980719476938248, + "val_left_hip_r2": 0.8988105654716492, + "val_right_hip_mse": 0.0007631260668858886, + "val_right_hip_rmse": 0.027624736503465305, + "val_right_hip_mae": 0.02111727185547352, + "val_right_hip_r2": 0.9027027487754822, + "val_left_knee_mse": 0.0009587103850208223, + "val_left_knee_rmse": 0.030963048703588965, + "val_left_knee_mae": 0.02044861763715744, + "val_left_knee_r2": 0.8905062079429626, + "val_right_knee_mse": 0.0008551102364435792, + "val_right_knee_rmse": 0.029242267977083774, + "val_right_knee_mae": 0.02032873034477234, + "val_right_knee_r2": 0.9132415652275085, + "val_left_foot_mse": 0.0018639072077348828, + "val_left_foot_rmse": 0.04317299164680255, + "val_left_foot_mae": 0.02368229441344738, + "val_left_foot_r2": 0.6137613654136658, + "val_right_foot_mse": 0.0014777411706745625, + "val_right_foot_rmse": 0.03844139917685831, + "val_right_foot_mae": 0.022566594183444977, + "val_right_foot_r2": 0.6480828523635864 + }, + "best_epoch": 100 + }, + { + "fold": 5, + "train_metrics": { + "train_mse": 0.000989423948340118, + "train_rmse": 0.031455110051311505, + "train_mae": 0.021496590226888657, + "train_r2": 0.9534255862236023, + "train_head_mse": 0.0008793224114924669, + "train_head_rmse": 0.02965337099711375, + "train_head_mae": 0.021577533334493637, + "train_head_r2": 0.9126991033554077, + "train_left_shoulder_mse": 0.0005666185752488673, + "train_left_shoulder_rmse": 0.023803751285225345, + "train_left_shoulder_mae": 0.017563728615641594, + "train_left_shoulder_r2": 0.8983926773071289, + "train_left_elbow_mse": 0.0009045019396580756, + "train_left_elbow_rmse": 0.030074938730745165, + "train_left_elbow_mae": 0.022666750475764275, + "train_left_elbow_r2": 0.9347050189971924, + "train_right_shoulder_mse": 0.0005077786627225578, + "train_right_shoulder_rmse": 0.02253394467736525, + "train_right_shoulder_mae": 0.01676481030881405, + "train_right_shoulder_r2": 0.9232202172279358, + "train_right_elbow_mse": 0.0010302858427166939, + "train_right_elbow_rmse": 0.03209806602767048, + "train_right_elbow_mae": 0.022837065160274506, + "train_right_elbow_r2": 0.9356580376625061, + "train_left_hand_mse": 0.00145688746124506, + "train_left_hand_rmse": 0.038169195187284996, + "train_left_hand_mae": 0.02906136028468609, + "train_left_hand_r2": 0.945020318031311, + "train_right_hand_mse": 0.0019343470921739936, + "train_right_hand_rmse": 0.04398121294568845, + "train_right_hand_mae": 0.030681142583489418, + "train_right_hand_r2": 0.93156898021698, + "train_left_hip_mse": 0.0006791924824938178, + "train_left_hip_rmse": 0.0260613215799548, + "train_left_hip_mae": 0.019195005297660828, + "train_left_hip_r2": 0.9131361842155457, + "train_right_hip_mse": 0.0006480540614575148, + "train_right_hip_rmse": 0.025456905967880597, + "train_right_hip_mae": 0.01897146739065647, + "train_right_hip_r2": 0.9206850528717041, + "train_left_knee_mse": 0.0008354582823812962, + "train_left_knee_rmse": 0.028904295223743066, + "train_left_knee_mae": 0.02018842101097107, + "train_left_knee_r2": 0.9120128750801086, + "train_right_knee_mse": 0.0009488193900324404, + "train_right_knee_rmse": 0.030802912038189512, + "train_right_knee_mae": 0.0200666394084692, + "train_right_knee_r2": 0.9103558659553528, + "train_left_foot_mse": 0.0012624843511730433, + "train_left_foot_rmse": 0.03553145579867286, + "train_left_foot_mae": 0.019969340413808823, + "train_left_foot_r2": 0.7268326282501221, + "train_right_foot_mse": 0.001208761241286993, + "train_right_foot_rmse": 0.03476724379767532, + "train_right_foot_mae": 0.019912393763661385, + "train_right_foot_r2": 0.7260886430740356 + }, + "val_metrics": { + "val_mse": 0.0012507449137046933, + "val_rmse": 0.035365872160950496, + "val_mae": 0.02407103218138218, + "val_r2": 0.9396548867225647, + "val_head_mse": 0.0010670314077287912, + "val_head_rmse": 0.03266544669415668, + "val_head_mae": 0.024321051314473152, + "val_head_r2": 0.8871725797653198, + "val_left_shoulder_mse": 0.0007049753912724555, + "val_left_shoulder_rmse": 0.026551372681510376, + "val_left_shoulder_mae": 0.019861722365021706, + "val_left_shoulder_r2": 0.8716084957122803, + "val_left_elbow_mse": 0.0010862088529393077, + "val_left_elbow_rmse": 0.03295768276046281, + "val_left_elbow_mae": 0.025121526792645454, + "val_left_elbow_r2": 0.9202998280525208, + "val_right_shoulder_mse": 0.0006753196357749403, + "val_right_shoulder_rmse": 0.025986912778838127, + "val_right_shoulder_mae": 0.01938425377011299, + "val_right_shoulder_r2": 0.8904924392700195, + "val_right_elbow_mse": 0.001322468277066946, + "val_right_elbow_rmse": 0.03636575692965768, + "val_right_elbow_mae": 0.02610102668404579, + "val_right_elbow_r2": 0.9108086824417114, + "val_left_hand_mse": 0.0018375293584540486, + "val_left_hand_rmse": 0.04286641294129996, + "val_left_hand_mae": 0.03293231874704361, + "val_left_hand_r2": 0.9277939200401306, + "val_right_hand_mse": 0.0025163658428937197, + "val_right_hand_rmse": 0.05016339146124113, + "val_right_hand_mae": 0.03609786182641983, + "val_right_hand_r2": 0.903942346572876, + "val_left_hip_mse": 0.0008731103735044599, + "val_left_hip_rmse": 0.029548441134930617, + "val_left_hip_mae": 0.021473867818713188, + "val_left_hip_r2": 0.8867710828781128, + "val_right_hip_mse": 0.0008481438853777945, + "val_right_hip_rmse": 0.02912290997441352, + "val_right_hip_mae": 0.020996352657675743, + "val_right_hip_r2": 0.8925349712371826, + "val_left_knee_mse": 0.0008716912707313895, + "val_left_knee_rmse": 0.029524418211564974, + "val_left_knee_mae": 0.022535109892487526, + "val_left_knee_r2": 0.907788872718811, + "val_right_knee_mse": 0.001695160404779017, + "val_right_knee_rmse": 0.04117232571496316, + "val_right_knee_mae": 0.024440951645374298, + "val_right_knee_r2": 0.8519278168678284, + "val_left_foot_mse": 0.0007806824869476259, + "val_left_foot_rmse": 0.02794069589232927, + "val_left_foot_mae": 0.017939938232302666, + "val_left_foot_r2": 0.768915593624115, + "val_right_foot_mse": 0.0019809966906905174, + "val_right_foot_rmse": 0.04450838899230703, + "val_right_foot_mae": 0.02171740122139454, + "val_right_foot_r2": 0.5834344625473022 + }, + "best_epoch": 100 + } + ], + "test_metrics": { + "test_mse": 0.001391171244904399, + "test_rmse": 0.037298408074667194, + "test_mae": 0.025799544528126717, + "test_r2": 0.9347655177116394, + "test_head_mse": 0.0011128688929602504, + "test_head_rmse": 0.03335968964124592, + "test_head_mae": 0.025346867740154266, + "test_head_r2": 0.89077228307724, + "test_left_shoulder_mse": 0.0008328401017934084, + "test_left_shoulder_rmse": 0.028858969174130396, + "test_left_shoulder_mae": 0.020718444138765335, + "test_left_shoulder_r2": 0.8539014458656311, + "test_left_elbow_mse": 0.0012530973181128502, + "test_left_elbow_rmse": 0.035399114651539665, + "test_left_elbow_mae": 0.025991136208176613, + "test_left_elbow_r2": 0.9075152277946472, + "test_right_shoulder_mse": 0.0008596102707087994, + "test_right_shoulder_rmse": 0.029319111014981327, + "test_right_shoulder_mae": 0.020807260647416115, + "test_right_shoulder_r2": 0.8708796501159668, + "test_right_elbow_mse": 0.00166393315885216, + "test_right_elbow_rmse": 0.040791336811290706, + "test_right_elbow_mae": 0.029100321233272552, + "test_right_elbow_r2": 0.8935747146606445, + "test_left_hand_mse": 0.0019411375978961587, + "test_left_hand_rmse": 0.04405834311337818, + "test_left_hand_mae": 0.033779602497816086, + "test_left_hand_r2": 0.9258810877799988, + "test_right_hand_mse": 0.002750793704763055, + "test_right_hand_rmse": 0.05244800954052551, + "test_right_hand_mae": 0.03874053433537483, + "test_right_hand_r2": 0.902238667011261, + "test_left_hip_mse": 0.0010438328608870506, + "test_left_hip_rmse": 0.03230840232643903, + "test_left_hip_mae": 0.023333478718996048, + "test_left_hip_r2": 0.8815681338310242, + "test_right_hip_mse": 0.0010959749342873693, + "test_right_hip_rmse": 0.03310551214355956, + "test_right_hip_mae": 0.023453062400221825, + "test_right_hip_r2": 0.8803987503051758, + "test_left_knee_mse": 0.0012220198987051845, + "test_left_knee_rmse": 0.0349574012006783, + "test_left_knee_mae": 0.023554084822535515, + "test_left_knee_r2": 0.8743963837623596, + "test_right_knee_mse": 0.0010661155683919787, + "test_right_knee_rmse": 0.032651425212262615, + "test_right_knee_mae": 0.023204246535897255, + "test_right_knee_r2": 0.9005808234214783, + "test_left_foot_mse": 0.001649297308176756, + "test_left_foot_rmse": 0.040611541563658426, + "test_left_foot_mae": 0.0230257548391819, + "test_left_foot_r2": 0.6453574299812317, + "test_right_foot_mse": 0.001593703986145556, + "test_right_foot_rmse": 0.03992122225265098, + "test_right_foot_mae": 0.024339277297258377, + "test_right_foot_r2": 0.6964194774627686 + } +} \ No newline at end of file diff --git a/A9/cv_results_loss_optimizer/conv1d_v3_adam_mae_fold1_best.h5 b/A9/cv_results_loss_optimizer/conv1d_v3_adam_mae_fold1_best.h5 new file mode 100644 index 0000000000000000000000000000000000000000..eee40262b8135cbdacbef46048c5aa26d2f6278b --- /dev/null +++ b/A9/cv_results_loss_optimizer/conv1d_v3_adam_mae_fold1_best.h5 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:aea548c58e1c390beade04dc3911aafcd5349cc31aab88df5a12a7c98da1d8e3 +size 2658520 diff --git a/A9/cv_results_loss_optimizer/conv1d_v3_adam_mae_fold2_best.h5 b/A9/cv_results_loss_optimizer/conv1d_v3_adam_mae_fold2_best.h5 new file mode 100644 index 0000000000000000000000000000000000000000..3f36452b7da8f18b5e734f02b74022b8665e1a1f --- /dev/null +++ b/A9/cv_results_loss_optimizer/conv1d_v3_adam_mae_fold2_best.h5 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:376205864e97aa2afbe1db885c4f224ed5577a688d81cdd084163bafb42c7764 +size 2658520 diff --git a/A9/cv_results_loss_optimizer/conv1d_v3_adam_mae_fold3_best.h5 b/A9/cv_results_loss_optimizer/conv1d_v3_adam_mae_fold3_best.h5 new file mode 100644 index 0000000000000000000000000000000000000000..a3f7830bd9b0f07b269595424740c3cd7fcdd763 --- /dev/null +++ b/A9/cv_results_loss_optimizer/conv1d_v3_adam_mae_fold3_best.h5 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:596bc082ff5d3b062ccbf01969f825709130e44c71238e6c7a62dfb6acaa03f8 +size 2658520 diff --git a/A9/cv_results_loss_optimizer/conv1d_v3_adam_mae_fold4_best.h5 b/A9/cv_results_loss_optimizer/conv1d_v3_adam_mae_fold4_best.h5 new file mode 100644 index 0000000000000000000000000000000000000000..fcc0e5780a61904c4b51ccefdb319e68068a601a --- /dev/null +++ b/A9/cv_results_loss_optimizer/conv1d_v3_adam_mae_fold4_best.h5 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:2546f4ed86502c625139219b29cf204a1bf552dfa814dc5f4e9ba3c46153de01 +size 2658520 diff --git a/A9/cv_results_loss_optimizer/conv1d_v3_adam_mae_fold5_best.h5 b/A9/cv_results_loss_optimizer/conv1d_v3_adam_mae_fold5_best.h5 new file mode 100644 index 0000000000000000000000000000000000000000..cdeaf3592cbe7772563e1604e25e32e40f9fefaf --- /dev/null +++ b/A9/cv_results_loss_optimizer/conv1d_v3_adam_mae_fold5_best.h5 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:8c2b6425f02152abea1c3e0f51c597f1ea2d89eeaf54fe136ad6cb6a07d21ee3 +size 2658520 diff --git a/A9/cv_results_loss_optimizer/conv1d_v3_adam_mae_predictions.png b/A9/cv_results_loss_optimizer/conv1d_v3_adam_mae_predictions.png new file mode 100644 index 0000000000000000000000000000000000000000..3e203efaf70d354047cb5ced833d62b9a01714da --- /dev/null +++ b/A9/cv_results_loss_optimizer/conv1d_v3_adam_mae_predictions.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:029026602742133f86a71d248b5b0f5c651d66146c8e8caf139cd5d0ea15988b +size 582414 diff --git a/A9/cv_results_loss_optimizer/conv1d_v3_adam_mae_test_predictions.png b/A9/cv_results_loss_optimizer/conv1d_v3_adam_mae_test_predictions.png new file mode 100644 index 0000000000000000000000000000000000000000..009e5f2032688a57c7088194bc0723a52a873ad8 --- /dev/null +++ b/A9/cv_results_loss_optimizer/conv1d_v3_adam_mae_test_predictions.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:f602c258fbc903840356f537151ca2f82721a334a2047bc580f50d81d1894169 +size 462996 diff --git a/A9/cv_results_loss_optimizer/conv1d_v3_adam_mae_training_history.png b/A9/cv_results_loss_optimizer/conv1d_v3_adam_mae_training_history.png new file mode 100644 index 0000000000000000000000000000000000000000..80403e22b016dbe244d1247347c488f39c690c25 --- /dev/null +++ b/A9/cv_results_loss_optimizer/conv1d_v3_adam_mae_training_history.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:9ea2746207398a10a176dbf04958a869d4d93ff8668cddcb24f771b7b3d08b3c +size 331705 diff --git a/A9/cv_results_loss_optimizer/conv1d_v3_adam_mse_cv_results.json b/A9/cv_results_loss_optimizer/conv1d_v3_adam_mse_cv_results.json new file mode 100644 index 0000000000000000000000000000000000000000..4230489e8110cc4d48d848535ce4bacccdeebdc7 --- /dev/null +++ b/A9/cv_results_loss_optimizer/conv1d_v3_adam_mse_cv_results.json @@ -0,0 +1,1243 @@ +{ + "model_name": "conv1d_v3_adam_mse", + "n_splits": 5, + "epochs": 100, + "batch_size": 32, + "patience": 10, + "aggregate_results": { + "n_folds": 5, + "train_metrics": { + "train_mse": { + "mean": 0.0008170344750396908, + "std": 1.796635137308971e-05 + }, + "train_rmse": { + "mean": 0.028582082263179775, + "std": 0.0003147197809394673 + }, + "train_mae": { + "mean": 0.021105413138866425, + "std": 0.0002403709787376843 + }, + "train_r2": { + "mean": 0.961344575881958, + "std": 0.00108251500713593 + }, + "train_head_mse": { + "mean": 0.0008609365439042449, + "std": 2.5353909509037964e-05 + }, + "train_head_rmse": { + "mean": 0.029338558805149306, + "std": 0.0004307100428839488 + }, + "train_head_mae": { + "mean": 0.02215624451637268, + "std": 0.0004302308255938736 + }, + "train_head_r2": { + "mean": 0.9134527683258057, + "std": 0.0027419652084781764 + }, + "train_left_shoulder_mse": { + "mean": 0.0005471199285238982, + "std": 9.860998190444608e-06 + }, + "train_left_shoulder_rmse": { + "mean": 0.023389638355352016, + "std": 0.00021153281008851938 + }, + "train_left_shoulder_mae": { + "mean": 0.01786181814968586, + "std": 0.00020003183696035547 + }, + "train_left_shoulder_r2": { + "mean": 0.9015629887580872, + "std": 0.002311455118613421 + }, + "train_left_elbow_mse": { + "mean": 0.0008355446043424308, + "std": 2.6214559806818892e-05 + }, + "train_left_elbow_rmse": { + "mean": 0.02890222319945432, + "std": 0.0004539806949147617 + }, + "train_left_elbow_mae": { + "mean": 0.022262413799762727, + "std": 0.000198079911156683 + }, + "train_left_elbow_r2": { + "mean": 0.9394777059555054, + "std": 0.0020325030505952573 + }, + "train_right_shoulder_mse": { + "mean": 0.0004862230212893337, + "std": 2.455276791159383e-05 + }, + "train_right_shoulder_rmse": { + "mean": 0.02204337291561482, + "std": 0.0005592242774350327 + }, + "train_right_shoulder_mae": { + "mean": 0.017011000588536263, + "std": 0.0004100467941657685 + }, + "train_right_shoulder_r2": { + "mean": 0.9254408717155457, + "std": 0.004189908220110148 + }, + "train_right_elbow_mse": { + "mean": 0.0007884912192821503, + "std": 3.1883715092770635e-05 + }, + "train_right_elbow_rmse": { + "mean": 0.02807424833110489, + "std": 0.0005725381433623813 + }, + "train_right_elbow_mae": { + "mean": 0.02161908559501171, + "std": 0.0004271389191876324 + }, + "train_right_elbow_r2": { + "mean": 0.9499905824661254, + "std": 0.0024993888216753636 + }, + "train_left_hand_mse": { + "mean": 0.0012179816840216517, + "std": 2.7495901997453332e-05 + }, + "train_left_hand_rmse": { + "mean": 0.03489736067312728, + "std": 0.00039484436341621383 + }, + "train_left_hand_mae": { + "mean": 0.02746143937110901, + "std": 0.00025953961305740333 + }, + "train_left_hand_r2": { + "mean": 0.9536584615707397, + "std": 0.0012817052715376116 + }, + "train_right_hand_mse": { + "mean": 0.0013592813396826387, + "std": 2.8455038963297654e-05 + }, + "train_right_hand_rmse": { + "mean": 0.03686639192296935, + "std": 0.000387925591652206 + }, + "train_right_hand_mae": { + "mean": 0.02812798023223877, + "std": 0.00034757157635303504 + }, + "train_right_hand_r2": { + "mean": 0.9511808156967163, + "std": 0.0014682734886926367 + }, + "train_left_hip_mse": { + "mean": 0.0006536097265779972, + "std": 2.7437697760042113e-05 + }, + "train_left_hip_rmse": { + "mean": 0.025560165795201862, + "std": 0.0005363311475105154 + }, + "train_left_hip_mae": { + "mean": 0.019498487561941148, + "std": 0.00041770454904992155 + }, + "train_left_hip_r2": { + "mean": 0.9161490559577942, + "std": 0.0037808054445543798 + }, + "train_right_hip_mse": { + "mean": 0.0006396665936335922, + "std": 3.222132193998072e-05 + }, + "train_right_hip_rmse": { + "mean": 0.02528360325152368, + "std": 0.0006371814915182128 + }, + "train_right_hip_mae": { + "mean": 0.019183660671114922, + "std": 0.0004564818324530987 + }, + "train_right_hip_r2": { + "mean": 0.9211291909217835, + "std": 0.004512458366059876 + }, + "train_left_knee_mse": { + "mean": 0.0006888612173497677, + "std": 3.430854576939173e-05 + }, + "train_left_knee_rmse": { + "mean": 0.026238107632045655, + "std": 0.0006503270246231723 + }, + "train_left_knee_mae": { + "mean": 0.019345495849847794, + "std": 0.0005149135422990339 + }, + "train_left_knee_r2": { + "mean": 0.9273600578308105, + "std": 0.003955218626707371 + }, + "train_right_knee_mse": { + "mean": 0.0006959178601391614, + "std": 3.388622493286968e-05 + }, + "train_right_knee_rmse": { + "mean": 0.02637229189374917, + "std": 0.0006481361045746023 + }, + "train_right_knee_mae": { + "mean": 0.019567203521728516, + "std": 0.00046823922309419634 + }, + "train_right_knee_r2": { + "mean": 0.935268223285675, + "std": 0.003686558219805545 + }, + "train_left_foot_mse": { + "mean": 0.0009252572199329734, + "std": 4.674665156249195e-05 + }, + "train_left_foot_rmse": { + "mean": 0.030408123118937785, + "std": 0.0007767034932936569 + }, + "train_left_foot_mae": { + "mean": 0.01961165592074394, + "std": 0.000264999345411343 + }, + "train_left_foot_r2": { + "mean": 0.7882539510726929, + "std": 0.011963830702987947 + }, + "train_right_foot_mse": { + "mean": 0.0009225573157891631, + "std": 4.626300402671263e-05 + }, + "train_right_foot_rmse": { + "mean": 0.030364254351169673, + "std": 0.000754568410839355 + }, + "train_right_foot_mae": { + "mean": 0.020663885027170183, + "std": 0.000474241696205098 + }, + "train_right_foot_r2": { + "mean": 0.7941907525062561, + "std": 0.011357537987227632 + } + }, + "val_metrics": { + "val_mse": { + "mean": 0.0010867850854992867, + "std": 5.567981310493115e-05 + }, + "val_rmse": { + "mean": 0.03295570360995441, + "std": 0.0008406456281484063 + }, + "val_mae": { + "mean": 0.023639746755361558, + "std": 0.0004889960693011201 + }, + "val_r2": { + "mean": 0.9485708117485047, + "std": 0.002204361947451878 + }, + "val_head_mse": { + "mean": 0.0010481817647814752, + "std": 6.732768859696303e-05 + }, + "val_head_rmse": { + "mean": 0.03235910691102896, + "std": 0.0010343909812409742 + }, + "val_head_mae": { + "mean": 0.024257607385516168, + "std": 0.000877859902711901 + }, + "val_head_r2": { + "mean": 0.8945841073989869, + "std": 0.003347518833981188 + }, + "val_left_shoulder_mse": { + "mean": 0.0006774062174372375, + "std": 6.853989805860463e-05 + }, + "val_left_shoulder_rmse": { + "mean": 0.025995326387538285, + "std": 0.0012842209477318374 + }, + "val_left_shoulder_mae": { + "mean": 0.01963539533317089, + "std": 0.0006412526295045318 + }, + "val_left_shoulder_r2": { + "mean": 0.8782560348510742, + "std": 0.007027505402006656 + }, + "val_left_elbow_mse": { + "mean": 0.001088447915390134, + "std": 0.00012124532100558284 + }, + "val_left_elbow_rmse": { + "mean": 0.03294103922899068, + "std": 0.0018264309196438088 + }, + "val_left_elbow_mae": { + "mean": 0.025225446000695227, + "std": 0.0012150430911028179 + }, + "val_left_elbow_r2": { + "mean": 0.9211379885673523, + "std": 0.007850651587815314 + }, + "val_right_shoulder_mse": { + "mean": 0.0006360255647450686, + "std": 3.323144121447375e-05 + }, + "val_right_shoulder_rmse": { + "mean": 0.025211018432852672, + "std": 0.000655831017436098 + }, + "val_right_shoulder_mae": { + "mean": 0.019012654945254327, + "std": 0.0004551102557804147 + }, + "val_right_shoulder_r2": { + "mean": 0.9023910641670227, + "std": 0.0031631841345448495 + }, + "val_right_elbow_mse": { + "mean": 0.0011270801536738873, + "std": 8.068367399625249e-05 + }, + "val_right_elbow_rmse": { + "mean": 0.0335504127688508, + "std": 0.0012041414840543165 + }, + "val_right_elbow_mae": { + "mean": 0.025309786200523376, + "std": 0.0009663138528753437 + }, + "val_right_elbow_r2": { + "mean": 0.9285561680793762, + "std": 0.0023069933698880913 + }, + "val_left_hand_mse": { + "mean": 0.0016907936660572886, + "std": 0.00013258091065307992 + }, + "val_left_hand_rmse": { + "mean": 0.04108619808071704, + "std": 0.001648633776607707 + }, + "val_left_hand_mae": { + "mean": 0.031744121387600896, + "std": 0.0011233689388838322 + }, + "val_left_hand_r2": { + "mean": 0.9354806780815125, + "std": 0.0059281049385666055 + }, + "val_right_hand_mse": { + "mean": 0.0019399255281314254, + "std": 0.00018191935530437637 + }, + "val_right_hand_rmse": { + "mean": 0.043994740188325554, + "std": 0.0020948422119952677 + }, + "val_right_hand_mae": { + "mean": 0.03276348300278187, + "std": 0.0013642843064787717 + }, + "val_right_hand_r2": { + "mean": 0.9302879333496094, + "std": 0.0055637914334194525 + }, + "val_left_hip_mse": { + "mean": 0.0008158349082805216, + "std": 6.93089259597926e-05 + }, + "val_left_hip_rmse": { + "mean": 0.028537055723253102, + "std": 0.001212995848496211 + }, + "val_left_hip_mae": { + "mean": 0.021349154785275458, + "std": 0.00048603375705415295 + }, + "val_left_hip_r2": { + "mean": 0.8948533296585083, + "std": 0.01095343185686127 + }, + "val_right_hip_mse": { + "mean": 0.0008037451189011335, + "std": 6.872765204432651e-05 + }, + "val_right_hip_rmse": { + "mean": 0.02832476480889231, + "std": 0.0012053287609898333 + }, + "val_right_hip_mae": { + "mean": 0.020990831032395362, + "std": 0.0006166514706112613 + }, + "val_right_hip_r2": { + "mean": 0.9004216313362121, + "std": 0.010650996092133767 + }, + "val_left_knee_mse": { + "mean": 0.0009054047404788435, + "std": 7.19296371195461e-05 + }, + "val_left_knee_rmse": { + "mean": 0.03006581968364832, + "std": 0.0012046689292867158 + }, + "val_left_knee_mae": { + "mean": 0.02155601903796196, + "std": 0.00045897427061244154 + }, + "val_left_knee_r2": { + "mean": 0.904265558719635, + "std": 0.008530641570409664 + }, + "val_right_knee_mse": { + "mean": 0.0009136367938481271, + "std": 0.00011710072267982595 + }, + "val_right_knee_rmse": { + "mean": 0.030165055171998773, + "std": 0.0019251598163262092 + }, + "val_right_knee_mae": { + "mean": 0.02165892720222473, + "std": 0.0011634002798049043 + }, + "val_right_knee_r2": { + "mean": 0.9150413274765015, + "std": 0.009655689003008363 + }, + "val_left_foot_mse": { + "mean": 0.0011936229770071805, + "std": 0.00036916343474743287 + }, + "val_left_foot_rmse": { + "mean": 0.03408873021136283, + "std": 0.0056197375013517775 + }, + "val_left_foot_mae": { + "mean": 0.021269287541508676, + "std": 0.0014359114569855015 + }, + "val_left_foot_r2": { + "mean": 0.7318541526794433, + "std": 0.06128671175174108 + }, + "val_right_foot_mse": { + "mean": 0.0012881008675321936, + "std": 0.00022345022021789468 + }, + "val_right_foot_rmse": { + "mean": 0.0357622308194708, + "std": 0.0030271627553017954 + }, + "val_right_foot_mae": { + "mean": 0.022543999552726745, + "std": 0.0007878444209290629 + }, + "val_right_foot_r2": { + "mean": 0.7110320925712585, + "std": 0.058026363035473615 + } + }, + "best_fold": "1", + "best_fold_metrics": { + "fold": 1, + "train_metrics": { + "train_mse": 0.0008266802178695798, + "train_rmse": 0.028752047194409998, + "train_mae": 0.021389417350292206, + "train_r2": 0.9611653089523315, + "train_head_mse": 0.0008964512380771339, + "train_head_rmse": 0.029940795548500942, + "train_head_mae": 0.022725453600287437, + "train_head_r2": 0.9112128615379333, + "train_left_shoulder_mse": 0.0005458136438392103, + "train_left_shoulder_rmse": 0.023362654897061897, + "train_left_shoulder_mae": 0.01807209476828575, + "train_left_shoulder_r2": 0.9033848643302917, + "train_left_elbow_mse": 0.0008148289052769542, + "train_left_elbow_rmse": 0.02854520809657821, + "train_left_elbow_mae": 0.022171810269355774, + "train_left_elbow_r2": 0.9409623146057129, + "train_right_shoulder_mse": 0.000515289546456188, + "train_right_shoulder_rmse": 0.022699990010045995, + "train_right_shoulder_mae": 0.017417393624782562, + "train_right_shoulder_r2": 0.9220199584960938, + "train_right_elbow_mse": 0.0007829813985154033, + "train_right_elbow_rmse": 0.02798180477587897, + "train_right_elbow_mae": 0.021562235429883003, + "train_right_elbow_r2": 0.9508142471313477, + "train_left_hand_mse": 0.0012159181060269475, + "train_left_hand_rmse": 0.034870017293183946, + "train_left_hand_mae": 0.02742750756442547, + "train_left_hand_r2": 0.9533361792564392, + "train_right_hand_mse": 0.0013560008956119418, + "train_right_hand_rmse": 0.036823917439782826, + "train_right_hand_mae": 0.02833450958132744, + "train_right_hand_r2": 0.9514773488044739, + "train_left_hip_mse": 0.0006933070253580809, + "train_left_hip_rmse": 0.026330723980895036, + "train_left_hip_mae": 0.020166747272014618, + "train_left_hip_r2": 0.9132720232009888, + "train_right_hip_mse": 0.0006838225526735187, + "train_right_hip_rmse": 0.026150001007141828, + "train_right_hip_mae": 0.019824938848614693, + "train_right_hip_r2": 0.9176005721092224, + "train_left_knee_mse": 0.0007464407244697213, + "train_left_knee_rmse": 0.02732106741087766, + "train_left_knee_mae": 0.020169487223029137, + "train_left_knee_r2": 0.921923041343689, + "train_right_knee_mse": 0.0007346547208726406, + "train_right_knee_rmse": 0.027104514769178963, + "train_right_knee_mae": 0.02036399580538273, + "train_right_knee_r2": 0.9316885471343994, + "train_left_foot_mse": 0.0008428630535490811, + "train_left_foot_rmse": 0.02903210384297151, + "train_left_foot_mae": 0.019294455647468567, + "train_left_foot_r2": 0.8086743354797363, + "train_right_foot_mse": 0.0009184712544083595, + "train_right_foot_rmse": 0.03030629067385779, + "train_right_foot_mae": 0.020531784743070602, + "train_right_foot_r2": 0.7930861711502075 + }, + "val_metrics": { + "val_mse": 0.0010198858799412847, + "val_rmse": 0.031935652176545336, + "val_mae": 0.02304425649344921, + "val_r2": 0.950356662273407, + "val_head_mse": 0.0010161129757761955, + "val_head_rmse": 0.031876527034421356, + "val_head_mae": 0.0236684400588274, + "val_head_r2": 0.8911224007606506, + "val_left_shoulder_mse": 0.0006082829204387963, + "val_left_shoulder_rmse": 0.024663392314091676, + "val_left_shoulder_mae": 0.018625659868121147, + "val_left_shoulder_r2": 0.8826888799667358, + "val_left_elbow_mse": 0.0009294748306274414, + "val_left_elbow_rmse": 0.030487289656961004, + "val_left_elbow_mae": 0.023702412843704224, + "val_left_elbow_r2": 0.9326358437538147, + "val_right_shoulder_mse": 0.0006312417099252343, + "val_right_shoulder_rmse": 0.025124524073606536, + "val_right_shoulder_mae": 0.018916338682174683, + "val_right_shoulder_r2": 0.897865355014801, + "val_right_elbow_mse": 0.0010294223902747035, + "val_right_elbow_rmse": 0.032084612983090566, + "val_right_elbow_mae": 0.02456171065568924, + "val_right_elbow_r2": 0.9322312474250793, + "val_left_hand_mse": 0.0014439364895224571, + "val_left_hand_rmse": 0.037999164326632986, + "val_left_hand_mae": 0.029895564541220665, + "val_left_hand_r2": 0.9468197226524353, + "val_right_hand_mse": 0.0016341489972546697, + "val_right_hand_rmse": 0.040424608807688785, + "val_right_hand_mae": 0.030808093026280403, + "val_right_hand_r2": 0.9404938220977783, + "val_left_hip_mse": 0.0007928804261609912, + "val_left_hip_rmse": 0.028158132504855347, + "val_left_hip_mae": 0.020829634740948677, + "val_left_hip_r2": 0.8869178891181946, + "val_right_hip_mse": 0.0007835486903786659, + "val_right_hip_rmse": 0.027991939739479756, + "val_right_hip_mae": 0.02039255015552044, + "val_right_hip_r2": 0.8938288688659668, + "val_left_knee_mse": 0.0009098558803088963, + "val_left_knee_rmse": 0.030163817402790654, + "val_left_knee_mae": 0.02154775708913803, + "val_left_knee_r2": 0.9009236097335815, + "val_right_knee_mse": 0.0009827071335166693, + "val_right_knee_rmse": 0.03134815997019074, + "val_right_knee_mae": 0.022728169336915016, + "val_right_knee_r2": 0.9086107015609741, + "val_left_foot_mse": 0.001267193234525621, + "val_left_foot_rmse": 0.03559765771122618, + "val_left_foot_mae": 0.021626893430948257, + "val_left_foot_r2": 0.701554536819458, + "val_right_foot_mse": 0.0012297116918489337, + "val_right_foot_rmse": 0.03506724528458051, + "val_right_foot_mae": 0.022272100672125816, + "val_right_foot_r2": 0.7364853620529175 + }, + "best_epoch": 100 + } + }, + "fold_details": [ + { + "fold": 1, + "train_metrics": { + "train_mse": 0.0008266802178695798, + "train_rmse": 0.028752047194409998, + "train_mae": 0.021389417350292206, + "train_r2": 0.9611653089523315, + "train_head_mse": 0.0008964512380771339, + "train_head_rmse": 0.029940795548500942, + "train_head_mae": 0.022725453600287437, + "train_head_r2": 0.9112128615379333, + "train_left_shoulder_mse": 0.0005458136438392103, + "train_left_shoulder_rmse": 0.023362654897061897, + "train_left_shoulder_mae": 0.01807209476828575, + "train_left_shoulder_r2": 0.9033848643302917, + "train_left_elbow_mse": 0.0008148289052769542, + "train_left_elbow_rmse": 0.02854520809657821, + "train_left_elbow_mae": 0.022171810269355774, + "train_left_elbow_r2": 0.9409623146057129, + "train_right_shoulder_mse": 0.000515289546456188, + "train_right_shoulder_rmse": 0.022699990010045995, + "train_right_shoulder_mae": 0.017417393624782562, + "train_right_shoulder_r2": 0.9220199584960938, + "train_right_elbow_mse": 0.0007829813985154033, + "train_right_elbow_rmse": 0.02798180477587897, + "train_right_elbow_mae": 0.021562235429883003, + "train_right_elbow_r2": 0.9508142471313477, + "train_left_hand_mse": 0.0012159181060269475, + "train_left_hand_rmse": 0.034870017293183946, + "train_left_hand_mae": 0.02742750756442547, + "train_left_hand_r2": 0.9533361792564392, + "train_right_hand_mse": 0.0013560008956119418, + "train_right_hand_rmse": 0.036823917439782826, + "train_right_hand_mae": 0.02833450958132744, + "train_right_hand_r2": 0.9514773488044739, + "train_left_hip_mse": 0.0006933070253580809, + "train_left_hip_rmse": 0.026330723980895036, + "train_left_hip_mae": 0.020166747272014618, + "train_left_hip_r2": 0.9132720232009888, + "train_right_hip_mse": 0.0006838225526735187, + "train_right_hip_rmse": 0.026150001007141828, + "train_right_hip_mae": 0.019824938848614693, + "train_right_hip_r2": 0.9176005721092224, + "train_left_knee_mse": 0.0007464407244697213, + "train_left_knee_rmse": 0.02732106741087766, + "train_left_knee_mae": 0.020169487223029137, + "train_left_knee_r2": 0.921923041343689, + "train_right_knee_mse": 0.0007346547208726406, + "train_right_knee_rmse": 0.027104514769178963, + "train_right_knee_mae": 0.02036399580538273, + "train_right_knee_r2": 0.9316885471343994, + "train_left_foot_mse": 0.0008428630535490811, + "train_left_foot_rmse": 0.02903210384297151, + "train_left_foot_mae": 0.019294455647468567, + "train_left_foot_r2": 0.8086743354797363, + "train_right_foot_mse": 0.0009184712544083595, + "train_right_foot_rmse": 0.03030629067385779, + "train_right_foot_mae": 0.020531784743070602, + "train_right_foot_r2": 0.7930861711502075 + }, + "val_metrics": { + "val_mse": 0.0010198858799412847, + "val_rmse": 0.031935652176545336, + "val_mae": 0.02304425649344921, + "val_r2": 0.950356662273407, + "val_head_mse": 0.0010161129757761955, + "val_head_rmse": 0.031876527034421356, + "val_head_mae": 0.0236684400588274, + "val_head_r2": 0.8911224007606506, + "val_left_shoulder_mse": 0.0006082829204387963, + "val_left_shoulder_rmse": 0.024663392314091676, + "val_left_shoulder_mae": 0.018625659868121147, + "val_left_shoulder_r2": 0.8826888799667358, + "val_left_elbow_mse": 0.0009294748306274414, + "val_left_elbow_rmse": 0.030487289656961004, + "val_left_elbow_mae": 0.023702412843704224, + "val_left_elbow_r2": 0.9326358437538147, + "val_right_shoulder_mse": 0.0006312417099252343, + "val_right_shoulder_rmse": 0.025124524073606536, + "val_right_shoulder_mae": 0.018916338682174683, + "val_right_shoulder_r2": 0.897865355014801, + "val_right_elbow_mse": 0.0010294223902747035, + "val_right_elbow_rmse": 0.032084612983090566, + "val_right_elbow_mae": 0.02456171065568924, + "val_right_elbow_r2": 0.9322312474250793, + "val_left_hand_mse": 0.0014439364895224571, + "val_left_hand_rmse": 0.037999164326632986, + "val_left_hand_mae": 0.029895564541220665, + "val_left_hand_r2": 0.9468197226524353, + "val_right_hand_mse": 0.0016341489972546697, + "val_right_hand_rmse": 0.040424608807688785, + "val_right_hand_mae": 0.030808093026280403, + "val_right_hand_r2": 0.9404938220977783, + "val_left_hip_mse": 0.0007928804261609912, + "val_left_hip_rmse": 0.028158132504855347, + "val_left_hip_mae": 0.020829634740948677, + "val_left_hip_r2": 0.8869178891181946, + "val_right_hip_mse": 0.0007835486903786659, + "val_right_hip_rmse": 0.027991939739479756, + "val_right_hip_mae": 0.02039255015552044, + "val_right_hip_r2": 0.8938288688659668, + "val_left_knee_mse": 0.0009098558803088963, + "val_left_knee_rmse": 0.030163817402790654, + "val_left_knee_mae": 0.02154775708913803, + "val_left_knee_r2": 0.9009236097335815, + "val_right_knee_mse": 0.0009827071335166693, + "val_right_knee_rmse": 0.03134815997019074, + "val_right_knee_mae": 0.022728169336915016, + "val_right_knee_r2": 0.9086107015609741, + "val_left_foot_mse": 0.001267193234525621, + "val_left_foot_rmse": 0.03559765771122618, + "val_left_foot_mae": 0.021626893430948257, + "val_left_foot_r2": 0.701554536819458, + "val_right_foot_mse": 0.0012297116918489337, + "val_right_foot_rmse": 0.03506724528458051, + "val_right_foot_mae": 0.022272100672125816, + "val_right_foot_r2": 0.7364853620529175 + }, + "best_epoch": 100 + }, + { + "fold": 2, + "train_metrics": { + "train_mse": 0.0008391687297262251, + "train_rmse": 0.028968409168026902, + "train_mae": 0.021331805735826492, + "train_r2": 0.9599905014038086, + "train_head_mse": 0.0008869636803865433, + "train_head_rmse": 0.02978193547079409, + "train_head_mae": 0.02261972613632679, + "train_head_r2": 0.9092808365821838, + "train_left_shoulder_mse": 0.0005495596560649574, + "train_left_shoulder_rmse": 0.023442688755024612, + "train_left_shoulder_mae": 0.017772821709513664, + "train_left_shoulder_r2": 0.9006490707397461, + "train_left_elbow_mse": 0.0008618637803010643, + "train_left_elbow_rmse": 0.029357516589470986, + "train_left_elbow_mae": 0.02238028682768345, + "train_left_elbow_r2": 0.9380030632019043, + "train_right_shoulder_mse": 0.0005083830328658223, + "train_right_shoulder_rmse": 0.022547350905723323, + "train_right_shoulder_mae": 0.01742667518556118, + "train_right_shoulder_r2": 0.9207848310470581, + "train_right_elbow_mse": 0.0008253277046605945, + "train_right_elbow_rmse": 0.02872851727222612, + "train_right_elbow_mae": 0.021982943639159203, + "train_right_elbow_r2": 0.9472631812095642, + "train_left_hand_mse": 0.0012308781733736396, + "train_left_hand_rmse": 0.03508387340892735, + "train_left_hand_mae": 0.027294425293803215, + "train_left_hand_r2": 0.9533971548080444, + "train_right_hand_mse": 0.0013890113914385438, + "train_right_hand_rmse": 0.03726944313292786, + "train_right_hand_mae": 0.028119489550590515, + "train_right_hand_r2": 0.9498741626739502, + "train_left_hip_mse": 0.0006735184579156339, + "train_left_hip_rmse": 0.025952234160388464, + "train_left_hip_mae": 0.019752586260437965, + "train_left_hip_r2": 0.9108401536941528, + "train_right_hip_mse": 0.000667242449708283, + "train_right_hip_rmse": 0.02583103655892041, + "train_right_hip_mae": 0.019556812942028046, + "train_right_hip_r2": 0.9146190881729126, + "train_left_knee_mse": 0.0006993560818955302, + "train_left_knee_rmse": 0.026445341402514172, + "train_left_knee_mae": 0.01960599049925804, + "train_left_knee_r2": 0.9247483015060425, + "train_right_knee_mse": 0.0007128438446670771, + "train_right_knee_rmse": 0.026699135653932265, + "train_right_knee_mae": 0.019666140899062157, + "train_right_knee_r2": 0.9338017106056213, + "train_left_foot_mse": 0.0009739958331920207, + "train_left_foot_rmse": 0.031208906312013254, + "train_left_foot_mae": 0.019981740042567253, + "train_left_foot_r2": 0.7744649648666382, + "train_right_foot_mse": 0.0009302497492171824, + "train_right_foot_rmse": 0.030499995888805993, + "train_right_foot_mae": 0.02115384303033352, + "train_right_foot_r2": 0.7963595390319824 + }, + "val_metrics": { + "val_mse": 0.001067714998498559, + "val_rmse": 0.0326759085336362, + "val_mae": 0.02410397119820118, + "val_r2": 0.9509819746017456, + "val_head_mse": 0.0011426527053117752, + "val_head_rmse": 0.03380314638183515, + "val_head_mae": 0.02540011890232563, + "val_head_r2": 0.8923918008804321, + "val_left_shoulder_mse": 0.000657710712403059, + "val_left_shoulder_rmse": 0.025645871254513054, + "val_left_shoulder_mae": 0.019938606768846512, + "val_left_shoulder_r2": 0.8836363554000854, + "val_left_elbow_mse": 0.001124913920648396, + "val_left_elbow_rmse": 0.03353973644273902, + "val_left_elbow_mae": 0.02572336234152317, + "val_left_elbow_r2": 0.9159864187240601, + "val_right_shoulder_mse": 0.0006886268965899944, + "val_right_shoulder_rmse": 0.026241701480467963, + "val_right_shoulder_mae": 0.019883878529071808, + "val_right_shoulder_r2": 0.9006509780883789, + "val_right_elbow_mse": 0.0011662066681310534, + "val_right_elbow_rmse": 0.03414976820025362, + "val_right_elbow_mae": 0.026423536241054535, + "val_right_elbow_r2": 0.9281284809112549, + "val_left_hand_mse": 0.0018137124134227633, + "val_left_hand_rmse": 0.04258770260794498, + "val_left_hand_mae": 0.032669417560100555, + "val_left_hand_r2": 0.9294633865356445, + "val_right_hand_mse": 0.0020469308365136385, + "val_right_hand_rmse": 0.045243019754583565, + "val_right_hand_mae": 0.033664632588624954, + "val_right_hand_r2": 0.9277625679969788, + "val_left_hip_mse": 0.0007906691753305495, + "val_left_hip_rmse": 0.02811884022022511, + "val_left_hip_mae": 0.021461840718984604, + "val_left_hip_r2": 0.90972900390625, + "val_right_hip_mse": 0.0008158496348187327, + "val_right_hip_rmse": 0.028563081675805443, + "val_right_hip_mae": 0.021269500255584717, + "val_right_hip_r2": 0.9123044610023499, + "val_left_knee_mse": 0.0009041365119628608, + "val_left_knee_rmse": 0.03006886283122228, + "val_left_knee_mae": 0.022007610648870468, + "val_left_knee_r2": 0.9115896224975586, + "val_right_knee_mse": 0.000762073090299964, + "val_right_knee_rmse": 0.027605671343040435, + "val_right_knee_mae": 0.021069034934043884, + "val_right_knee_r2": 0.9287885427474976, + "val_left_foot_mse": 0.0009216222097165883, + "val_left_foot_rmse": 0.030358231333801187, + "val_left_foot_mae": 0.021172182634472847, + "val_left_foot_r2": 0.798751950263977, + "val_right_foot_mse": 0.0010451903799548745, + "val_right_foot_rmse": 0.032329404262294636, + "val_right_foot_mae": 0.022667910903692245, + "val_right_foot_r2": 0.7478564381599426 + }, + "best_epoch": 100 + }, + { + "fold": 3, + "train_metrics": { + "train_mse": 0.0008277649758383632, + "train_rmse": 0.02877090502292834, + "train_mae": 0.02114626206457615, + "train_r2": 0.9604041576385498, + "train_head_mse": 0.0008377624326385558, + "train_head_rmse": 0.02894412604724067, + "train_head_mae": 0.021934615448117256, + "train_head_r2": 0.9148684740066528, + "train_left_shoulder_mse": 0.000551520730368793, + "train_left_shoulder_rmse": 0.023484478498974445, + "train_left_shoulder_mae": 0.01783871464431286, + "train_left_shoulder_r2": 0.8987858295440674, + "train_left_elbow_mse": 0.0008664450142532587, + "train_left_elbow_rmse": 0.029435438067969343, + "train_left_elbow_mae": 0.022592274472117424, + "train_left_elbow_r2": 0.936639130115509, + "train_right_shoulder_mse": 0.0004883183282800019, + "train_right_shoulder_rmse": 0.022097925881856013, + "train_right_shoulder_mae": 0.01709161140024662, + "train_right_shoulder_r2": 0.9241292476654053, + "train_right_elbow_mse": 0.0008089124457910657, + "train_right_elbow_rmse": 0.02844138614398155, + "train_right_elbow_mae": 0.021868502721190453, + "train_right_elbow_r2": 0.9479285478591919, + "train_left_hand_mse": 0.0012563413474708796, + "train_left_hand_rmse": 0.035444905804232006, + "train_left_hand_mae": 0.027817759662866592, + "train_left_hand_r2": 0.9517296552658081, + "train_right_hand_mse": 0.0013775277184322476, + "train_right_hand_rmse": 0.03711506053386209, + "train_right_hand_mae": 0.02840971201658249, + "train_right_hand_r2": 0.9496750235557556, + "train_left_hip_mse": 0.000635413161944598, + "train_left_hip_rmse": 0.02520740291947185, + "train_left_hip_mae": 0.019105229526758194, + "train_left_hip_r2": 0.9181087017059326, + "train_right_hip_mse": 0.0006230846629478037, + "train_right_hip_rmse": 0.024961663865772324, + "train_right_hip_mae": 0.018940232694149017, + "train_right_hip_r2": 0.9231482744216919, + "train_left_knee_mse": 0.0006869036005809903, + "train_left_knee_rmse": 0.0262088458460305, + "train_left_knee_mae": 0.0193291325122118, + "train_left_knee_r2": 0.9270417094230652, + "train_right_knee_mse": 0.0006810201448388398, + "train_right_knee_rmse": 0.026096362674496224, + "train_right_knee_mae": 0.019267242401838303, + "train_right_knee_r2": 0.9363250136375427, + "train_left_foot_mse": 0.0009425205062143505, + "train_left_foot_rmse": 0.030700496839861573, + "train_left_foot_mae": 0.019530558958649635, + "train_left_foot_r2": 0.7788186073303223, + "train_right_foot_mse": 0.0010051748249679804, + "train_right_foot_rmse": 0.03170449218908861, + "train_right_foot_mae": 0.021175799891352654, + "train_right_foot_r2": 0.7738831639289856 + }, + "val_metrics": { + "val_mse": 0.0011737592285498977, + "val_rmse": 0.03426016970988173, + "val_mae": 0.02432257868349552, + "val_r2": 0.9468417763710022, + "val_head_mse": 0.0011144138406962156, + "val_head_rmse": 0.03338283751714668, + "val_head_mae": 0.025250932201743126, + "val_head_r2": 0.8926703333854675, + "val_left_shoulder_mse": 0.0008071360643953085, + "val_left_shoulder_rmse": 0.028410140168526245, + "val_left_shoulder_mae": 0.020556125789880753, + "val_left_shoulder_r2": 0.8654775619506836, + "val_left_elbow_mse": 0.00128572853282094, + "val_left_elbow_rmse": 0.035857056945891974, + "val_left_elbow_mae": 0.02724616415798664, + "val_left_elbow_r2": 0.9102495908737183, + "val_right_shoulder_mse": 0.0006561246118508279, + "val_right_shoulder_rmse": 0.025614929471908136, + "val_right_shoulder_mae": 0.018965944647789, + "val_right_shoulder_r2": 0.9045822620391846, + "val_right_elbow_mse": 0.0012379701947793365, + "val_right_elbow_rmse": 0.035184800621565794, + "val_right_elbow_mae": 0.026529215276241302, + "val_right_elbow_r2": 0.9260286688804626, + "val_left_hand_mse": 0.001792966853827238, + "val_left_hand_rmse": 0.04234343932449557, + "val_left_hand_mae": 0.033128444105386734, + "val_left_hand_r2": 0.9343199729919434, + "val_right_hand_mse": 0.002173864748328924, + "val_right_hand_rmse": 0.04662472250136106, + "val_right_hand_mae": 0.03474586457014084, + "val_right_hand_r2": 0.9269508719444275, + "val_left_hip_mse": 0.0008604403119534254, + "val_left_hip_rmse": 0.029333262893060933, + "val_left_hip_mae": 0.021409068256616592, + "val_left_hip_r2": 0.8917308449745178, + "val_right_hip_mse": 0.0007937807240523398, + "val_right_hip_rmse": 0.028174114432442058, + "val_right_hip_mae": 0.02069556526839733, + "val_right_hip_r2": 0.9025007486343384, + "val_left_knee_mse": 0.001011873479001224, + "val_left_knee_rmse": 0.031809958802256004, + "val_left_knee_mae": 0.02213096246123314, + "val_left_knee_r2": 0.8964857459068298, + "val_right_knee_mse": 0.000833040161523968, + "val_right_knee_rmse": 0.028862435128103242, + "val_right_knee_mae": 0.020666297525167465, + "val_right_knee_r2": 0.9242597222328186, + "val_left_foot_mse": 0.0015569725073873997, + "val_left_foot_rmse": 0.03945849094158822, + "val_left_foot_mae": 0.022341586649417877, + "val_left_foot_r2": 0.6757157444953918, + "val_right_foot_mse": 0.0011345583479851484, + "val_right_foot_rmse": 0.03368320572607584, + "val_right_foot_mae": 0.022527365013957024, + "val_right_foot_r2": 0.755253255367279 + }, + "best_epoch": 89 + }, + { + "fold": 4, + "train_metrics": { + "train_mse": 0.0007933024899102747, + "train_rmse": 0.028165626034410716, + "train_mae": 0.020870866253972054, + "train_r2": 0.9627386331558228, + "train_head_mse": 0.0008428292348980904, + "train_head_rmse": 0.02903152140171249, + "train_head_mae": 0.021816473454236984, + "train_head_r2": 0.9153656959533691, + "train_left_shoulder_mse": 0.0005592500674538314, + "train_left_shoulder_rmse": 0.023648468606948556, + "train_left_shoulder_mae": 0.01807972602546215, + "train_left_shoulder_r2": 0.8999382853507996, + "train_left_elbow_mse": 0.0008359081693924963, + "train_left_elbow_rmse": 0.02891207653200469, + "train_left_elbow_mae": 0.0221111997961998, + "train_left_elbow_r2": 0.9394396543502808, + "train_right_shoulder_mse": 0.00047079293290153146, + "train_right_shoulder_rmse": 0.021697763315639967, + "train_right_shoulder_mae": 0.016770420596003532, + "train_right_shoulder_r2": 0.9280612468719482, + "train_right_elbow_mse": 0.00079375907080248, + "train_right_elbow_rmse": 0.028173730154214226, + "train_right_elbow_mae": 0.021869834512472153, + "train_right_elbow_r2": 0.9496280550956726, + "train_left_hand_mse": 0.00117184326518327, + "train_left_hand_rmse": 0.03423219632426862, + "train_left_hand_mae": 0.027093300595879555, + "train_left_hand_r2": 0.9556779861450195, + "train_right_hand_mse": 0.0013671162305399776, + "train_right_hand_rmse": 0.03697453489281478, + "train_right_hand_mae": 0.028316371142864227, + "train_right_hand_r2": 0.9511061906814575, + "train_left_hip_mse": 0.0006155352457426488, + "train_left_hip_rmse": 0.02480998278400549, + "train_left_hip_mae": 0.01905142329633236, + "train_left_hip_r2": 0.9216898679733276, + "train_right_hip_mse": 0.0005935825174674392, + "train_right_hip_rmse": 0.024363548950582695, + "train_right_hip_mae": 0.018538430333137512, + "train_right_hip_r2": 0.9274568557739258, + "train_left_knee_mse": 0.0006441775476559997, + "train_left_knee_rmse": 0.02538065301870698, + "train_left_knee_mae": 0.0187475997954607, + "train_left_knee_r2": 0.9333781003952026, + "train_right_knee_mse": 0.0006374465301632881, + "train_right_knee_rmse": 0.025247703463152606, + "train_right_knee_mae": 0.01896589994430542, + "train_right_knee_r2": 0.9419487714767456, + "train_left_foot_mse": 0.0009076694259420037, + "train_left_foot_rmse": 0.030127552604584457, + "train_left_foot_mae": 0.019394926726818085, + "train_left_foot_r2": 0.7868647575378418, + "train_right_foot_mse": 0.0008730217232368886, + "train_right_foot_rmse": 0.02954694101318931, + "train_right_foot_mae": 0.020565679296851158, + "train_right_foot_r2": 0.8083672523498535 + }, + "val_metrics": { + "val_mse": 0.0011256338329985738, + "val_rmse": 0.03355046695649069, + "val_mae": 0.02342354878783226, + "val_r2": 0.9451853036880493, + "val_head_mse": 0.000988956424407661, + "val_head_rmse": 0.03144767756778966, + "val_head_mae": 0.02357901632785797, + "val_head_r2": 0.9002310037612915, + "val_left_shoulder_mse": 0.000675060844514519, + "val_left_shoulder_rmse": 0.025981933040374786, + "val_left_shoulder_mae": 0.01972043141722679, + "val_left_shoulder_r2": 0.8758242726325989, + "val_left_elbow_mse": 0.0011019764933735132, + "val_left_elbow_rmse": 0.03319603128950076, + "val_left_elbow_mae": 0.025070562958717346, + "val_left_elbow_r2": 0.920203447341919, + "val_right_shoulder_mse": 0.0005987840704619884, + "val_right_shoulder_rmse": 0.024470064782545806, + "val_right_shoulder_mae": 0.018649572506546974, + "val_right_shoulder_r2": 0.9070183634757996, + "val_right_elbow_mse": 0.0011634068796411157, + "val_right_elbow_rmse": 0.03410875077807916, + "val_right_elbow_mae": 0.02476770617067814, + "val_right_elbow_r2": 0.9264249801635742, + "val_left_hand_mse": 0.001725809066556394, + "val_left_hand_rmse": 0.04154285818954197, + "val_left_hand_mae": 0.031430479139089584, + "val_left_hand_r2": 0.9327197074890137, + "val_right_hand_mse": 0.0018714598845690489, + "val_right_hand_rmse": 0.043260373144126354, + "val_right_hand_mae": 0.03190559893846512, + "val_right_hand_r2": 0.9315565824508667, + "val_left_hip_mse": 0.0007154585910029709, + "val_left_hip_rmse": 0.02674805770524228, + "val_left_hip_mae": 0.020876558497548103, + "val_left_hip_r2": 0.9051631689071655, + "val_right_hip_mse": 0.0007060007774271071, + "val_right_hip_rmse": 0.02657067514059639, + "val_right_hip_mae": 0.020527075976133347, + "val_right_hip_r2": 0.909986138343811, + "val_left_knee_mse": 0.0009157732711173594, + "val_left_knee_rmse": 0.030261746002459268, + "val_left_knee_mae": 0.021034972742199898, + "val_left_knee_r2": 0.8954100608825684, + "val_right_knee_mse": 0.0008926497539505363, + "val_right_knee_rmse": 0.029877244751659017, + "val_right_knee_mae": 0.020473483949899673, + "val_right_knee_r2": 0.9094328880310059, + "val_left_foot_mse": 0.0015888606430962682, + "val_left_foot_rmse": 0.039860514837320754, + "val_left_foot_mae": 0.02261955477297306, + "val_left_foot_r2": 0.670756459236145, + "val_right_foot_mse": 0.001689043128862977, + "val_right_foot_rmse": 0.041097969887367636, + "val_right_foot_mae": 0.02385113388299942, + "val_right_foot_r2": 0.5977623462677002 + }, + "best_epoch": 100 + }, + { + "fold": 5, + "train_metrics": { + "train_mse": 0.0007982559618540108, + "train_rmse": 0.02825342389612294, + "train_mae": 0.020788714289665222, + "train_r2": 0.9624242782592773, + "train_head_mse": 0.0008406761335209012, + "train_head_rmse": 0.02899441555749833, + "train_head_mae": 0.021684953942894936, + "train_head_r2": 0.9165359735488892, + "train_left_shoulder_mse": 0.0005294555448926985, + "train_left_shoulder_rmse": 0.02300990101875057, + "train_left_shoulder_mae": 0.017545733600854874, + "train_left_shoulder_r2": 0.905056893825531, + "train_left_elbow_mse": 0.0007986771524883807, + "train_left_elbow_rmse": 0.028260876711248373, + "train_left_elbow_mae": 0.022056497633457184, + "train_left_elbow_r2": 0.9423443675041199, + "train_right_shoulder_mse": 0.0004483312659431249, + "train_right_shoulder_rmse": 0.021173834464808798, + "train_right_shoulder_mae": 0.016348902136087418, + "train_right_shoulder_r2": 0.9322090744972229, + "train_right_elbow_mse": 0.0007314754766412079, + "train_right_elbow_rmse": 0.027045803309223557, + "train_right_elbow_mae": 0.020811911672353745, + "train_right_elbow_r2": 0.9543188810348511, + "train_left_hand_mse": 0.0012149275280535221, + "train_left_hand_rmse": 0.034855810535024456, + "train_left_hand_mae": 0.027674203738570213, + "train_left_hand_r2": 0.9541513323783875, + "train_right_hand_mse": 0.0013067504623904824, + "train_right_hand_rmse": 0.0361490036154592, + "train_right_hand_mae": 0.027459818869829178, + "train_right_hand_r2": 0.9537713527679443, + "train_left_hip_mse": 0.0006502747419290245, + "train_left_hip_rmse": 0.025500485131248474, + "train_left_hip_mae": 0.019416451454162598, + "train_left_hip_r2": 0.9168345332145691, + "train_right_hip_mse": 0.0006306007853709161, + "train_right_hip_rmse": 0.02511176587520113, + "train_right_hip_mae": 0.01905788853764534, + "train_right_hip_r2": 0.9228211641311646, + "train_left_knee_mse": 0.0006674281321465969, + "train_left_knee_rmse": 0.025834630482098962, + "train_left_knee_mae": 0.01887526921927929, + "train_left_knee_r2": 0.9297091364860535, + "train_right_knee_mse": 0.0007136240601539612, + "train_right_knee_rmse": 0.026713742907985793, + "train_right_knee_mae": 0.01957273855805397, + "train_right_knee_r2": 0.9325770735740662, + "train_left_foot_mse": 0.000959237280767411, + "train_left_foot_rmse": 0.030971555995258148, + "train_left_foot_mae": 0.01985659822821617, + "train_left_foot_r2": 0.7924470901489258, + "train_right_foot_mse": 0.0008858690271154046, + "train_right_foot_rmse": 0.029763551990906672, + "train_right_foot_mae": 0.019892318174242973, + "train_right_foot_r2": 0.7992576360702515 + }, + "val_metrics": { + "val_mse": 0.0010469314875081182, + "val_rmse": 0.03235632067321806, + "val_mae": 0.023304378613829613, + "val_r2": 0.9494883418083191, + "val_head_mse": 0.000978772877715528, + "val_head_rmse": 0.031285346053951966, + "val_head_mae": 0.023389529436826706, + "val_head_r2": 0.8965049982070923, + "val_left_shoulder_mse": 0.0006388405454345047, + "val_left_shoulder_rmse": 0.025275295160185663, + "val_left_shoulder_mae": 0.01933615282177925, + "val_left_shoulder_r2": 0.8836531043052673, + "val_left_elbow_mse": 0.0010001457994803786, + "val_left_elbow_rmse": 0.03162508180986064, + "val_left_elbow_mae": 0.02438472770154476, + "val_left_elbow_r2": 0.9266146421432495, + "val_right_shoulder_mse": 0.0006053505348972976, + "val_right_shoulder_rmse": 0.024603872355734933, + "val_right_shoulder_mae": 0.018647540360689163, + "val_right_shoulder_r2": 0.9018383622169495, + "val_right_elbow_mse": 0.0010383946355432272, + "val_right_elbow_rmse": 0.03222413126126486, + "val_right_elbow_mae": 0.024266762658953667, + "val_right_elbow_r2": 0.92996746301651, + "val_left_hand_mse": 0.0016775435069575906, + "val_left_hand_rmse": 0.04095782595496971, + "val_left_hand_mae": 0.031596701592206955, + "val_left_hand_r2": 0.9340806007385254, + "val_right_hand_mse": 0.0019732231739908457, + "val_right_hand_rmse": 0.04442097673386804, + "val_right_hand_mae": 0.032693225890398026, + "val_right_hand_r2": 0.9246758222579956, + "val_left_hip_mse": 0.0009197260369546711, + "val_left_hip_rmse": 0.030326985292881834, + "val_left_hip_mae": 0.02216867171227932, + "val_left_hip_r2": 0.8807257413864136, + "val_right_hip_mse": 0.0009195457678288221, + "val_right_hip_rmse": 0.030324013056137904, + "val_right_hip_mae": 0.02206946350634098, + "val_right_hip_r2": 0.8834879398345947, + "val_left_knee_mse": 0.0007853845600038767, + "val_left_knee_rmse": 0.02802471337951339, + "val_left_knee_mae": 0.021058792248368263, + "val_left_knee_r2": 0.9169187545776367, + "val_right_knee_mse": 0.0010977138299494982, + "val_right_knee_rmse": 0.03313176466700043, + "val_right_knee_mae": 0.023357650265097618, + "val_right_knee_r2": 0.9041147828102112, + "val_left_foot_mse": 0.0006334662903100252, + "val_left_foot_rmse": 0.0251687562328778, + "val_left_foot_mae": 0.01858622021973133, + "val_left_foot_r2": 0.8124920725822449, + "val_right_foot_mse": 0.0013420007890090346, + "val_right_foot_rmse": 0.03663332893703539, + "val_right_foot_mae": 0.021401487290859222, + "val_right_foot_r2": 0.7178030610084534 + }, + "best_epoch": 100 + } + ], + "test_metrics": { + "test_mse": 0.0010053022997453809, + "test_rmse": 0.03170650248364491, + "test_mae": 0.02348632737994194, + "test_r2": 0.9528595805168152, + "test_head_mse": 0.001045786659233272, + "test_head_rmse": 0.0323386248816067, + "test_head_mae": 0.024737700819969177, + "test_head_r2": 0.897356390953064, + "test_left_shoulder_mse": 0.0006564909708686173, + "test_left_shoulder_rmse": 0.025622079752990724, + "test_left_shoulder_mae": 0.019473403692245483, + "test_left_shoulder_r2": 0.8848369717597961, + "test_left_elbow_mse": 0.0009413087391294539, + "test_left_elbow_rmse": 0.03068075519164178, + "test_left_elbow_mae": 0.023488754406571388, + "test_left_elbow_r2": 0.9305267333984375, + "test_right_shoulder_mse": 0.0007046430255286396, + "test_right_shoulder_rmse": 0.026545113025350627, + "test_right_shoulder_mae": 0.019808512181043625, + "test_right_shoulder_r2": 0.8941569328308105, + "test_right_elbow_mse": 0.0011103218421339989, + "test_right_elbow_rmse": 0.03332149219548847, + "test_right_elbow_mae": 0.025394394993782043, + "test_right_elbow_r2": 0.9289836883544922, + "test_left_hand_mse": 0.0013964829267933965, + "test_left_hand_rmse": 0.03736954544536763, + "test_left_hand_mae": 0.02890772558748722, + "test_left_hand_r2": 0.9466777443885803, + "test_right_hand_mse": 0.00182151363696903, + "test_right_hand_rmse": 0.042679194427367416, + "test_right_hand_mae": 0.03280480578541756, + "test_right_hand_r2": 0.9352646470069885, + "test_left_hip_mse": 0.0008462700061500072, + "test_left_hip_rmse": 0.02909072027554504, + "test_left_hip_mae": 0.02208428643643856, + "test_left_hip_r2": 0.9039833545684814, + "test_right_hip_mse": 0.0008784330566413701, + "test_right_hip_rmse": 0.02963837135608787, + "test_right_hip_mae": 0.0219794400036335, + "test_right_hip_r2": 0.9041385650634766, + "test_left_knee_mse": 0.0008693316485732794, + "test_left_knee_rmse": 0.029484430613007935, + "test_left_knee_mae": 0.021937692537903786, + "test_left_knee_r2": 0.9106469750404358, + "test_right_knee_mse": 0.0008832779130898416, + "test_right_knee_rmse": 0.029719991808374403, + "test_right_knee_mae": 0.022207044064998627, + "test_right_knee_r2": 0.9176310896873474, + "test_left_foot_mse": 0.0008368164417333901, + "test_left_foot_rmse": 0.02892777975810432, + "test_left_foot_mae": 0.020607056096196175, + "test_left_foot_r2": 0.8200623393058777, + "test_right_foot_mse": 0.0010782527970150113, + "test_right_foot_rmse": 0.032836759843428695, + "test_right_foot_mae": 0.021891439333558083, + "test_right_foot_r2": 0.7946063876152039 + } +} \ No newline at end of file diff --git a/A9/cv_results_loss_optimizer/conv1d_v3_adam_mse_fold1_best.h5 b/A9/cv_results_loss_optimizer/conv1d_v3_adam_mse_fold1_best.h5 new file mode 100644 index 0000000000000000000000000000000000000000..90e29abe52f69f1e78c8023bf151d2804e73ed04 --- /dev/null +++ b/A9/cv_results_loss_optimizer/conv1d_v3_adam_mse_fold1_best.h5 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:d4c9389dedea5d07fd44a1bea364207fe44841cd5ef0e376110a14995042691b +size 2658520 diff --git a/A9/cv_results_loss_optimizer/conv1d_v3_adam_mse_fold2_best.h5 b/A9/cv_results_loss_optimizer/conv1d_v3_adam_mse_fold2_best.h5 new file mode 100644 index 0000000000000000000000000000000000000000..67a1ece73ecf947b755f5559b3946b69a4902993 --- /dev/null +++ b/A9/cv_results_loss_optimizer/conv1d_v3_adam_mse_fold2_best.h5 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:8dd1cffa71a44e3ca320a99c800683ff60725b4aa4c713fb2f0ddef5b897ad95 +size 2658520 diff --git a/A9/cv_results_loss_optimizer/conv1d_v3_adam_mse_fold3_best.h5 b/A9/cv_results_loss_optimizer/conv1d_v3_adam_mse_fold3_best.h5 new file mode 100644 index 0000000000000000000000000000000000000000..7de14ce34f5be9a2b8d71a076de1a018a6b85433 --- /dev/null +++ b/A9/cv_results_loss_optimizer/conv1d_v3_adam_mse_fold3_best.h5 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7824772e5c1faf34544d5cb9a8518c3808cf6348cc71e80f1251da4cb89b0afc +size 2658520 diff --git a/A9/cv_results_loss_optimizer/conv1d_v3_adam_mse_fold4_best.h5 b/A9/cv_results_loss_optimizer/conv1d_v3_adam_mse_fold4_best.h5 new file mode 100644 index 0000000000000000000000000000000000000000..2b5cf3cdd52102654efb99139b0b33b647036448 --- /dev/null +++ b/A9/cv_results_loss_optimizer/conv1d_v3_adam_mse_fold4_best.h5 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:1a093463228a0d9d95a8bf8658e90fa3823bf24ac70c118d57f2808199fc115e +size 2658520 diff --git a/A9/cv_results_loss_optimizer/conv1d_v3_adam_mse_fold5_best.h5 b/A9/cv_results_loss_optimizer/conv1d_v3_adam_mse_fold5_best.h5 new file mode 100644 index 0000000000000000000000000000000000000000..5cbefd0e4ea013bbc5469cf1f3dd03477a74eace --- /dev/null +++ b/A9/cv_results_loss_optimizer/conv1d_v3_adam_mse_fold5_best.h5 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:1e1c7228f8b69c225b7da17adb196382529e4cbeecd31734f7df3e566217fe4a +size 2658520 diff --git a/A9/cv_results_loss_optimizer/conv1d_v3_adam_mse_predictions.png b/A9/cv_results_loss_optimizer/conv1d_v3_adam_mse_predictions.png new file mode 100644 index 0000000000000000000000000000000000000000..02fc548922c5a76ddf40fb8085dbfc5357a8af48 --- /dev/null +++ b/A9/cv_results_loss_optimizer/conv1d_v3_adam_mse_predictions.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7dd982c73c3e9a323e07c8f851c3ea40e3429d27eaf331b8364310c768c34050 +size 550898 diff --git a/A9/cv_results_loss_optimizer/conv1d_v3_adam_mse_test_predictions.png b/A9/cv_results_loss_optimizer/conv1d_v3_adam_mse_test_predictions.png new file mode 100644 index 0000000000000000000000000000000000000000..3f8283d0d3a32c4e1abbf2ca81dc22a70e82b1d4 --- /dev/null +++ b/A9/cv_results_loss_optimizer/conv1d_v3_adam_mse_test_predictions.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:011a12676a4d6151135629a23adc15105440d4bfefdab534602660a28bc33843 +size 467758 diff --git a/A9/cv_results_loss_optimizer/conv1d_v3_adam_mse_training_history.png b/A9/cv_results_loss_optimizer/conv1d_v3_adam_mse_training_history.png new file mode 100644 index 0000000000000000000000000000000000000000..bb9c323b482ad959e24a5666d1afae4699cef927 --- /dev/null +++ b/A9/cv_results_loss_optimizer/conv1d_v3_adam_mse_training_history.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:9da2834bbbc0b21dcf44065b1c358a28747d045f2813b3e6fbc326e032fc7783 +size 368237 diff --git a/A9/cv_results_loss_optimizer/conv1d_v3_rmsprop_mae_cv_results.json b/A9/cv_results_loss_optimizer/conv1d_v3_rmsprop_mae_cv_results.json new file mode 100644 index 0000000000000000000000000000000000000000..c556f5675eb0395b4aa18c0ee49a32ef76146323 --- /dev/null +++ b/A9/cv_results_loss_optimizer/conv1d_v3_rmsprop_mae_cv_results.json @@ -0,0 +1,1243 @@ +{ + "model_name": "conv1d_v3_rmsprop_mae", + "n_splits": 5, + "epochs": 100, + "batch_size": 32, + "patience": 10, + "aggregate_results": { + "n_folds": 5, + "train_metrics": { + "train_mse": { + "mean": 0.0009206908871419728, + "std": 3.615028597513794e-05 + }, + "train_rmse": { + "mean": 0.03033709586464539, + "std": 0.0005928757384882228 + }, + "train_mae": { + "mean": 0.020942456647753717, + "std": 0.0002982157412235282 + }, + "train_r2": { + "mean": 0.9564496397972106, + "std": 0.0016354589357046232 + }, + "train_head_mse": { + "mean": 0.0008362963097169995, + "std": 5.923022668974774e-05 + }, + "train_head_rmse": { + "mean": 0.028900155228039636, + "std": 0.001037948704037256 + }, + "train_head_mae": { + "mean": 0.021307482197880746, + "std": 0.0006335075361347412 + }, + "train_head_r2": { + "mean": 0.9159852862358093, + "std": 0.005222537604169388 + }, + "train_left_shoulder_mse": { + "mean": 0.0005473337369039655, + "std": 1.884065525978251e-05 + }, + "train_left_shoulder_rmse": { + "mean": 0.023391646374229514, + "std": 0.00040572996803399115 + }, + "train_left_shoulder_mae": { + "mean": 0.017211272194981576, + "std": 0.00029856900225332084 + }, + "train_left_shoulder_r2": { + "mean": 0.9015494108200073, + "std": 0.0030007426553122136 + }, + "train_left_elbow_mse": { + "mean": 0.0008903976762667298, + "std": 6.708339782199921e-05 + }, + "train_left_elbow_rmse": { + "mean": 0.02981843968405219, + "std": 0.0011217535715438902 + }, + "train_left_elbow_mae": { + "mean": 0.022223565727472305, + "std": 0.0008061349797119755 + }, + "train_left_elbow_r2": { + "mean": 0.9355279207229614, + "std": 0.004597560148555312 + }, + "train_right_shoulder_mse": { + "mean": 0.0004762952332384884, + "std": 1.67906081318233e-05 + }, + "train_right_shoulder_rmse": { + "mean": 0.021820849153103873, + "std": 0.0003818055473331052 + }, + "train_right_shoulder_mae": { + "mean": 0.016313813999295233, + "std": 0.00018438739944256257 + }, + "train_right_shoulder_r2": { + "mean": 0.9269890546798706, + "std": 0.002492210268487843 + }, + "train_right_elbow_mse": { + "mean": 0.0009255822282284498, + "std": 3.633677919414003e-05 + }, + "train_right_elbow_rmse": { + "mean": 0.030417563578586982, + "std": 0.0005950245129980798 + }, + "train_right_elbow_mae": { + "mean": 0.022208529710769653, + "std": 0.00036665353014056886 + }, + "train_right_elbow_r2": { + "mean": 0.9413130879402161, + "std": 0.0025166930366086002 + }, + "train_left_hand_mse": { + "mean": 0.0013688717503100634, + "std": 6.686010932493916e-05 + }, + "train_left_hand_rmse": { + "mean": 0.03698739172283083, + "std": 0.0008969971303906569 + }, + "train_left_hand_mae": { + "mean": 0.028276403620839118, + "std": 0.0007198080824229456 + }, + "train_left_hand_r2": { + "mean": 0.9479272723197937, + "std": 0.0024683159047710445 + }, + "train_right_hand_mse": { + "mean": 0.0017338651232421398, + "std": 9.382043336332914e-05 + }, + "train_right_hand_rmse": { + "mean": 0.04162433859567777, + "std": 0.0011311762526283315 + }, + "train_right_hand_mae": { + "mean": 0.029975622519850732, + "std": 0.0005388508361448516 + }, + "train_right_hand_r2": { + "mean": 0.9377394556999207, + "std": 0.0034108679651967894 + }, + "train_left_hip_mse": { + "mean": 0.0006590691045857966, + "std": 2.2143752760062025e-05 + }, + "train_left_hip_rmse": { + "mean": 0.025668694663433134, + "std": 0.0004326879490303692 + }, + "train_left_hip_mae": { + "mean": 0.01910896860063076, + "std": 0.00043664060284864646 + }, + "train_left_hip_r2": { + "mean": 0.9154826402664185, + "std": 0.0020965581216877247 + }, + "train_right_hip_mse": { + "mean": 0.000650475115980953, + "std": 2.4043116531931362e-05 + }, + "train_right_hip_rmse": { + "mean": 0.02550013881902159, + "std": 0.00046694345651411367 + }, + "train_right_hip_mae": { + "mean": 0.018941498547792434, + "std": 0.000558813598574222 + }, + "train_right_hip_r2": { + "mean": 0.9198240637779236, + "std": 0.003005769552976306 + }, + "train_left_knee_mse": { + "mean": 0.0007729625678621233, + "std": 6.535780070149381e-05 + }, + "train_left_knee_rmse": { + "mean": 0.02777749867335349, + "std": 0.0011718085654100672 + }, + "train_left_knee_mae": { + "mean": 0.019136035069823264, + "std": 0.0006602235115357534 + }, + "train_left_knee_r2": { + "mean": 0.9184973239898682, + "std": 0.007101581877407361 + }, + "train_right_knee_mse": { + "mean": 0.0008152929600328207, + "std": 7.284418173223433e-05 + }, + "train_right_knee_rmse": { + "mean": 0.028524720882349013, + "std": 0.001277990069172197 + }, + "train_right_knee_mae": { + "mean": 0.01922696866095066, + "std": 0.000451288485522082 + }, + "train_right_knee_r2": { + "mean": 0.9242021560668945, + "std": 0.006744460752416853 + }, + "train_left_foot_mse": { + "mean": 0.0011701066512614488, + "std": 0.00016084569881917848 + }, + "train_left_foot_rmse": { + "mean": 0.03412649315249415, + "std": 0.002342886334037221 + }, + "train_left_foot_mae": { + "mean": 0.019034701213240624, + "std": 0.0009645013489008851 + }, + "train_left_foot_r2": { + "mean": 0.7330427169799805, + "std": 0.030499224271650447 + }, + "train_right_foot_mse": { + "mean": 0.001122433179989457, + "std": 0.00010550135974700284 + }, + "train_right_foot_rmse": { + "mean": 0.03346672578026462, + "std": 0.0015528829762825578 + }, + "train_right_foot_mae": { + "mean": 0.01928706429898739, + "std": 0.0004585858712313208 + }, + "train_right_foot_r2": { + "mean": 0.7497075200080872, + "std": 0.022863198573460267 + } + }, + "val_metrics": { + "val_mse": { + "mean": 0.0012004459975287318, + "std": 8.743791041899788e-05 + }, + "val_rmse": { + "mean": 0.034623441558392365, + "std": 0.0012896868539766832 + }, + "val_mae": { + "mean": 0.023638730123639106, + "std": 0.0008009697416940648 + }, + "val_r2": { + "mean": 0.9431959390640259, + "std": 0.003857514093878985 + }, + "val_head_mse": { + "mean": 0.0010684540029615164, + "std": 6.186871457439652e-05 + }, + "val_head_rmse": { + "mean": 0.03267351976573415, + "std": 0.0009461020450756606 + }, + "val_head_mae": { + "mean": 0.0240256629884243, + "std": 0.0008338477954030818 + }, + "val_head_r2": { + "mean": 0.8923399090766907, + "std": 0.00703728154130753 + }, + "val_left_shoulder_mse": { + "mean": 0.0007099067559465766, + "std": 6.121197057347107e-05 + }, + "val_left_shoulder_rmse": { + "mean": 0.02661952903668343, + "std": 0.0011434289710096678 + }, + "val_left_shoulder_mae": { + "mean": 0.019455020129680634, + "std": 0.0008339399806206126 + }, + "val_left_shoulder_r2": { + "mean": 0.8722980976104736, + "std": 0.0068256424776896125 + }, + "val_left_elbow_mse": { + "mean": 0.0011511969147250055, + "std": 7.668260401356469e-05 + }, + "val_left_elbow_rmse": { + "mean": 0.03390981261960042, + "std": 0.0011495750643582612 + }, + "val_left_elbow_mae": { + "mean": 0.025299175456166268, + "std": 0.0009294430404566033 + }, + "val_left_elbow_r2": { + "mean": 0.9165043473243714, + "std": 0.005497131136710882 + }, + "val_right_shoulder_mse": { + "mean": 0.0006496857851743699, + "std": 5.755260803797681e-05 + }, + "val_right_shoulder_rmse": { + "mean": 0.025464108321712607, + "std": 0.0011247099868217164 + }, + "val_right_shoulder_mae": { + "mean": 0.018786928430199624, + "std": 0.0007150953047113064 + }, + "val_right_shoulder_r2": { + "mean": 0.900288724899292, + "std": 0.00799400164779036 + }, + "val_right_elbow_mse": { + "mean": 0.0013069996144622563, + "std": 0.0001308978660361773 + }, + "val_right_elbow_rmse": { + "mean": 0.036105254371378064, + "std": 0.001846678434471352 + }, + "val_right_elbow_mae": { + "mean": 0.025786950066685677, + "std": 0.0012381707710350611 + }, + "val_right_elbow_r2": { + "mean": 0.9171055436134339, + "std": 0.007074607377599327 + }, + "val_left_hand_mse": { + "mean": 0.0018339209025725722, + "std": 0.0001592735418256824 + }, + "val_left_hand_rmse": { + "mean": 0.04278130119482852, + "std": 0.0019186377067952063 + }, + "val_left_hand_mae": { + "mean": 0.03239019326865673, + "std": 0.0013026996092340298 + }, + "val_left_hand_r2": { + "mean": 0.9299574375152588, + "std": 0.007526929399587549 + }, + "val_right_hand_mse": { + "mean": 0.0023565222974866627, + "std": 0.0002724585370486451 + }, + "val_right_hand_rmse": { + "mean": 0.048459885886620244, + "std": 0.0028568789513044924 + }, + "val_right_hand_mae": { + "mean": 0.03409670665860176, + "std": 0.0019771827251433894 + }, + "val_right_hand_r2": { + "mean": 0.9152791976928711, + "std": 0.0092438641795481 + }, + "val_left_hip_mse": { + "mean": 0.0008484723512083292, + "std": 8.846362972249359e-05 + }, + "val_left_hip_rmse": { + "mean": 0.029089279640932147, + "std": 0.0015120056811995723 + }, + "val_left_hip_mae": { + "mean": 0.0211775578558445, + "std": 0.0009735528998174647 + }, + "val_left_hip_r2": { + "mean": 0.8907995462417603, + "std": 0.011662127687666472 + }, + "val_right_hip_mse": { + "mean": 0.0008437502547167242, + "std": 7.077726104525985e-05 + }, + "val_right_hip_rmse": { + "mean": 0.029021059234895463, + "std": 0.0012362748890975058 + }, + "val_right_hip_mae": { + "mean": 0.021117371320724488, + "std": 0.0009816131094361313 + }, + "val_right_hip_r2": { + "mean": 0.8957228899002075, + "std": 0.00826771573158941 + }, + "val_left_knee_mse": { + "mean": 0.0009724668576382101, + "std": 4.940447995774943e-05 + }, + "val_left_knee_rmse": { + "mean": 0.031174281038258127, + "std": 0.000794392463401965 + }, + "val_left_knee_mae": { + "mean": 0.0215570617467165, + "std": 0.0008594643978702686 + }, + "val_left_knee_r2": { + "mean": 0.8972449421882629, + "std": 0.005647094962209474 + }, + "val_right_knee_mse": { + "mean": 0.0010995324235409498, + "std": 0.0003196793695279386 + }, + "val_right_knee_rmse": { + "mean": 0.032825336178787, + "std": 0.004693583736396613 + }, + "val_right_knee_mae": { + "mean": 0.021864574030041696, + "std": 0.0019059560918154037 + }, + "val_right_knee_r2": { + "mean": 0.89881272315979, + "std": 0.024850543610111075 + }, + "val_left_foot_mse": { + "mean": 0.0013409191509708763, + "std": 0.0003427913069188681 + }, + "val_left_foot_rmse": { + "mean": 0.036324993290821825, + "std": 0.004627527784100861 + }, + "val_left_foot_mae": { + "mean": 0.020553989335894585, + "std": 0.00121515759266364 + }, + "val_left_foot_r2": { + "mean": 0.6959450006484985, + "std": 0.05123716820489879 + }, + "val_right_foot_mse": { + "mean": 0.0014239709125831723, + "std": 0.00035403522959816795 + }, + "val_right_foot_rmse": { + "mean": 0.037481086731307664, + "std": 0.004374819998967435 + }, + "val_right_foot_mae": { + "mean": 0.021192300692200662, + "std": 0.0005895036619545261 + }, + "val_right_foot_r2": { + "mean": 0.6836122989654541, + "std": 0.0674151125790412 + } + }, + "best_fold": "1", + "best_fold_metrics": { + "fold": 1, + "train_metrics": { + "train_mse": 0.0008825858822092414, + "train_rmse": 0.0297083470124011, + "train_mae": 0.020710114389657974, + "train_r2": 0.9585390686988831, + "train_head_mse": 0.0009037221898324788, + "train_head_rmse": 0.030061972487388094, + "train_head_mae": 0.021911976858973503, + "train_head_r2": 0.9104927182197571, + "train_left_shoulder_mse": 0.0005428819567896426, + "train_left_shoulder_rmse": 0.02329982739828007, + "train_left_shoulder_mae": 0.017068706452846527, + "train_left_shoulder_r2": 0.9039038419723511, + "train_left_elbow_mse": 0.000811553152743727, + "train_left_elbow_rmse": 0.028487771986305404, + "train_left_elbow_mae": 0.02100607380270958, + "train_left_elbow_r2": 0.9411996603012085, + "train_right_shoulder_mse": 0.0005061260308139026, + "train_right_shoulder_rmse": 0.02249724496052578, + "train_right_shoulder_mae": 0.016533508896827698, + "train_right_shoulder_r2": 0.923406720161438, + "train_right_elbow_mse": 0.0008782222867012024, + "train_right_elbow_rmse": 0.029634815449082898, + "train_right_elbow_mae": 0.02148911915719509, + "train_right_elbow_r2": 0.944831371307373, + "train_left_hand_mse": 0.0013044702354818583, + "train_left_hand_rmse": 0.03611745056730691, + "train_left_hand_mae": 0.027253253385424614, + "train_left_hand_r2": 0.9499377608299255, + "train_right_hand_mse": 0.0015913662500679493, + "train_right_hand_rmse": 0.03989193214257677, + "train_right_hand_mae": 0.029037902131676674, + "train_right_hand_r2": 0.9430550932884216, + "train_left_hip_mse": 0.0006794539513066411, + "train_left_hip_rmse": 0.02606633751232883, + "train_left_hip_mae": 0.019465599209070206, + "train_left_hip_r2": 0.9150049090385437, + "train_right_hip_mse": 0.0006558697787113488, + "train_right_hip_rmse": 0.025609954679994044, + "train_right_hip_mae": 0.019084792584180832, + "train_right_hip_r2": 0.9209688305854797, + "train_left_knee_mse": 0.0007495898753404617, + "train_left_knee_rmse": 0.02737863903375151, + "train_left_knee_mae": 0.019129153341054916, + "train_left_knee_r2": 0.9215936064720154, + "train_right_knee_mse": 0.0007377680740319192, + "train_right_knee_rmse": 0.027161886422557607, + "train_right_knee_mae": 0.018852168694138527, + "train_right_knee_r2": 0.9313990473747253, + "train_left_foot_mse": 0.0010896617313846946, + "train_left_foot_rmse": 0.03301002471045265, + "train_left_foot_mae": 0.01852666586637497, + "train_left_foot_r2": 0.7526522874832153, + "train_right_foot_mse": 0.0010229311883449554, + "train_right_foot_rmse": 0.03198329545786293, + "train_right_foot_mae": 0.019872542470693588, + "train_right_foot_r2": 0.7695533633232117 + }, + "val_metrics": { + "val_mse": 0.0010353884426876903, + "val_rmse": 0.03217745239585773, + "val_mae": 0.02242676168680191, + "val_r2": 0.9496020674705505, + "val_head_mse": 0.0009844066808000207, + "val_head_rmse": 0.03137525586827972, + "val_head_mae": 0.022794252261519432, + "val_head_r2": 0.8945198059082031, + "val_left_shoulder_mse": 0.000633683695923537, + "val_left_shoulder_rmse": 0.025173074820600223, + "val_left_shoulder_mae": 0.01837519183754921, + "val_left_shoulder_r2": 0.8777901530265808, + "val_left_elbow_mse": 0.001010555773973465, + "val_left_elbow_rmse": 0.031789239908709126, + "val_left_elbow_mae": 0.02358384057879448, + "val_left_elbow_r2": 0.9267594814300537, + "val_right_shoulder_mse": 0.0005801356746815145, + "val_right_shoulder_rmse": 0.024086005785134125, + "val_right_shoulder_mae": 0.017741873860359192, + "val_right_shoulder_r2": 0.9061342477798462, + "val_right_elbow_mse": 0.0010796069400385022, + "val_right_elbow_rmse": 0.03285737268922307, + "val_right_elbow_mae": 0.02370341494679451, + "val_right_elbow_r2": 0.928927481174469, + "val_left_hand_mse": 0.0015234581660479307, + "val_left_hand_rmse": 0.039031502226380306, + "val_left_hand_mae": 0.030279429629445076, + "val_left_hand_r2": 0.9438909292221069, + "val_right_hand_mse": 0.0018899973947554827, + "val_right_hand_rmse": 0.043474100275399406, + "val_right_hand_mae": 0.031543582677841187, + "val_right_hand_r2": 0.9311773180961609, + "val_left_hip_mse": 0.0007276084506884217, + "val_left_hip_rmse": 0.02697421825907883, + "val_left_hip_mae": 0.019707106053829193, + "val_left_hip_r2": 0.8962271213531494, + "val_right_hip_mse": 0.0007198604289442301, + "val_right_hip_rmse": 0.026830214850877176, + "val_right_hip_mae": 0.019679415971040726, + "val_right_hip_r2": 0.902458667755127, + "val_left_knee_mse": 0.0008989588241092861, + "val_left_knee_rmse": 0.029982642046845805, + "val_left_knee_mae": 0.020985377952456474, + "val_left_knee_r2": 0.90211021900177, + "val_right_knee_mse": 0.0009380837436765432, + "val_right_knee_rmse": 0.030628152795696695, + "val_right_knee_mae": 0.021724969148635864, + "val_right_knee_r2": 0.9127605557441711, + "val_left_foot_mse": 0.001230311580002308, + "val_left_foot_rmse": 0.03507579763886073, + "val_left_foot_mae": 0.020148754119873047, + "val_left_foot_r2": 0.7102408409118652, + "val_right_foot_mse": 0.0012433826923370361, + "val_right_foot_rmse": 0.035261632014656326, + "val_right_foot_mae": 0.02128068543970585, + "val_right_foot_r2": 0.7335558533668518 + }, + "best_epoch": 100 + } + }, + "fold_details": [ + { + "fold": 1, + "train_metrics": { + "train_mse": 0.0008825858822092414, + "train_rmse": 0.0297083470124011, + "train_mae": 0.020710114389657974, + "train_r2": 0.9585390686988831, + "train_head_mse": 0.0009037221898324788, + "train_head_rmse": 0.030061972487388094, + "train_head_mae": 0.021911976858973503, + "train_head_r2": 0.9104927182197571, + "train_left_shoulder_mse": 0.0005428819567896426, + "train_left_shoulder_rmse": 0.02329982739828007, + "train_left_shoulder_mae": 0.017068706452846527, + "train_left_shoulder_r2": 0.9039038419723511, + "train_left_elbow_mse": 0.000811553152743727, + "train_left_elbow_rmse": 0.028487771986305404, + "train_left_elbow_mae": 0.02100607380270958, + "train_left_elbow_r2": 0.9411996603012085, + "train_right_shoulder_mse": 0.0005061260308139026, + "train_right_shoulder_rmse": 0.02249724496052578, + "train_right_shoulder_mae": 0.016533508896827698, + "train_right_shoulder_r2": 0.923406720161438, + "train_right_elbow_mse": 0.0008782222867012024, + "train_right_elbow_rmse": 0.029634815449082898, + "train_right_elbow_mae": 0.02148911915719509, + "train_right_elbow_r2": 0.944831371307373, + "train_left_hand_mse": 0.0013044702354818583, + "train_left_hand_rmse": 0.03611745056730691, + "train_left_hand_mae": 0.027253253385424614, + "train_left_hand_r2": 0.9499377608299255, + "train_right_hand_mse": 0.0015913662500679493, + "train_right_hand_rmse": 0.03989193214257677, + "train_right_hand_mae": 0.029037902131676674, + "train_right_hand_r2": 0.9430550932884216, + "train_left_hip_mse": 0.0006794539513066411, + "train_left_hip_rmse": 0.02606633751232883, + "train_left_hip_mae": 0.019465599209070206, + "train_left_hip_r2": 0.9150049090385437, + "train_right_hip_mse": 0.0006558697787113488, + "train_right_hip_rmse": 0.025609954679994044, + "train_right_hip_mae": 0.019084792584180832, + "train_right_hip_r2": 0.9209688305854797, + "train_left_knee_mse": 0.0007495898753404617, + "train_left_knee_rmse": 0.02737863903375151, + "train_left_knee_mae": 0.019129153341054916, + "train_left_knee_r2": 0.9215936064720154, + "train_right_knee_mse": 0.0007377680740319192, + "train_right_knee_rmse": 0.027161886422557607, + "train_right_knee_mae": 0.018852168694138527, + "train_right_knee_r2": 0.9313990473747253, + "train_left_foot_mse": 0.0010896617313846946, + "train_left_foot_rmse": 0.03301002471045265, + "train_left_foot_mae": 0.01852666586637497, + "train_left_foot_r2": 0.7526522874832153, + "train_right_foot_mse": 0.0010229311883449554, + "train_right_foot_rmse": 0.03198329545786293, + "train_right_foot_mae": 0.019872542470693588, + "train_right_foot_r2": 0.7695533633232117 + }, + "val_metrics": { + "val_mse": 0.0010353884426876903, + "val_rmse": 0.03217745239585773, + "val_mae": 0.02242676168680191, + "val_r2": 0.9496020674705505, + "val_head_mse": 0.0009844066808000207, + "val_head_rmse": 0.03137525586827972, + "val_head_mae": 0.022794252261519432, + "val_head_r2": 0.8945198059082031, + "val_left_shoulder_mse": 0.000633683695923537, + "val_left_shoulder_rmse": 0.025173074820600223, + "val_left_shoulder_mae": 0.01837519183754921, + "val_left_shoulder_r2": 0.8777901530265808, + "val_left_elbow_mse": 0.001010555773973465, + "val_left_elbow_rmse": 0.031789239908709126, + "val_left_elbow_mae": 0.02358384057879448, + "val_left_elbow_r2": 0.9267594814300537, + "val_right_shoulder_mse": 0.0005801356746815145, + "val_right_shoulder_rmse": 0.024086005785134125, + "val_right_shoulder_mae": 0.017741873860359192, + "val_right_shoulder_r2": 0.9061342477798462, + "val_right_elbow_mse": 0.0010796069400385022, + "val_right_elbow_rmse": 0.03285737268922307, + "val_right_elbow_mae": 0.02370341494679451, + "val_right_elbow_r2": 0.928927481174469, + "val_left_hand_mse": 0.0015234581660479307, + "val_left_hand_rmse": 0.039031502226380306, + "val_left_hand_mae": 0.030279429629445076, + "val_left_hand_r2": 0.9438909292221069, + "val_right_hand_mse": 0.0018899973947554827, + "val_right_hand_rmse": 0.043474100275399406, + "val_right_hand_mae": 0.031543582677841187, + "val_right_hand_r2": 0.9311773180961609, + "val_left_hip_mse": 0.0007276084506884217, + "val_left_hip_rmse": 0.02697421825907883, + "val_left_hip_mae": 0.019707106053829193, + "val_left_hip_r2": 0.8962271213531494, + "val_right_hip_mse": 0.0007198604289442301, + "val_right_hip_rmse": 0.026830214850877176, + "val_right_hip_mae": 0.019679415971040726, + "val_right_hip_r2": 0.902458667755127, + "val_left_knee_mse": 0.0008989588241092861, + "val_left_knee_rmse": 0.029982642046845805, + "val_left_knee_mae": 0.020985377952456474, + "val_left_knee_r2": 0.90211021900177, + "val_right_knee_mse": 0.0009380837436765432, + "val_right_knee_rmse": 0.030628152795696695, + "val_right_knee_mae": 0.021724969148635864, + "val_right_knee_r2": 0.9127605557441711, + "val_left_foot_mse": 0.001230311580002308, + "val_left_foot_rmse": 0.03507579763886073, + "val_left_foot_mae": 0.020148754119873047, + "val_left_foot_r2": 0.7102408409118652, + "val_right_foot_mse": 0.0012433826923370361, + "val_right_foot_rmse": 0.035261632014656326, + "val_right_foot_mae": 0.02128068543970585, + "val_right_foot_r2": 0.7335558533668518 + }, + "best_epoch": 100 + }, + { + "fold": 2, + "train_metrics": { + "train_mse": 0.000927658227737993, + "train_rmse": 0.030457482294798974, + "train_mae": 0.02080542966723442, + "train_r2": 0.9557715058326721, + "train_head_mse": 0.000811937206890434, + "train_head_rmse": 0.028494511873173648, + "train_head_mae": 0.02079087123274803, + "train_head_r2": 0.9169545769691467, + "train_left_shoulder_mse": 0.000567353330552578, + "train_left_shoulder_rmse": 0.023819179888329028, + "train_left_shoulder_mae": 0.01731901429593563, + "train_left_shoulder_r2": 0.897432267665863, + "train_left_elbow_mse": 0.0009537457372061908, + "train_left_elbow_rmse": 0.030882774117721206, + "train_left_elbow_mae": 0.02262716554105282, + "train_left_elbow_r2": 0.9313936233520508, + "train_right_shoulder_mse": 0.00048358453204855323, + "train_right_shoulder_rmse": 0.021990555519325865, + "train_right_shoulder_mae": 0.01625783182680607, + "train_right_shoulder_r2": 0.9246488809585571, + "train_right_elbow_mse": 0.0009846545290201902, + "train_right_elbow_rmse": 0.03137920535992252, + "train_right_elbow_mae": 0.022459521889686584, + "train_right_elbow_r2": 0.9370825290679932, + "train_left_hand_mse": 0.001344221062026918, + "train_left_hand_rmse": 0.036663620416250736, + "train_left_hand_mae": 0.028055815026164055, + "train_left_hand_r2": 0.9491057991981506, + "train_right_hand_mse": 0.0018411485943943262, + "train_right_hand_rmse": 0.042908607462772856, + "train_right_hand_mae": 0.030438877642154694, + "train_right_hand_r2": 0.9335576891899109, + "train_left_hip_mse": 0.0006269916775636375, + "train_left_hip_rmse": 0.02503980186749962, + "train_left_hip_mae": 0.018541380763053894, + "train_left_hip_r2": 0.916999340057373, + "train_right_hip_mse": 0.0006307634757831693, + "train_right_hip_rmse": 0.025115004992696484, + "train_right_hip_mae": 0.018387803807854652, + "train_right_hip_r2": 0.9192869663238525, + "train_left_knee_mse": 0.000805127841886133, + "train_left_knee_rmse": 0.028374774745998126, + "train_left_knee_mae": 0.01877642795443535, + "train_left_knee_r2": 0.9133671522140503, + "train_right_knee_mse": 0.000728983839508146, + "train_right_knee_rmse": 0.026999700729973767, + "train_right_knee_mae": 0.018754106014966965, + "train_right_knee_r2": 0.9323028922080994, + "train_left_foot_mse": 0.0012566036311909556, + "train_left_foot_rmse": 0.03544860549007472, + "train_left_foot_mae": 0.019169803708791733, + "train_left_foot_r2": 0.7090252637863159, + "train_right_foot_mse": 0.0010244412114843726, + "train_right_foot_rmse": 0.0320068931870054, + "train_right_foot_mae": 0.018891965970396996, + "train_right_foot_r2": 0.7757401466369629 + }, + "val_metrics": { + "val_mse": 0.001210120040923357, + "val_rmse": 0.03478677968601516, + "val_mae": 0.023928327485919, + "val_r2": 0.9444442391395569, + "val_head_mse": 0.0011387421982362866, + "val_head_rmse": 0.033745254455053185, + "val_head_mae": 0.02480490133166313, + "val_head_r2": 0.8927600383758545, + "val_left_shoulder_mse": 0.0006899299332872033, + "val_left_shoulder_rmse": 0.026266517342183056, + "val_left_shoulder_mae": 0.01914707012474537, + "val_left_shoulder_r2": 0.877936065196991, + "val_left_elbow_mse": 0.0011652118992060423, + "val_left_elbow_rmse": 0.03413520029538485, + "val_left_elbow_mae": 0.025532294064760208, + "val_left_elbow_r2": 0.9129767417907715, + "val_right_shoulder_mse": 0.0007348133949562907, + "val_right_shoulder_rmse": 0.027107441689622626, + "val_right_shoulder_mae": 0.019253676757216454, + "val_right_shoulder_r2": 0.8939875960350037, + "val_right_elbow_mse": 0.0014528457541018724, + "val_right_elbow_rmse": 0.038116213795468624, + "val_right_elbow_mae": 0.027071649208664894, + "val_right_elbow_r2": 0.9104633331298828, + "val_left_hand_mse": 0.0019658685196191072, + "val_left_hand_rmse": 0.04433811587809192, + "val_left_hand_mae": 0.03339832276105881, + "val_left_hand_r2": 0.9235459566116333, + "val_right_hand_mse": 0.002737270900979638, + "val_right_hand_rmse": 0.05231893444040731, + "val_right_hand_mae": 0.03548651561141014, + "val_right_hand_r2": 0.9034000039100647, + "val_left_hip_mse": 0.0007920048083178699, + "val_left_hip_rmse": 0.028142579986878778, + "val_left_hip_mae": 0.02102036029100418, + "val_left_hip_r2": 0.9095764756202698, + "val_right_hip_mse": 0.0008591497899033129, + "val_right_hip_rmse": 0.02931125705088939, + "val_right_hip_mae": 0.021341871470212936, + "val_right_hip_r2": 0.9076501727104187, + "val_left_knee_mse": 0.001003874116577208, + "val_left_knee_rmse": 0.031683972550442724, + "val_left_knee_mae": 0.02187899686396122, + "val_left_knee_r2": 0.9018368124961853, + "val_right_knee_mse": 0.0009464414324611425, + "val_right_knee_rmse": 0.030764288265148317, + "val_right_knee_mae": 0.021825246512889862, + "val_right_knee_r2": 0.9115604162216187, + "val_left_foot_mse": 0.001095280982553959, + "val_left_foot_rmse": 0.03309502957475577, + "val_left_foot_mae": 0.02020801417529583, + "val_left_foot_r2": 0.760831356048584, + "val_right_foot_mse": 0.0011501277331262827, + "val_right_foot_rmse": 0.03391353318553351, + "val_right_foot_mae": 0.020099354907870293, + "val_right_foot_r2": 0.7225411534309387 + }, + "best_epoch": 100 + }, + { + "fold": 3, + "train_metrics": { + "train_mse": 0.0008847408462315798, + "train_rmse": 0.029744593563059148, + "train_mae": 0.020830968394875526, + "train_r2": 0.9576787352561951, + "train_head_mse": 0.0007341221207752824, + "train_head_rmse": 0.02709468805458521, + "train_head_mae": 0.020325681194663048, + "train_head_r2": 0.9254001975059509, + "train_left_shoulder_mse": 0.0005135019891895354, + "train_left_shoulder_rmse": 0.022660582278254356, + "train_left_shoulder_mae": 0.01678767241537571, + "train_left_shoulder_r2": 0.9057629704475403, + "train_left_elbow_mse": 0.0008272295817732811, + "train_left_elbow_rmse": 0.02876159908234035, + "train_left_elbow_mae": 0.02166995219886303, + "train_left_elbow_r2": 0.9395068287849426, + "train_right_shoulder_mse": 0.000462483789306134, + "train_right_shoulder_rmse": 0.021505436273327124, + "train_right_shoulder_mae": 0.016256950795650482, + "train_right_shoulder_r2": 0.9281431436538696, + "train_right_elbow_mse": 0.0009132610284723341, + "train_right_elbow_rmse": 0.03022020894157309, + "train_right_elbow_mae": 0.022300943732261658, + "train_right_elbow_r2": 0.9412114024162292, + "train_left_hand_mse": 0.001406094990670681, + "train_left_hand_rmse": 0.03749793315198427, + "train_left_hand_mae": 0.028967471793293953, + "train_left_hand_r2": 0.9459758996963501, + "train_right_hand_mse": 0.0017342888750135899, + "train_right_hand_rmse": 0.04164479409258245, + "train_right_hand_mae": 0.030562052503228188, + "train_right_hand_r2": 0.9366415143013, + "train_left_hip_mse": 0.0006820462294854224, + "train_left_hip_rmse": 0.0261160148086461, + "train_left_hip_mae": 0.019746506586670876, + "train_left_hip_r2": 0.9120986461639404, + "train_right_hip_mse": 0.0006951207760721445, + "train_right_hip_rmse": 0.026365143202193013, + "train_right_hip_mae": 0.019884854555130005, + "train_right_hip_r2": 0.9142632484436035, + "train_left_knee_mse": 0.0006803437136113644, + "train_left_knee_rmse": 0.02608339919587484, + "train_left_knee_mae": 0.01857244223356247, + "train_left_knee_r2": 0.9277384877204895, + "train_right_knee_mse": 0.0008177290437743068, + "train_right_knee_rmse": 0.02859596201868905, + "train_right_knee_mae": 0.019107310101389885, + "train_right_knee_r2": 0.923542857170105, + "train_left_foot_mse": 0.0009469252545386553, + "train_left_foot_rmse": 0.030772150632327525, + "train_left_foot_mae": 0.017845408990979195, + "train_left_foot_r2": 0.7777849435806274, + "train_right_foot_mse": 0.0010884832590818405, + "train_right_foot_rmse": 0.03299216966314644, + "train_right_foot_mae": 0.018775319680571556, + "train_right_foot_r2": 0.7551426887512207 + }, + "val_metrics": { + "val_mse": 0.001260576187632978, + "val_rmse": 0.03550459389477618, + "val_mae": 0.024509267881512642, + "val_r2": 0.9429099559783936, + "val_head_mse": 0.00103144277818501, + "val_head_rmse": 0.032116082858670826, + "val_head_mae": 0.024015288800001144, + "val_head_r2": 0.9006613492965698, + "val_left_shoulder_mse": 0.0008003240218386054, + "val_left_shoulder_rmse": 0.02828999861856846, + "val_left_shoulder_mae": 0.019956035539507866, + "val_left_shoulder_r2": 0.8666128516197205, + "val_left_elbow_mse": 0.0012279492802917957, + "val_left_elbow_rmse": 0.03504210724673668, + "val_left_elbow_mae": 0.025854093953967094, + "val_left_elbow_r2": 0.9142829179763794, + "val_right_shoulder_mse": 0.0006413479568436742, + "val_right_shoulder_rmse": 0.025324848604555847, + "val_right_shoulder_mae": 0.019032785668969154, + "val_right_shoulder_r2": 0.9067311882972717, + "val_right_elbow_mse": 0.0013188716256991029, + "val_right_elbow_rmse": 0.036316272188911446, + "val_right_elbow_mae": 0.026971297338604927, + "val_right_elbow_r2": 0.9211946725845337, + "val_left_hand_mse": 0.0018678645137697458, + "val_left_hand_rmse": 0.0432187981527685, + "val_left_hand_mae": 0.03387869894504547, + "val_left_hand_r2": 0.9315763711929321, + "val_right_hand_mse": 0.0024488642811775208, + "val_right_hand_rmse": 0.049486000860622395, + "val_right_hand_mae": 0.036860499531030655, + "val_right_hand_r2": 0.9177099466323853, + "val_left_hip_mse": 0.0009909099899232388, + "val_left_hip_rmse": 0.03147872281277051, + "val_left_hip_mae": 0.022657960653305054, + "val_left_hip_r2": 0.8753138184547424, + "val_right_hip_mse": 0.0009391375933773816, + "val_right_hip_rmse": 0.030645351904936276, + "val_right_hip_mae": 0.022399840876460075, + "val_right_hip_r2": 0.8846467137336731, + "val_left_knee_mse": 0.0010395629797130823, + "val_left_knee_rmse": 0.032242254569323815, + "val_left_knee_mae": 0.022147221490740776, + "val_left_knee_r2": 0.893653154373169, + "val_right_knee_mse": 0.0012408829061314464, + "val_right_knee_rmse": 0.03522616791720959, + "val_right_knee_mae": 0.022213371470570564, + "val_right_knee_r2": 0.8871784806251526, + "val_left_foot_mse": 0.0015739506343379617, + "val_left_foot_rmse": 0.03967304669845715, + "val_left_foot_mae": 0.02135366015136242, + "val_left_foot_r2": 0.6721795797348022, + "val_right_foot_mse": 0.001266381237655878, + "val_right_foot_rmse": 0.035586250682754964, + "val_right_foot_mae": 0.0212797150015831, + "val_right_foot_r2": 0.7268164157867432 + }, + "best_epoch": 100 + }, + { + "fold": 4, + "train_metrics": { + "train_mse": 0.0009269318543374538, + "train_rmse": 0.030445555576100986, + "train_mae": 0.0208336990326643, + "train_r2": 0.956462025642395, + "train_head_mse": 0.0008762200595811009, + "train_head_rmse": 0.029601014502565633, + "train_head_mae": 0.0216806810349226, + "train_head_r2": 0.9120126962661743, + "train_left_shoulder_mse": 0.000551835575606674, + "train_left_shoulder_rmse": 0.023491180804861088, + "train_left_shoulder_mae": 0.017186248674988747, + "train_left_shoulder_r2": 0.9012649059295654, + "train_left_elbow_mse": 0.0008788025006651878, + "train_left_elbow_rmse": 0.029644603230017903, + "train_left_elbow_mae": 0.02248154580593109, + "train_left_elbow_r2": 0.9363320469856262, + "train_right_shoulder_mse": 0.00046319011016748846, + "train_right_shoulder_rmse": 0.021521851922348328, + "train_right_shoulder_mae": 0.016026239842176437, + "train_right_shoulder_r2": 0.9292229413986206, + "train_right_elbow_mse": 0.0009069484658539295, + "train_right_elbow_rmse": 0.030115585098980387, + "train_right_elbow_mae": 0.022317806258797646, + "train_right_elbow_r2": 0.9424450397491455, + "train_left_hand_mse": 0.0013085671234875917, + "train_left_hand_rmse": 0.03617412229049368, + "train_left_hand_mae": 0.027895158156752586, + "train_left_hand_r2": 0.9505067467689514, + "train_right_hand_mse": 0.001675135106779635, + "train_right_hand_rmse": 0.04092841441809877, + "train_right_hand_mae": 0.029857559129595757, + "train_right_hand_r2": 0.9400901198387146, + "train_left_hip_mse": 0.0006678422214463353, + "train_left_hip_rmse": 0.025842643468622466, + "train_left_hip_mae": 0.01883869804441929, + "train_left_hip_r2": 0.9150352478027344, + "train_right_hip_mse": 0.0006321837427094579, + "train_right_hip_rmse": 0.025143264360648516, + "train_right_hip_mae": 0.018356971442699432, + "train_right_hip_r2": 0.9227392673492432, + "train_left_knee_mse": 0.0007528851856477559, + "train_left_knee_rmse": 0.02743875335447578, + "train_left_knee_mae": 0.018794866278767586, + "train_left_knee_r2": 0.9221354126930237, + "train_right_knee_mse": 0.0009005428873933852, + "train_right_knee_rmse": 0.030009046759158896, + "train_right_knee_mae": 0.01941732130944729, + "train_right_knee_r2": 0.9179890155792236, + "train_left_foot_mse": 0.001133721205405891, + "train_left_foot_rmse": 0.03367077672709513, + "train_left_foot_mae": 0.018883002921938896, + "train_left_foot_r2": 0.7337841987609863, + "train_right_foot_mse": 0.001302240532822907, + "train_right_foot_rmse": 0.036086569978634804, + "train_right_foot_mae": 0.019101988524198532, + "train_right_foot_r2": 0.7141515016555786 + }, + "val_metrics": { + "val_mse": 0.00121107860468328, + "val_rmse": 0.03480055466056942, + "val_mae": 0.022995375096797943, + "val_r2": 0.9410244226455688, + "val_head_mse": 0.0010465334635227919, + "val_head_rmse": 0.03235016945122223, + "val_head_mae": 0.02346768230199814, + "val_head_r2": 0.8944224715232849, + "val_left_shoulder_mse": 0.0006662497180514038, + "val_left_shoulder_rmse": 0.025811813536661923, + "val_left_shoulder_mae": 0.01901177130639553, + "val_left_shoulder_r2": 0.8774450421333313, + "val_left_elbow_mse": 0.001142294961027801, + "val_left_elbow_rmse": 0.033797854384972445, + "val_left_elbow_mae": 0.025222117081284523, + "val_left_elbow_r2": 0.9172838926315308, + "val_right_shoulder_mse": 0.000599258579313755, + "val_right_shoulder_rmse": 0.024479758563224333, + "val_right_shoulder_mae": 0.01820112019777298, + "val_right_shoulder_r2": 0.9069446325302124, + "val_right_elbow_mse": 0.00141298642847687, + "val_right_elbow_rmse": 0.03758971173708133, + "val_right_elbow_mae": 0.025280525907874107, + "val_right_elbow_r2": 0.9106413125991821, + "val_left_hand_mse": 0.0018793701892718673, + "val_left_hand_rmse": 0.04335170341834179, + "val_left_hand_mae": 0.03159584850072861, + "val_left_hand_r2": 0.9267331957817078, + "val_right_hand_mse": 0.0023547711316496134, + "val_right_hand_rmse": 0.048525984087389853, + "val_right_hand_mae": 0.03225317224860191, + "val_right_hand_r2": 0.9138808250427246, + "val_left_hip_mse": 0.0008806693949736655, + "val_left_hip_rmse": 0.02967607445356723, + "val_left_hip_mae": 0.020833058282732964, + "val_left_hip_r2": 0.8832638263702393, + "val_right_hip_mse": 0.000838539912365377, + "val_right_hip_rmse": 0.02895755363226281, + "val_right_hip_mae": 0.020357400178909302, + "val_right_hip_r2": 0.89308762550354, + "val_left_knee_mse": 0.00098232866730541, + "val_left_knee_rmse": 0.03134212289085425, + "val_left_knee_mae": 0.020191719755530357, + "val_left_knee_r2": 0.8878087997436523, + "val_right_knee_mse": 0.0007243398576974869, + "val_right_knee_rmse": 0.026913562709115397, + "val_right_knee_mae": 0.018779123201966286, + "val_right_knee_r2": 0.9265093803405762, + "val_left_foot_mse": 0.0018784793792292476, + "val_left_foot_rmse": 0.043341427978658566, + "val_left_foot_mae": 0.022326767444610596, + "val_left_foot_r2": 0.6107417345046997, + "val_right_foot_mse": 0.0013382014585658908, + "val_right_foot_rmse": 0.03658143598283002, + "val_right_foot_mae": 0.02141956239938736, + "val_right_foot_r2": 0.6813136339187622 + }, + "best_epoch": 100 + }, + { + "fold": 5, + "train_metrics": { + "train_mse": 0.0009815376251935959, + "train_rmse": 0.03132950087686677, + "train_mae": 0.021532071754336357, + "train_r2": 0.9537968635559082, + "train_head_mse": 0.0008554799715057015, + "train_head_rmse": 0.02924858922248561, + "train_head_mae": 0.021828200668096542, + "train_head_r2": 0.9150662422180176, + "train_left_shoulder_mse": 0.0005610958323813975, + "train_left_shoulder_rmse": 0.023687461501423014, + "train_left_shoulder_mae": 0.01769471913576126, + "train_left_shoulder_r2": 0.8993830680847168, + "train_left_elbow_mse": 0.0009806574089452624, + "train_left_elbow_rmse": 0.03131545000387608, + "train_left_elbow_mae": 0.023333091288805008, + "train_left_elbow_r2": 0.929207444190979, + "train_right_shoulder_mse": 0.0004660917038563639, + "train_right_shoulder_rmse": 0.02158915708999228, + "train_right_shoulder_mae": 0.016494538635015488, + "train_right_shoulder_r2": 0.9295235872268677, + "train_right_elbow_mse": 0.0009448248310945928, + "train_right_elbow_rmse": 0.03073800304337601, + "train_right_elbow_mae": 0.022475257515907288, + "train_right_elbow_r2": 0.9409950971603394, + "train_left_hand_mse": 0.0014810053398832679, + "train_left_hand_rmse": 0.03848383218811853, + "train_left_hand_mae": 0.029210319742560387, + "train_left_hand_r2": 0.9441101551055908, + "train_right_hand_mse": 0.0018273867899551988, + "train_right_hand_rmse": 0.04274794486235799, + "train_right_hand_mae": 0.029981721192598343, + "train_right_hand_r2": 0.9353528618812561, + "train_left_hip_mse": 0.0006390114431269467, + "train_left_hip_rmse": 0.02527867566006864, + "train_left_hip_mae": 0.018952658399939537, + "train_left_hip_r2": 0.9182750582695007, + "train_right_hip_mse": 0.0006384378066286445, + "train_right_hip_rmse": 0.025267326859575875, + "train_right_hip_mae": 0.018993070349097252, + "train_right_hip_r2": 0.921862006187439, + "train_left_knee_mse": 0.0008768662228249013, + "train_left_knee_rmse": 0.02961192703666719, + "train_left_knee_mae": 0.020407285541296005, + "train_left_knee_r2": 0.907651960849762, + "train_right_knee_mse": 0.0008914409554563463, + "train_right_knee_rmse": 0.02985700848136575, + "train_right_knee_mae": 0.02000393718481064, + "train_right_knee_r2": 0.9157769680023193, + "train_left_foot_mse": 0.0014236214337870479, + "train_left_foot_rmse": 0.03773090820252075, + "train_left_foot_mae": 0.020748624578118324, + "train_left_foot_r2": 0.6919668912887573, + "train_right_foot_mse": 0.00117406970821321, + "train_right_foot_rmse": 0.03426470061467355, + "train_right_foot_mae": 0.01979350484907627, + "train_right_foot_r2": 0.7339498996734619 + }, + "val_metrics": { + "val_mse": 0.001285066711716354, + "val_rmse": 0.03584782715474334, + "val_mae": 0.02433391846716404, + "val_r2": 0.9379990100860596, + "val_head_mse": 0.0011411448940634727, + "val_head_rmse": 0.03378083619544479, + "val_head_mae": 0.02504619024693966, + "val_head_r2": 0.879335880279541, + "val_left_shoulder_mse": 0.0007593464106321335, + "val_left_shoulder_rmse": 0.027556240865403495, + "val_left_shoulder_mae": 0.020785031840205193, + "val_left_shoulder_r2": 0.8617063760757446, + "val_left_elbow_mse": 0.0012099726591259241, + "val_left_elbow_rmse": 0.034784661262198945, + "val_left_elbow_mae": 0.026303531602025032, + "val_left_elbow_r2": 0.9112187027931213, + "val_right_shoulder_mse": 0.0006928733200766146, + "val_right_shoulder_rmse": 0.026322486966026112, + "val_right_shoulder_mae": 0.019705185666680336, + "val_right_shoulder_r2": 0.887645959854126, + "val_right_elbow_mse": 0.0012706873239949346, + "val_right_elbow_rmse": 0.03564670144620585, + "val_right_elbow_mae": 0.025907862931489944, + "val_right_elbow_r2": 0.9143009185791016, + "val_left_hand_mse": 0.00193304312415421, + "val_left_hand_rmse": 0.04396638629856006, + "val_left_hand_mae": 0.03279866650700569, + "val_left_hand_r2": 0.9240407347679138, + "val_right_hand_mse": 0.0023517077788710594, + "val_right_hand_rmse": 0.048494409769282265, + "val_right_hand_mae": 0.03433976322412491, + "val_right_hand_r2": 0.91022789478302, + "val_left_hip_mse": 0.0008511691121384501, + "val_left_hip_rmse": 0.02917480269236538, + "val_left_hip_mae": 0.021669303998351097, + "val_left_hip_r2": 0.8896164894104004, + "val_right_hip_mse": 0.0008620635489933193, + "val_right_hip_rmse": 0.029360918735511655, + "val_right_hip_mae": 0.021808328106999397, + "val_right_hip_r2": 0.8907712697982788, + "val_left_knee_mse": 0.000937609700486064, + "val_left_knee_rmse": 0.030620413133824045, + "val_left_knee_mae": 0.02258199267089367, + "val_left_knee_r2": 0.9008157253265381, + "val_right_knee_mse": 0.00164791417773813, + "val_right_knee_rmse": 0.04059450920676502, + "val_right_knee_mae": 0.024780159816145897, + "val_right_knee_r2": 0.8560547828674316, + "val_left_foot_mse": 0.0009265731787309051, + "val_left_foot_rmse": 0.030439664563376927, + "val_left_foot_mae": 0.018732750788331032, + "val_left_foot_r2": 0.7257314920425415, + "val_right_foot_mse": 0.002121761441230774, + "val_right_foot_rmse": 0.04606258179076347, + "val_right_foot_mae": 0.021882185712456703, + "val_right_foot_r2": 0.5538344383239746 + }, + "best_epoch": 100 + } + ], + "test_metrics": { + "test_mse": 0.0011878072982653975, + "test_rmse": 0.03446458034367164, + "test_mae": 0.023634184151887894, + "test_r2": 0.9443016052246094, + "test_head_mse": 0.0010633665369823575, + "test_head_rmse": 0.032609301387523736, + "test_head_mae": 0.02437138743698597, + "test_head_r2": 0.895630955696106, + "test_left_shoulder_mse": 0.0007250756025314331, + "test_left_shoulder_rmse": 0.026927227902839034, + "test_left_shoulder_mae": 0.019380683079361916, + "test_left_shoulder_r2": 0.8728057146072388, + "test_left_elbow_mse": 0.0010455182055011392, + "test_left_elbow_rmse": 0.032334473948112084, + "test_left_elbow_mae": 0.023429356515407562, + "test_left_elbow_r2": 0.9228355884552002, + "test_right_shoulder_mse": 0.0007288106135092676, + "test_right_shoulder_rmse": 0.026996492614954033, + "test_right_shoulder_mae": 0.019620945677161217, + "test_right_shoulder_r2": 0.8905267715454102, + "test_right_elbow_mse": 0.001342307892628014, + "test_right_elbow_rmse": 0.036637520284921225, + "test_right_elbow_mae": 0.025898402556777, + "test_right_elbow_r2": 0.9141458868980408, + "test_left_hand_mse": 0.0015924208564683795, + "test_left_hand_rmse": 0.039905148245162295, + "test_left_hand_mae": 0.029757991433143616, + "test_left_hand_r2": 0.9391962289810181, + "test_right_hand_mse": 0.0021561961621046066, + "test_right_hand_rmse": 0.04643485934192766, + "test_right_hand_mae": 0.034087229520082474, + "test_right_hand_r2": 0.9233702421188354, + "test_left_hip_mse": 0.000896528537850827, + "test_left_hip_rmse": 0.029942086397758375, + "test_left_hip_mae": 0.02181306481361389, + "test_left_hip_r2": 0.8982810974121094, + "test_right_hip_mse": 0.0008973250514827669, + "test_right_hip_rmse": 0.029955384348773876, + "test_right_hip_mae": 0.021620769053697586, + "test_right_hip_r2": 0.9020769596099854, + "test_left_knee_mse": 0.0010164525592699647, + "test_left_knee_rmse": 0.031881853134188495, + "test_left_knee_mae": 0.021726343780755997, + "test_left_knee_r2": 0.895525336265564, + "test_right_knee_mse": 0.0009778690291568637, + "test_right_knee_rmse": 0.0312708974792356, + "test_right_knee_mae": 0.02189616672694683, + "test_right_knee_r2": 0.9088101387023926, + "test_left_foot_mse": 0.0016503861406818032, + "test_left_foot_rmse": 0.040624944808353936, + "test_left_foot_mae": 0.021257631480693817, + "test_left_foot_r2": 0.645123302936554, + "test_right_foot_mse": 0.0013492393773049116, + "test_right_foot_rmse": 0.03673199391953712, + "test_right_foot_mae": 0.02238442189991474, + "test_right_foot_r2": 0.7429869174957275 + } +} \ No newline at end of file diff --git a/A9/cv_results_loss_optimizer/conv1d_v3_rmsprop_mae_fold1_best.h5 b/A9/cv_results_loss_optimizer/conv1d_v3_rmsprop_mae_fold1_best.h5 new file mode 100644 index 0000000000000000000000000000000000000000..89921147dde28c6f1f086ec966061a501be3afc2 --- /dev/null +++ b/A9/cv_results_loss_optimizer/conv1d_v3_rmsprop_mae_fold1_best.h5 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e7ab0c90005d3e913344b801e7a186f7bd4f06b5a38fbf66c3d2bd8dd031ce6e +size 2658520 diff --git a/A9/cv_results_loss_optimizer/conv1d_v3_rmsprop_mae_fold2_best.h5 b/A9/cv_results_loss_optimizer/conv1d_v3_rmsprop_mae_fold2_best.h5 new file mode 100644 index 0000000000000000000000000000000000000000..6a4728c99d39cde5facee3f47f41d69673973bcf --- /dev/null +++ b/A9/cv_results_loss_optimizer/conv1d_v3_rmsprop_mae_fold2_best.h5 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:3f36a9b3ec93ba7e5d90d47ca4200ec81f08f215e4124e22abfdb652f624d6a0 +size 2658520 diff --git a/A9/cv_results_loss_optimizer/conv1d_v3_rmsprop_mae_fold3_best.h5 b/A9/cv_results_loss_optimizer/conv1d_v3_rmsprop_mae_fold3_best.h5 new file mode 100644 index 0000000000000000000000000000000000000000..1d20be8a0079ae18f6716b7d4759590fc1c8794a --- /dev/null +++ b/A9/cv_results_loss_optimizer/conv1d_v3_rmsprop_mae_fold3_best.h5 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b2b2514ab915b662288fc50d5ea84df9d7d690313d3e0042e50c0d876bce3e5e +size 2658520 diff --git a/A9/cv_results_loss_optimizer/conv1d_v3_rmsprop_mae_fold4_best.h5 b/A9/cv_results_loss_optimizer/conv1d_v3_rmsprop_mae_fold4_best.h5 new file mode 100644 index 0000000000000000000000000000000000000000..8db886d0ca4f85efeb27a72c10ea05f803d33a0b --- /dev/null +++ b/A9/cv_results_loss_optimizer/conv1d_v3_rmsprop_mae_fold4_best.h5 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b8d4306714eaeee99a19ad9de9ef31fd0ff8f2764f6627a1455103909178a5e6 +size 2658520 diff --git a/A9/cv_results_loss_optimizer/conv1d_v3_rmsprop_mae_fold5_best.h5 b/A9/cv_results_loss_optimizer/conv1d_v3_rmsprop_mae_fold5_best.h5 new file mode 100644 index 0000000000000000000000000000000000000000..c4d38416d4bd6e6c98c22634d0f4fad9007fc72b --- /dev/null +++ b/A9/cv_results_loss_optimizer/conv1d_v3_rmsprop_mae_fold5_best.h5 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:0f766c22338cc6d7ea31117d70e7c312a7d73eb886f230e3aeb9316eba324a7e +size 2658520 diff --git a/A9/cv_results_loss_optimizer/conv1d_v3_rmsprop_mae_predictions.png b/A9/cv_results_loss_optimizer/conv1d_v3_rmsprop_mae_predictions.png new file mode 100644 index 0000000000000000000000000000000000000000..4def4f4c75d23c8e690d13bff252d8d22cdbed4b --- /dev/null +++ b/A9/cv_results_loss_optimizer/conv1d_v3_rmsprop_mae_predictions.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b30684e16f7257d56a41943b95427d79b9845afda7ba2ec65955c2eadb47c125 +size 550170 diff --git a/A9/cv_results_loss_optimizer/conv1d_v3_rmsprop_mae_test_predictions.png b/A9/cv_results_loss_optimizer/conv1d_v3_rmsprop_mae_test_predictions.png new file mode 100644 index 0000000000000000000000000000000000000000..2178469870276b530ca802c1377fd64436039950 --- /dev/null +++ b/A9/cv_results_loss_optimizer/conv1d_v3_rmsprop_mae_test_predictions.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:48ce0c3ffe4683a4ba0f04bd20fd55081e689c9c48561e136f1873965bb3c895 +size 441468 diff --git a/A9/cv_results_loss_optimizer/conv1d_v3_rmsprop_mae_training_history.png b/A9/cv_results_loss_optimizer/conv1d_v3_rmsprop_mae_training_history.png new file mode 100644 index 0000000000000000000000000000000000000000..eac0d13d1056c0cb1a02e0f73c43463db941ec1f --- /dev/null +++ b/A9/cv_results_loss_optimizer/conv1d_v3_rmsprop_mae_training_history.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:d979ba15f44920f7d9d0bc9168937257bd1216a29eb7da452423a679852ee559 +size 319716 diff --git a/A9/cv_results_loss_optimizer/conv1d_v3_rmsprop_mse_cv_results.json b/A9/cv_results_loss_optimizer/conv1d_v3_rmsprop_mse_cv_results.json new file mode 100644 index 0000000000000000000000000000000000000000..9fd29c1d9df624a0bec35e30225c197417fe325a --- /dev/null +++ b/A9/cv_results_loss_optimizer/conv1d_v3_rmsprop_mse_cv_results.json @@ -0,0 +1,1243 @@ +{ + "model_name": "conv1d_v3_rmsprop_mse", + "n_splits": 5, + "epochs": 100, + "batch_size": 32, + "patience": 10, + "aggregate_results": { + "n_folds": 5, + "train_metrics": { + "train_mse": { + "mean": 0.000796480756253004, + "std": 7.444720371720353e-05 + }, + "train_rmse": { + "mean": 0.02819088105926194, + "std": 0.0013247570930365205 + }, + "train_mae": { + "mean": 0.02092903107404709, + "std": 0.0008763380956310857 + }, + "train_r2": { + "mean": 0.9623377442359924, + "std": 0.0033590586778876634 + }, + "train_head_mse": { + "mean": 0.0008489995962008834, + "std": 7.775306426685101e-05 + }, + "train_head_rmse": { + "mean": 0.029106824410313918, + "std": 0.0013387938407527099 + }, + "train_head_mae": { + "mean": 0.022153063490986825, + "std": 0.0010697372437146672 + }, + "train_head_r2": { + "mean": 0.9147328019142151, + "std": 0.006944739377242398 + }, + "train_left_shoulder_mse": { + "mean": 0.0005482923705130815, + "std": 4.335946991483168e-05 + }, + "train_left_shoulder_rmse": { + "mean": 0.023397214379091452, + "std": 0.000928832499387058 + }, + "train_left_shoulder_mae": { + "mean": 0.017886571958661078, + "std": 0.0007139105466980459 + }, + "train_left_shoulder_r2": { + "mean": 0.9014155030250549, + "std": 0.007117927204114238 + }, + "train_left_elbow_mse": { + "mean": 0.0008330822689458728, + "std": 7.300813515249173e-05 + }, + "train_left_elbow_rmse": { + "mean": 0.028835104967787534, + "std": 0.001272395552706561 + }, + "train_left_elbow_mae": { + "mean": 0.022121883928775787, + "std": 0.0009507893039638193 + }, + "train_left_elbow_r2": { + "mean": 0.9396776676177978, + "std": 0.005104415647675789 + }, + "train_right_shoulder_mse": { + "mean": 0.00047872722498141227, + "std": 3.436834975768271e-05 + }, + "train_right_shoulder_rmse": { + "mean": 0.021865628598935285, + "std": 0.0007883596608374544 + }, + "train_right_shoulder_mae": { + "mean": 0.016969437897205352, + "std": 0.0005917836229943771 + }, + "train_right_shoulder_r2": { + "mean": 0.9266458868980407, + "std": 0.004789468877022828 + }, + "train_right_elbow_mse": { + "mean": 0.0007669288781471551, + "std": 5.850704402920195e-05 + }, + "train_right_elbow_rmse": { + "mean": 0.027673059678191208, + "std": 0.0010633184821223515 + }, + "train_right_elbow_mae": { + "mean": 0.021420005708932877, + "std": 0.0007674621240139273 + }, + "train_right_elbow_r2": { + "mean": 0.9514127731323242, + "std": 0.0032538876898056297 + }, + "train_left_hand_mse": { + "mean": 0.001222145138308406, + "std": 0.00012735451786674044 + }, + "train_left_hand_rmse": { + "mean": 0.03491119666560227, + "std": 0.0018312524905229497 + }, + "train_left_hand_mae": { + "mean": 0.02729208767414093, + "std": 0.0013297571946987385 + }, + "train_left_hand_r2": { + "mean": 0.9535073041915894, + "std": 0.004843856418978533 + }, + "train_right_hand_mse": { + "mean": 0.001324491831474006, + "std": 0.00013912468207609384 + }, + "train_right_hand_rmse": { + "mean": 0.036342405431522465, + "std": 0.0019290927724847824 + }, + "train_right_hand_mae": { + "mean": 0.027640661224722862, + "std": 0.0012212577287092766 + }, + "train_right_hand_r2": { + "mean": 0.952479898929596, + "std": 0.004636938070618124 + }, + "train_left_hip_mse": { + "mean": 0.000642840622458607, + "std": 5.657779678718383e-05 + }, + "train_left_hip_rmse": { + "mean": 0.025329009967155414, + "std": 0.0011321998685517316 + }, + "train_left_hip_mae": { + "mean": 0.019339507818222045, + "std": 0.0008351207970781456 + }, + "train_left_hip_r2": { + "mean": 0.9175400376319885, + "std": 0.007296777807273149 + }, + "train_right_hip_mse": { + "mean": 0.0006328940042294561, + "std": 4.8757890469224454e-05 + }, + "train_right_hip_rmse": { + "mean": 0.02513824877090725, + "std": 0.0009810468701525962 + }, + "train_right_hip_mae": { + "mean": 0.01908394992351532, + "std": 0.0006106698463428748 + }, + "train_right_hip_r2": { + "mean": 0.9219738602638244, + "std": 0.006257677014342614 + }, + "train_left_knee_mse": { + "mean": 0.0006787988706491888, + "std": 7.915604433647892e-05 + }, + "train_left_knee_rmse": { + "mean": 0.026009140617678367, + "std": 0.0015242949120895695 + }, + "train_left_knee_mae": { + "mean": 0.019214300066232683, + "std": 0.0009635552789927117 + }, + "train_left_knee_r2": { + "mean": 0.9284448862075806, + "std": 0.008300400368214932 + }, + "train_right_knee_mse": { + "mean": 0.0006514313514344394, + "std": 6.880959548941124e-05 + }, + "train_right_knee_rmse": { + "mean": 0.025487323206452195, + "std": 0.0013519272185597826 + }, + "train_right_knee_mae": { + "mean": 0.019253762438893318, + "std": 0.000766153401225821 + }, + "train_right_knee_r2": { + "mean": 0.9393937349319458, + "std": 0.006820888674963618 + }, + "train_left_foot_mse": { + "mean": 0.0008797234739176929, + "std": 0.0001016283162751079 + }, + "train_left_foot_rmse": { + "mean": 0.029613348628652142, + "std": 0.001665249805764671 + }, + "train_left_foot_mae": { + "mean": 0.019426900520920755, + "std": 0.001028762582072227 + }, + "train_left_foot_r2": { + "mean": 0.7993392229080201, + "std": 0.016539011911609242 + }, + "train_right_foot_mse": { + "mean": 0.0008458942407742143, + "std": 9.623444808177527e-05 + }, + "train_right_foot_rmse": { + "mean": 0.029037252145626323, + "std": 0.0016529454333200956 + }, + "train_right_foot_mae": { + "mean": 0.020275282859802245, + "std": 0.0011997981492818568 + }, + "train_right_foot_r2": { + "mean": 0.811170244216919, + "std": 0.023081124648684406 + } + }, + "val_metrics": { + "val_mse": { + "mean": 0.0010544779244810343, + "std": 4.0910378221683397e-05 + }, + "val_rmse": { + "mean": 0.03246663755571122, + "std": 0.0006287847860088607 + }, + "val_mae": { + "mean": 0.023508354648947717, + "std": 0.0007427242681017118 + }, + "val_r2": { + "mean": 0.9500379204750061, + "std": 0.002828407245393275 + }, + "val_head_mse": { + "mean": 0.0010626421542838215, + "std": 7.815587845743406e-05 + }, + "val_head_rmse": { + "mean": 0.032575532438867696, + "std": 0.0012152533102610357 + }, + "val_head_mae": { + "mean": 0.02430778481066227, + "std": 0.00089030375679105 + }, + "val_head_r2": { + "mean": 0.8927295088768006, + "std": 0.010764489121804598 + }, + "val_left_shoulder_mse": { + "mean": 0.0006989713874645531, + "std": 3.8621099452425585e-05 + }, + "val_left_shoulder_rmse": { + "mean": 0.026427770374197364, + "std": 0.0007377943570191868 + }, + "val_left_shoulder_mae": { + "mean": 0.01981595382094383, + "std": 0.0007330943793440925 + }, + "val_left_shoulder_r2": { + "mean": 0.8738594889640808, + "std": 0.008982325027637416 + }, + "val_left_elbow_mse": { + "mean": 0.0010944330831989646, + "std": 4.2934825749184165e-05 + }, + "val_left_elbow_rmse": { + "mean": 0.033075852678796806, + "std": 0.000648885790794162 + }, + "val_left_elbow_mae": { + "mean": 0.025304192677140236, + "std": 0.0006277206049564899 + }, + "val_left_elbow_r2": { + "mean": 0.9205707669258117, + "std": 0.0041424841391178475 + }, + "val_right_shoulder_mse": { + "mean": 0.0006316444836556911, + "std": 7.122141042654293e-05 + }, + "val_right_shoulder_rmse": { + "mean": 0.025090210571880434, + "std": 0.0014580182146977805 + }, + "val_right_shoulder_mae": { + "mean": 0.01922006271779537, + "std": 0.0009278690550935061 + }, + "val_right_shoulder_r2": { + "mean": 0.9028211951255798, + "std": 0.01250963394515613 + }, + "val_right_elbow_mse": { + "mean": 0.0010765020386315882, + "std": 7.128896381384846e-05 + }, + "val_right_elbow_rmse": { + "mean": 0.03279213999656338, + "std": 0.0010851696076543624 + }, + "val_right_elbow_mae": { + "mean": 0.02501179128885269, + "std": 0.001088995095204632 + }, + "val_right_elbow_r2": { + "mean": 0.9314642906188965, + "std": 0.006602183528915347 + }, + "val_left_hand_mse": { + "mean": 0.0017040885286405683, + "std": 9.010785477073149e-05 + }, + "val_left_hand_rmse": { + "mean": 0.04126637177679658, + "std": 0.0010840152304163572 + }, + "val_left_hand_mae": { + "mean": 0.03179153650999069, + "std": 0.0009942361487109887 + }, + "val_left_hand_r2": { + "mean": 0.9349550247192383, + "std": 0.005021019918542181 + }, + "val_right_hand_mse": { + "mean": 0.0018612712388858198, + "std": 0.00012541338838522575 + }, + "val_right_hand_rmse": { + "mean": 0.04311751724836864, + "std": 0.0014666101126247317 + }, + "val_right_hand_mae": { + "mean": 0.03199148662388325, + "std": 0.0014088889279018533 + }, + "val_right_hand_r2": { + "mean": 0.9329439759254455, + "std": 0.005840392394548321 + }, + "val_left_hip_mse": { + "mean": 0.0007949785445816815, + "std": 7.37607104192516e-05 + }, + "val_left_hip_rmse": { + "mean": 0.02816614816726946, + "std": 0.0012832154928556128 + }, + "val_left_hip_mae": { + "mean": 0.02113706208765507, + "std": 0.0009371674812661731 + }, + "val_left_hip_r2": { + "mean": 0.8975067257881164, + "std": 0.011738062398398275 + }, + "val_right_hip_mse": { + "mean": 0.0007998691755346954, + "std": 8.368488217238693e-05 + }, + "val_right_hip_rmse": { + "mean": 0.028244101451252807, + "std": 0.0014628426935369696 + }, + "val_right_hip_mae": { + "mean": 0.020961252227425575, + "std": 0.0008579227925157762 + }, + "val_right_hip_r2": { + "mean": 0.9009030342102051, + "std": 0.012226857319581412 + }, + "val_left_knee_mse": { + "mean": 0.000860901060514152, + "std": 5.162362122503315e-05 + }, + "val_left_knee_rmse": { + "mean": 0.029327588903104457, + "std": 0.0008908364859256328 + }, + "val_left_knee_mae": { + "mean": 0.02129170373082161, + "std": 0.0009007522075376602 + }, + "val_left_knee_r2": { + "mean": 0.9091060996055603, + "std": 0.004623031018055739 + }, + "val_right_knee_mse": { + "mean": 0.0008836076245643199, + "std": 0.00013481484139615346 + }, + "val_right_knee_rmse": { + "mean": 0.029639833199575073, + "std": 0.002255640145432566 + }, + "val_right_knee_mae": { + "mean": 0.02165335826575756, + "std": 0.001680956233126082 + }, + "val_right_knee_r2": { + "mean": 0.9179434299468994, + "std": 0.010612712124941538 + }, + "val_left_foot_mse": { + "mean": 0.0010833978769369424, + "std": 0.0002639516130459695 + }, + "val_left_foot_rmse": { + "mean": 0.03265140195247032, + "std": 0.004157382286386432 + }, + "val_left_foot_mae": { + "mean": 0.02092561833560467, + "std": 0.00098502252669093 + }, + "val_left_foot_r2": { + "mean": 0.7536278009414673, + "std": 0.04471620065332273 + }, + "val_right_foot_mse": { + "mean": 0.001155906671192497, + "std": 0.00017220162860101892 + }, + "val_right_foot_rmse": { + "mean": 0.033900206411305846, + "std": 0.002585087322191449 + }, + "val_right_foot_mae": { + "mean": 0.022196802496910095, + "std": 0.0011549287740290814 + }, + "val_right_foot_r2": { + "mean": 0.741330885887146, + "std": 0.04037743920552572 + } + }, + "best_fold": "4", + "best_fold_metrics": { + "fold": 4, + "train_metrics": { + "train_mse": 0.0007396690198220313, + "train_rmse": 0.027196856800410434, + "train_mae": 0.020389899611473083, + "train_r2": 0.9652577638626099, + "train_head_mse": 0.0008147882181219757, + "train_head_rmse": 0.028544495408431653, + "train_head_mae": 0.02170136384665966, + "train_head_r2": 0.9181814789772034, + "train_left_shoulder_mse": 0.0005042082630097866, + "train_left_shoulder_rmse": 0.022454582227460537, + "train_left_shoulder_mae": 0.017156267538666725, + "train_left_shoulder_r2": 0.9097864031791687, + "train_left_elbow_mse": 0.0007951081497594714, + "train_left_elbow_rmse": 0.028197662132869658, + "train_left_elbow_mae": 0.02168133482336998, + "train_left_elbow_r2": 0.9423955678939819, + "train_right_shoulder_mse": 0.0004380958271212876, + "train_right_shoulder_rmse": 0.020930738809733583, + "train_right_shoulder_mae": 0.016288021579384804, + "train_right_shoulder_r2": 0.933057427406311, + "train_right_elbow_mse": 0.0007245641900226474, + "train_right_elbow_rmse": 0.026917730031015753, + "train_right_elbow_mae": 0.02116497792303562, + "train_right_elbow_r2": 0.9540191292762756, + "train_left_hand_mse": 0.0010757292620837688, + "train_left_hand_rmse": 0.0327983118785673, + "train_left_hand_mae": 0.02558828331530094, + "train_left_hand_r2": 0.9593132734298706, + "train_right_hand_mse": 0.0012341623660176992, + "train_right_hand_rmse": 0.035130647105023546, + "train_right_hand_mae": 0.02724386751651764, + "train_right_hand_r2": 0.9558611512184143, + "train_left_hip_mse": 0.000608258240390569, + "train_left_hip_rmse": 0.024662891971351797, + "train_left_hip_mae": 0.0188661627471447, + "train_left_hip_r2": 0.9226157069206238, + "train_right_hip_mse": 0.00060211744857952, + "train_right_hip_rmse": 0.02453808159941441, + "train_right_hip_mae": 0.018776990473270416, + "train_right_hip_r2": 0.9264137744903564, + "train_left_knee_mse": 0.000623013882432133, + "train_left_knee_rmse": 0.02496024604109769, + "train_left_knee_mae": 0.01885506883263588, + "train_left_knee_r2": 0.9355669021606445, + "train_right_knee_mse": 0.0006075026467442513, + "train_right_knee_rmse": 0.024647568779582525, + "train_right_knee_mae": 0.019012607634067535, + "train_right_knee_r2": 0.9446756839752197, + "train_left_foot_mse": 0.0008118184050545096, + "train_left_foot_rmse": 0.02849242715274551, + "train_left_foot_mae": 0.018963536247611046, + "train_left_foot_r2": 0.809372067451477, + "train_right_foot_mse": 0.0007763301837258041, + "train_right_foot_rmse": 0.027862702376578695, + "train_right_foot_mae": 0.019770218059420586, + "train_right_foot_r2": 0.8295915126800537 + }, + "val_metrics": { + "val_mse": 0.0010027959942817688, + "val_rmse": 0.031666954294370796, + "val_mae": 0.02263071946799755, + "val_r2": 0.951167106628418, + "val_head_mse": 0.0009305973071604967, + "val_head_rmse": 0.03050569302868723, + "val_head_mae": 0.023090656846761703, + "val_head_r2": 0.9061184525489807, + "val_left_shoulder_mse": 0.0006306541035883129, + "val_left_shoulder_rmse": 0.025112827471002002, + "val_left_shoulder_mae": 0.018864132463932037, + "val_left_shoulder_r2": 0.8839927911758423, + "val_left_elbow_mse": 0.0011112610809504986, + "val_left_elbow_rmse": 0.0333355828050223, + "val_left_elbow_mae": 0.02487783506512642, + "val_left_elbow_r2": 0.9195311069488525, + "val_right_shoulder_mse": 0.0005033456836827099, + "val_right_shoulder_rmse": 0.022435366805174144, + "val_right_shoulder_mae": 0.017676031216979027, + "val_right_shoulder_r2": 0.9218384027481079, + "val_right_elbow_mse": 0.0009742524125613272, + "val_right_elbow_rmse": 0.031213016716769418, + "val_right_elbow_mae": 0.023107292130589485, + "val_right_elbow_r2": 0.9383873343467712, + "val_left_hand_mse": 0.0016541049117222428, + "val_left_hand_rmse": 0.04067068860644288, + "val_left_hand_mae": 0.030412372201681137, + "val_left_hand_r2": 0.9355151057243347, + "val_right_hand_mse": 0.0016513929003849626, + "val_right_hand_rmse": 0.040637333824759744, + "val_right_hand_mae": 0.029748739674687386, + "val_right_hand_r2": 0.9396049380302429, + "val_left_hip_mse": 0.0007148400763981044, + "val_left_hip_rmse": 0.026736493345203376, + "val_left_hip_mae": 0.02058982476592064, + "val_left_hip_r2": 0.9052451848983765, + "val_right_hip_mse": 0.0007162583060562611, + "val_right_hip_rmse": 0.026763002560554767, + "val_right_hip_mae": 0.020733479410409927, + "val_right_hip_r2": 0.9086782932281494, + "val_left_knee_mse": 0.0007693525985814631, + "val_left_knee_rmse": 0.027737206034160383, + "val_left_knee_mae": 0.019984953105449677, + "val_left_knee_r2": 0.9121326804161072, + "val_right_knee_mse": 0.000785896263550967, + "val_right_knee_rmse": 0.028033841398405732, + "val_right_knee_mae": 0.020405160263180733, + "val_right_knee_r2": 0.920263946056366, + "val_left_foot_mse": 0.0012625186936929822, + "val_left_foot_rmse": 0.03553193906463567, + "val_left_foot_mae": 0.02177722565829754, + "val_left_foot_r2": 0.738381028175354, + "val_right_foot_mse": 0.0013318737037479877, + "val_right_foot_rmse": 0.03649484489277887, + "val_right_foot_mae": 0.022931654006242752, + "val_right_foot_r2": 0.6828205585479736 + }, + "best_epoch": 100 + } + }, + "fold_details": [ + { + "fold": 1, + "train_metrics": { + "train_mse": 0.0008665046188980341, + "train_rmse": 0.029436450514592177, + "train_mae": 0.02175077795982361, + "train_r2": 0.9592944979667664, + "train_head_mse": 0.0009298729128204286, + "train_head_rmse": 0.030493817616369857, + "train_head_mae": 0.023378323763608932, + "train_head_r2": 0.9079026579856873, + "train_left_shoulder_mse": 0.0005884800921194255, + "train_left_shoulder_rmse": 0.024258608618785735, + "train_left_shoulder_mae": 0.018525516614317894, + "train_left_shoulder_r2": 0.8958324193954468, + "train_left_elbow_mse": 0.0009074152912944555, + "train_left_elbow_rmse": 0.030123334664250826, + "train_left_elbow_mae": 0.023107698187232018, + "train_left_elbow_r2": 0.9342540502548218, + "train_right_shoulder_mse": 0.0005157479317858815, + "train_right_shoulder_rmse": 0.022710084363248884, + "train_right_shoulder_mae": 0.017532845959067345, + "train_right_shoulder_r2": 0.9219505786895752, + "train_right_elbow_mse": 0.0008220644667744637, + "train_right_elbow_rmse": 0.028671666620105354, + "train_right_elbow_mae": 0.022041283547878265, + "train_right_elbow_r2": 0.9483591318130493, + "train_left_hand_mse": 0.0013746485346928239, + "train_left_hand_rmse": 0.03707625297535909, + "train_left_hand_mae": 0.028885718435049057, + "train_left_hand_r2": 0.9472445249557495, + "train_right_hand_mse": 0.0014922658447176218, + "train_right_hand_rmse": 0.03862985690780671, + "train_right_hand_mae": 0.029225442558526993, + "train_right_hand_r2": 0.9466012716293335, + "train_left_hip_mse": 0.0006816406385041773, + "train_left_hip_rmse": 0.026108248476375762, + "train_left_hip_mae": 0.019838787615299225, + "train_left_hip_r2": 0.9147313833236694, + "train_right_hip_mse": 0.0006630380521528423, + "train_right_hip_rmse": 0.02574952528014531, + "train_right_hip_mae": 0.019360071048140526, + "train_right_hip_r2": 0.9201050996780396, + "train_left_knee_mse": 0.0007442144560627639, + "train_left_knee_rmse": 0.027280294281088024, + "train_left_knee_mae": 0.020114853978157043, + "train_left_knee_r2": 0.9221558570861816, + "train_right_knee_mse": 0.0006914628902450204, + "train_right_knee_rmse": 0.026295681969574783, + "train_right_knee_mae": 0.01960083842277527, + "train_right_knee_r2": 0.9357047080993652, + "train_left_foot_mse": 0.0008882460533641279, + "train_left_foot_rmse": 0.029803457070684398, + "train_left_foot_mae": 0.0196837130934, + "train_left_foot_r2": 0.7983726263046265, + "train_right_foot_mse": 0.0009654625901021063, + "train_right_foot_rmse": 0.03107189389306848, + "train_right_foot_mae": 0.02146502211689949, + "train_right_foot_r2": 0.7824999690055847 + }, + "val_metrics": { + "val_mse": 0.0010792264947667718, + "val_rmse": 0.032851582835028995, + "val_mae": 0.02367297001183033, + "val_r2": 0.9474682211875916, + "val_head_mse": 0.0011019998928532004, + "val_head_rmse": 0.0331963837315633, + "val_head_mae": 0.02436153218150139, + "val_head_r2": 0.8819195628166199, + "val_left_shoulder_mse": 0.0007106499979272485, + "val_left_shoulder_rmse": 0.026658019392431398, + "val_left_shoulder_mae": 0.019468018785119057, + "val_left_shoulder_r2": 0.8629467487335205, + "val_left_elbow_mse": 0.0010314153041690588, + "val_left_elbow_rmse": 0.03211565512595156, + "val_left_elbow_mae": 0.024876713752746582, + "val_left_elbow_r2": 0.9252476692199707, + "val_right_shoulder_mse": 0.0006507080979645252, + "val_right_shoulder_rmse": 0.025508980731587947, + "val_right_shoulder_mae": 0.019364558160305023, + "val_right_shoulder_r2": 0.8947156667709351, + "val_right_elbow_mse": 0.0010605755960568786, + "val_right_elbow_rmse": 0.032566479638684906, + "val_right_elbow_mae": 0.025204740464687347, + "val_right_elbow_r2": 0.9301803708076477, + "val_left_hand_mse": 0.0016170800663530827, + "val_left_hand_rmse": 0.040212934067947376, + "val_left_hand_mae": 0.03188423439860344, + "val_left_hand_r2": 0.9404428601264954, + "val_right_hand_mse": 0.0018234503222629428, + "val_right_hand_rmse": 0.04270187726860428, + "val_right_hand_mae": 0.0323280468583107, + "val_right_hand_r2": 0.9336005449295044, + "val_left_hip_mse": 0.0007856901502236724, + "val_left_hip_rmse": 0.028030165005288005, + "val_left_hip_mae": 0.02080843225121498, + "val_left_hip_r2": 0.8879433870315552, + "val_right_hip_mse": 0.0007942025549709797, + "val_right_hip_rmse": 0.02818159958148188, + "val_right_hip_mae": 0.020293118432164192, + "val_right_hip_r2": 0.8923852443695068, + "val_left_knee_mse": 0.0009185639792121947, + "val_left_knee_rmse": 0.030307820429918655, + "val_left_knee_mae": 0.02161259576678276, + "val_left_knee_r2": 0.8999753594398499, + "val_right_knee_mse": 0.00098401156719774, + "val_right_knee_rmse": 0.031368958656572264, + "val_right_knee_mae": 0.022405097261071205, + "val_right_knee_r2": 0.9084893465042114, + "val_left_foot_mse": 0.0012650052085518837, + "val_left_foot_rmse": 0.035566911709507244, + "val_left_foot_mae": 0.021658295765519142, + "val_left_foot_r2": 0.702069878578186, + "val_right_foot_mse": 0.0012865925673395395, + "val_right_foot_rmse": 0.03586910324136275, + "val_right_foot_mae": 0.023483235388994217, + "val_right_foot_r2": 0.7242964506149292 + }, + "best_epoch": 100 + }, + { + "fold": 2, + "train_metrics": { + "train_mse": 0.0007965312106534839, + "train_rmse": 0.02822288452042923, + "train_mae": 0.02077263966202736, + "train_r2": 0.9620233178138733, + "train_head_mse": 0.0008216450805775821, + "train_head_rmse": 0.028664352087175844, + "train_head_mae": 0.02182442881166935, + "train_head_r2": 0.9159616231918335, + "train_left_shoulder_mse": 0.0005548224435187876, + "train_left_shoulder_rmse": 0.02355466925088925, + "train_left_shoulder_mae": 0.01789220981299877, + "train_left_shoulder_r2": 0.8996976613998413, + "train_left_elbow_mse": 0.000821933033876121, + "train_left_elbow_rmse": 0.028669374493980875, + "train_left_elbow_mae": 0.021955404430627823, + "train_left_elbow_r2": 0.9408754110336304, + "train_right_shoulder_mse": 0.00048643999616615474, + "train_right_shoulder_rmse": 0.022055384743099694, + "train_right_shoulder_mae": 0.017074357718229294, + "train_right_shoulder_r2": 0.9242039918899536, + "train_right_elbow_mse": 0.0007746925693936646, + "train_right_elbow_rmse": 0.027833299649766007, + "train_right_elbow_mae": 0.02143443375825882, + "train_right_elbow_r2": 0.950498640537262, + "train_left_hand_mse": 0.0012506804196164012, + "train_left_hand_rmse": 0.03536496033670052, + "train_left_hand_mae": 0.02750825323164463, + "train_left_hand_r2": 0.9526473879814148, + "train_right_hand_mse": 0.001332068582996726, + "train_right_hand_rmse": 0.03649751475096251, + "train_right_hand_mae": 0.027385566383600235, + "train_right_hand_r2": 0.9519290924072266, + "train_left_hip_mse": 0.0006687247077934444, + "train_left_hip_rmse": 0.02585971205936842, + "train_left_hip_mae": 0.019806906580924988, + "train_left_hip_r2": 0.9114747643470764, + "train_right_hip_mse": 0.000658244825899601, + "train_right_hip_rmse": 0.025656282386573488, + "train_right_hip_mae": 0.01941586658358574, + "train_right_hip_r2": 0.915770411491394, + "train_left_knee_mse": 0.0006730671157129109, + "train_left_knee_rmse": 0.025943537070201337, + "train_left_knee_mae": 0.0188412144780159, + "train_left_knee_r2": 0.9275770783424377, + "train_right_knee_mse": 0.0006546431104652584, + "train_right_knee_rmse": 0.02558599442009746, + "train_right_knee_mae": 0.0189324039965868, + "train_right_knee_r2": 0.9392065405845642, + "train_left_foot_mse": 0.0008407228742726147, + "train_left_foot_rmse": 0.028995221576539377, + "train_left_foot_mae": 0.01848563924431801, + "train_left_foot_r2": 0.8053251504898071, + "train_right_foot_mse": 0.0008172208908945322, + "train_right_foot_rmse": 0.028587075591856755, + "train_right_foot_mae": 0.019487636163830757, + "train_right_foot_r2": 0.8211026191711426 + }, + "val_metrics": { + "val_mse": 0.001053999993018806, + "val_rmse": 0.03246536605397829, + "val_mae": 0.023614903911948204, + "val_r2": 0.9516116380691528, + "val_head_mse": 0.0011140300193801522, + "val_head_rmse": 0.0333770882399911, + "val_head_mae": 0.024902120232582092, + "val_head_r2": 0.8950873017311096, + "val_left_shoulder_mse": 0.000695552269462496, + "val_left_shoulder_rmse": 0.02637332496031731, + "val_left_shoulder_mae": 0.019979014992713928, + "val_left_shoulder_r2": 0.8769413232803345, + "val_left_elbow_mse": 0.001100413384847343, + "val_left_elbow_rmse": 0.03317247932921721, + "val_left_elbow_mae": 0.025269215926527977, + "val_left_elbow_r2": 0.9178162217140198, + "val_right_shoulder_mse": 0.0006860295543447137, + "val_right_shoulder_rmse": 0.02619216589640333, + "val_right_shoulder_mae": 0.01987605355679989, + "val_right_shoulder_r2": 0.9010257124900818, + "val_right_elbow_mse": 0.0011187803465873003, + "val_right_elbow_rmse": 0.03344817403965873, + "val_right_elbow_mae": 0.025529494509100914, + "val_right_elbow_r2": 0.9310513138771057, + "val_left_hand_mse": 0.0018466977635398507, + "val_left_hand_rmse": 0.04297322147035117, + "val_left_hand_mae": 0.032440185546875, + "val_left_hand_r2": 0.9281805753707886, + "val_right_hand_mse": 0.0019435046706348658, + "val_right_hand_rmse": 0.044085197863170195, + "val_right_hand_mae": 0.03186493739485741, + "val_right_hand_r2": 0.9314125180244446, + "val_left_hip_mse": 0.0007960660150274634, + "val_left_hip_rmse": 0.028214641855381816, + "val_left_hip_mae": 0.021368002519011497, + "val_left_hip_r2": 0.909112811088562, + "val_right_hip_mse": 0.0008345854002982378, + "val_right_hip_rmse": 0.028889191755710954, + "val_right_hip_mae": 0.0212212223559618, + "val_right_hip_r2": 0.9102905988693237, + "val_left_knee_mse": 0.0008981445571407676, + "val_left_knee_rmse": 0.029969059997617002, + "val_left_knee_mae": 0.021922744810581207, + "val_left_knee_r2": 0.9121755361557007, + "val_right_knee_mse": 0.000849513104185462, + "val_right_knee_rmse": 0.02914640808376672, + "val_right_knee_mae": 0.02125907503068447, + "val_right_knee_r2": 0.9206178188323975, + "val_left_foot_mse": 0.0008227587677538395, + "val_left_foot_rmse": 0.028683771853677813, + "val_left_foot_mae": 0.02032771334052086, + "val_left_foot_r2": 0.820340096950531, + "val_right_foot_mse": 0.0009959248127415776, + "val_right_foot_rmse": 0.031558276453912654, + "val_right_foot_mae": 0.02103395201265812, + "val_right_foot_r2": 0.7597413063049316 + }, + "best_epoch": 100 + }, + { + "fold": 3, + "train_metrics": { + "train_mse": 0.0006911249365657568, + "train_rmse": 0.026289255154259446, + "train_mae": 0.019673969596624374, + "train_r2": 0.9669402837753296, + "train_head_mse": 0.0007354658446274698, + "train_head_rmse": 0.027119473531532094, + "train_head_mae": 0.02055048942565918, + "train_head_r2": 0.9252636432647705, + "train_left_shoulder_mse": 0.0004931440926156938, + "train_left_shoulder_rmse": 0.022206847876627918, + "train_left_shoulder_mae": 0.017034156247973442, + "train_left_shoulder_r2": 0.9094990491867065, + "train_left_elbow_mse": 0.0007223921711556613, + "train_left_elbow_rmse": 0.02687735424396645, + "train_left_elbow_mae": 0.020656827837228775, + "train_left_elbow_r2": 0.9471733570098877, + "train_right_shoulder_mse": 0.00043914903653785586, + "train_right_shoulder_rmse": 0.020955883100882575, + "train_right_shoulder_mae": 0.01628381572663784, + "train_right_shoulder_r2": 0.9317687153816223, + "train_right_elbow_mse": 0.0006791710620746017, + "train_right_elbow_rmse": 0.026060910614838492, + "train_right_elbow_mae": 0.02012937143445015, + "train_right_elbow_r2": 0.9562802910804749, + "train_left_hand_mse": 0.0010725700994953513, + "train_left_hand_rmse": 0.03275011602262427, + "train_left_hand_mae": 0.02594558522105217, + "train_left_hand_r2": 0.9587904214859009, + "train_right_hand_mse": 0.0011134594678878784, + "train_right_hand_rmse": 0.03336854009224675, + "train_right_hand_mae": 0.02570994757115841, + "train_right_hand_r2": 0.9593221545219421, + "train_left_hip_mse": 0.0005497487145476043, + "train_left_hip_rmse": 0.02344672076320278, + "train_left_hip_mae": 0.01793663017451763, + "train_left_hip_r2": 0.9291490316390991, + "train_right_hip_mse": 0.0005531455390155315, + "train_right_hip_rmse": 0.023519046303273682, + "train_right_hip_mae": 0.01805630512535572, + "train_right_hip_r2": 0.9317746162414551, + "train_left_knee_mse": 0.0005675407592207193, + "train_left_knee_rmse": 0.023823113969855397, + "train_left_knee_mae": 0.01779811456799507, + "train_left_knee_r2": 0.9397196173667908, + "train_right_knee_mse": 0.0005514470976777375, + "train_right_knee_rmse": 0.02348291075820324, + "train_right_knee_mae": 0.01821477711200714, + "train_right_knee_r2": 0.9484400153160095, + "train_left_foot_mse": 0.0007864419021643698, + "train_left_foot_rmse": 0.028043571494450734, + "train_left_foot_mae": 0.018684297800064087, + "train_left_foot_r2": 0.815445601940155, + "train_right_foot_mse": 0.0007209485629573464, + "train_right_foot_rmse": 0.026850485339325738, + "train_right_foot_mae": 0.01876131072640419, + "train_right_foot_r2": 0.8378206491470337 + }, + "val_metrics": { + "val_mse": 0.0010196610819548368, + "val_rmse": 0.031932132436698256, + "val_mae": 0.022872015833854675, + "val_r2": 0.9538207054138184, + "val_head_mse": 0.0010197964729741216, + "val_head_rmse": 0.031934252347191744, + "val_head_mae": 0.023601405322551727, + "val_head_r2": 0.9017829895019531, + "val_left_shoulder_mse": 0.0007085168035700917, + "val_left_shoulder_rmse": 0.02661797895352109, + "val_left_shoulder_mae": 0.01968163251876831, + "val_left_shoulder_r2": 0.8819140195846558, + "val_left_elbow_mse": 0.0010691593633964658, + "val_left_elbow_rmse": 0.03269800243740381, + "val_left_elbow_mae": 0.024970965459942818, + "val_left_elbow_r2": 0.9253672361373901, + "val_right_shoulder_mse": 0.0006139228935353458, + "val_right_shoulder_rmse": 0.02477746745604453, + "val_right_shoulder_mae": 0.01881948858499527, + "val_right_shoulder_r2": 0.9107195138931274, + "val_right_elbow_mse": 0.0010436192387714982, + "val_right_elbow_rmse": 0.032305096173382584, + "val_right_elbow_mae": 0.024807292968034744, + "val_right_elbow_r2": 0.9376415610313416, + "val_left_hand_mse": 0.0016299528069794178, + "val_left_hand_rmse": 0.040372674013240906, + "val_left_hand_mae": 0.03101324662566185, + "val_left_hand_r2": 0.9402915239334106, + "val_right_hand_mse": 0.0018642813665792346, + "val_right_hand_rmse": 0.0431773246806612, + "val_right_hand_mae": 0.03184603899717331, + "val_right_hand_r2": 0.9373539090156555, + "val_left_hip_mse": 0.0007475194870494306, + "val_left_hip_rmse": 0.027340802604339007, + "val_left_hip_mae": 0.02009522169828415, + "val_left_hip_r2": 0.9059396386146545, + "val_right_hip_mse": 0.0007146793068386614, + "val_right_hip_rmse": 0.02673348661956875, + "val_right_hip_mae": 0.020072555169463158, + "val_right_hip_r2": 0.9122166633605957, + "val_left_knee_mse": 0.0008698555757291615, + "val_left_knee_rmse": 0.029493314085215338, + "val_left_knee_mae": 0.02052437886595726, + "val_left_knee_r2": 0.9110141396522522, + "val_right_knee_mse": 0.0007127727149054408, + "val_right_knee_rmse": 0.02669780355957098, + "val_right_knee_mae": 0.019705740734934807, + "val_right_knee_r2": 0.9351944923400879, + "val_left_foot_mse": 0.0013576565543189645, + "val_left_foot_rmse": 0.03684639133373802, + "val_left_foot_mae": 0.021604036912322044, + "val_left_foot_r2": 0.7172290086746216, + "val_right_foot_mse": 0.0009038627031259239, + "val_right_foot_rmse": 0.030064309456994415, + "val_right_foot_mae": 0.020594190806150436, + "val_right_foot_r2": 0.8050189018249512 + }, + "best_epoch": 100 + }, + { + "fold": 4, + "train_metrics": { + "train_mse": 0.0007396690198220313, + "train_rmse": 0.027196856800410434, + "train_mae": 0.020389899611473083, + "train_r2": 0.9652577638626099, + "train_head_mse": 0.0008147882181219757, + "train_head_rmse": 0.028544495408431653, + "train_head_mae": 0.02170136384665966, + "train_head_r2": 0.9181814789772034, + "train_left_shoulder_mse": 0.0005042082630097866, + "train_left_shoulder_rmse": 0.022454582227460537, + "train_left_shoulder_mae": 0.017156267538666725, + "train_left_shoulder_r2": 0.9097864031791687, + "train_left_elbow_mse": 0.0007951081497594714, + "train_left_elbow_rmse": 0.028197662132869658, + "train_left_elbow_mae": 0.02168133482336998, + "train_left_elbow_r2": 0.9423955678939819, + "train_right_shoulder_mse": 0.0004380958271212876, + "train_right_shoulder_rmse": 0.020930738809733583, + "train_right_shoulder_mae": 0.016288021579384804, + "train_right_shoulder_r2": 0.933057427406311, + "train_right_elbow_mse": 0.0007245641900226474, + "train_right_elbow_rmse": 0.026917730031015753, + "train_right_elbow_mae": 0.02116497792303562, + "train_right_elbow_r2": 0.9540191292762756, + "train_left_hand_mse": 0.0010757292620837688, + "train_left_hand_rmse": 0.0327983118785673, + "train_left_hand_mae": 0.02558828331530094, + "train_left_hand_r2": 0.9593132734298706, + "train_right_hand_mse": 0.0012341623660176992, + "train_right_hand_rmse": 0.035130647105023546, + "train_right_hand_mae": 0.02724386751651764, + "train_right_hand_r2": 0.9558611512184143, + "train_left_hip_mse": 0.000608258240390569, + "train_left_hip_rmse": 0.024662891971351797, + "train_left_hip_mae": 0.0188661627471447, + "train_left_hip_r2": 0.9226157069206238, + "train_right_hip_mse": 0.00060211744857952, + "train_right_hip_rmse": 0.02453808159941441, + "train_right_hip_mae": 0.018776990473270416, + "train_right_hip_r2": 0.9264137744903564, + "train_left_knee_mse": 0.000623013882432133, + "train_left_knee_rmse": 0.02496024604109769, + "train_left_knee_mae": 0.01885506883263588, + "train_left_knee_r2": 0.9355669021606445, + "train_right_knee_mse": 0.0006075026467442513, + "train_right_knee_rmse": 0.024647568779582525, + "train_right_knee_mae": 0.019012607634067535, + "train_right_knee_r2": 0.9446756839752197, + "train_left_foot_mse": 0.0008118184050545096, + "train_left_foot_rmse": 0.02849242715274551, + "train_left_foot_mae": 0.018963536247611046, + "train_left_foot_r2": 0.809372067451477, + "train_right_foot_mse": 0.0007763301837258041, + "train_right_foot_rmse": 0.027862702376578695, + "train_right_foot_mae": 0.019770218059420586, + "train_right_foot_r2": 0.8295915126800537 + }, + "val_metrics": { + "val_mse": 0.0010027959942817688, + "val_rmse": 0.031666954294370796, + "val_mae": 0.02263071946799755, + "val_r2": 0.951167106628418, + "val_head_mse": 0.0009305973071604967, + "val_head_rmse": 0.03050569302868723, + "val_head_mae": 0.023090656846761703, + "val_head_r2": 0.9061184525489807, + "val_left_shoulder_mse": 0.0006306541035883129, + "val_left_shoulder_rmse": 0.025112827471002002, + "val_left_shoulder_mae": 0.018864132463932037, + "val_left_shoulder_r2": 0.8839927911758423, + "val_left_elbow_mse": 0.0011112610809504986, + "val_left_elbow_rmse": 0.0333355828050223, + "val_left_elbow_mae": 0.02487783506512642, + "val_left_elbow_r2": 0.9195311069488525, + "val_right_shoulder_mse": 0.0005033456836827099, + "val_right_shoulder_rmse": 0.022435366805174144, + "val_right_shoulder_mae": 0.017676031216979027, + "val_right_shoulder_r2": 0.9218384027481079, + "val_right_elbow_mse": 0.0009742524125613272, + "val_right_elbow_rmse": 0.031213016716769418, + "val_right_elbow_mae": 0.023107292130589485, + "val_right_elbow_r2": 0.9383873343467712, + "val_left_hand_mse": 0.0016541049117222428, + "val_left_hand_rmse": 0.04067068860644288, + "val_left_hand_mae": 0.030412372201681137, + "val_left_hand_r2": 0.9355151057243347, + "val_right_hand_mse": 0.0016513929003849626, + "val_right_hand_rmse": 0.040637333824759744, + "val_right_hand_mae": 0.029748739674687386, + "val_right_hand_r2": 0.9396049380302429, + "val_left_hip_mse": 0.0007148400763981044, + "val_left_hip_rmse": 0.026736493345203376, + "val_left_hip_mae": 0.02058982476592064, + "val_left_hip_r2": 0.9052451848983765, + "val_right_hip_mse": 0.0007162583060562611, + "val_right_hip_rmse": 0.026763002560554767, + "val_right_hip_mae": 0.020733479410409927, + "val_right_hip_r2": 0.9086782932281494, + "val_left_knee_mse": 0.0007693525985814631, + "val_left_knee_rmse": 0.027737206034160383, + "val_left_knee_mae": 0.019984953105449677, + "val_left_knee_r2": 0.9121326804161072, + "val_right_knee_mse": 0.000785896263550967, + "val_right_knee_rmse": 0.028033841398405732, + "val_right_knee_mae": 0.020405160263180733, + "val_right_knee_r2": 0.920263946056366, + "val_left_foot_mse": 0.0012625186936929822, + "val_left_foot_rmse": 0.03553193906463567, + "val_left_foot_mae": 0.02177722565829754, + "val_left_foot_r2": 0.738381028175354, + "val_right_foot_mse": 0.0013318737037479877, + "val_right_foot_rmse": 0.03649484489277887, + "val_right_foot_mae": 0.022931654006242752, + "val_right_foot_r2": 0.6828205585479736 + }, + "best_epoch": 100 + }, + { + "fold": 5, + "train_metrics": { + "train_mse": 0.0008885739953257143, + "train_rmse": 0.029808958306618404, + "train_mae": 0.022057868540287018, + "train_r2": 0.9581728577613831, + "train_head_mse": 0.0009432259248569608, + "train_head_rmse": 0.030711983408060133, + "train_head_mae": 0.023310711607336998, + "train_head_r2": 0.9063546061515808, + "train_left_shoulder_mse": 0.0006008069613017142, + "train_left_shoulder_rmse": 0.024511363921693834, + "train_left_shoulder_mae": 0.018824709579348564, + "train_left_shoulder_r2": 0.8922619819641113, + "train_left_elbow_mse": 0.0009185626986436546, + "train_left_elbow_rmse": 0.030307799303869863, + "train_left_elbow_mae": 0.02320815436542034, + "train_left_elbow_r2": 0.9336899518966675, + "train_right_shoulder_mse": 0.0005142033332958817, + "train_right_shoulder_rmse": 0.022676051977711678, + "train_right_shoulder_mae": 0.01766814850270748, + "train_right_shoulder_r2": 0.9222487211227417, + "train_right_elbow_mse": 0.000834152102470398, + "train_right_elbow_rmse": 0.02888169147523043, + "train_right_elbow_mae": 0.022329961881041527, + "train_right_elbow_r2": 0.9479066729545593, + "train_left_hand_mse": 0.0013370973756536841, + "train_left_hand_rmse": 0.03656634211476018, + "train_left_hand_mae": 0.028532598167657852, + "train_left_hand_r2": 0.949540913105011, + "train_right_hand_mse": 0.0014505028957501054, + "train_right_hand_rmse": 0.038085468301572785, + "train_right_hand_mae": 0.028638482093811035, + "train_right_hand_r2": 0.9486858248710632, + "train_left_hip_mse": 0.0007058308110572398, + "train_left_hip_rmse": 0.02656747656547832, + "train_left_hip_mae": 0.020249051973223686, + "train_left_hip_r2": 0.9097293019294739, + "train_right_hip_mse": 0.0006879241554997861, + "train_right_hip_rmse": 0.026228308285129374, + "train_right_hip_mae": 0.019810516387224197, + "train_right_hip_r2": 0.9158053994178772, + "train_left_knee_mse": 0.0007861581398174167, + "train_left_knee_rmse": 0.028038511726149386, + "train_left_knee_mae": 0.020462248474359512, + "train_left_knee_r2": 0.9172049760818481, + "train_right_knee_mse": 0.0007521010120399296, + "train_right_knee_rmse": 0.02742446010480297, + "train_right_knee_mae": 0.020508185029029846, + "train_right_knee_r2": 0.9289417266845703, + "train_left_foot_mse": 0.0010713881347328424, + "train_left_foot_rmse": 0.03273206584884068, + "train_left_foot_mae": 0.021317316219210625, + "train_left_foot_r2": 0.7681806683540344, + "train_right_foot_mse": 0.0009495089761912823, + "train_right_foot_rmse": 0.030814103527301948, + "train_right_foot_mae": 0.021892227232456207, + "train_right_foot_r2": 0.78483647108078 + }, + "val_metrics": { + "val_mse": 0.001116706058382988, + "val_rmse": 0.033417152158479756, + "val_mae": 0.02475116401910782, + "val_r2": 0.9461219310760498, + "val_head_mse": 0.001146787079051137, + "val_head_rmse": 0.033864244846905076, + "val_head_mae": 0.025583209469914436, + "val_head_r2": 0.8787392377853394, + "val_left_shoulder_mse": 0.0007494837627746165, + "val_left_shoulder_rmse": 0.027376701093715008, + "val_left_shoulder_mae": 0.02108697034418583, + "val_left_shoulder_r2": 0.863502562046051, + "val_left_elbow_mse": 0.0011599162826314569, + "val_left_elbow_rmse": 0.03405754369638916, + "val_left_elbow_mae": 0.026526233181357384, + "val_left_elbow_r2": 0.9148916006088257, + "val_right_shoulder_mse": 0.0007042161887511611, + "val_right_shoulder_rmse": 0.02653707197019221, + "val_right_shoulder_mae": 0.02036418206989765, + "val_right_shoulder_r2": 0.885806679725647, + "val_right_elbow_mse": 0.0011852825991809368, + "val_right_elbow_rmse": 0.03442793341432124, + "val_right_elbow_mae": 0.026410136371850967, + "val_right_elbow_r2": 0.9200608730316162, + "val_left_hand_mse": 0.0017726070946082473, + "val_left_hand_rmse": 0.04210234072600058, + "val_left_hand_mae": 0.033207643777132034, + "val_left_hand_r2": 0.9303450584411621, + "val_right_hand_mse": 0.002023726934567094, + "val_right_hand_rmse": 0.04498585260464776, + "val_right_hand_mae": 0.034169670194387436, + "val_right_hand_r2": 0.9227479696273804, + "val_left_hip_mse": 0.0009307769942097366, + "val_left_hip_rmse": 0.0305086380261351, + "val_left_hip_mae": 0.02282382920384407, + "val_left_hip_r2": 0.8792926073074341, + "val_right_hip_mse": 0.000939620309509337, + "val_right_hip_rmse": 0.03065322673894768, + "val_right_hip_mae": 0.0224858857691288, + "val_right_hip_r2": 0.8809443712234497, + "val_left_knee_mse": 0.0008485885919071734, + "val_left_knee_rmse": 0.02913054396861091, + "val_left_knee_mae": 0.022413846105337143, + "val_left_knee_r2": 0.9102327823638916, + "val_right_knee_mse": 0.0010858444729819894, + "val_right_knee_rmse": 0.03295215429955968, + "val_right_knee_mae": 0.024491718038916588, + "val_right_knee_r2": 0.9051515460014343, + "val_left_foot_mse": 0.0007090501603670418, + "val_left_foot_rmse": 0.026627995800792854, + "val_left_foot_mae": 0.019260820001363754, + "val_left_foot_r2": 0.7901189923286438, + "val_right_foot_mse": 0.0012612795690074563, + "val_right_foot_rmse": 0.03551449801148056, + "val_right_foot_mae": 0.02294098027050495, + "val_right_foot_r2": 0.7347772121429443 + }, + "best_epoch": 93 + } + ], + "test_metrics": { + "test_mse": 0.0009683615062385798, + "test_rmse": 0.031118507455187818, + "test_mae": 0.022788260132074356, + "test_r2": 0.9545918107032776, + "test_head_mse": 0.0009712303872220218, + "test_head_rmse": 0.031164569421412224, + "test_head_mae": 0.02439800836145878, + "test_head_r2": 0.9046740531921387, + "test_left_shoulder_mse": 0.0006193611770868301, + "test_left_shoulder_rmse": 0.02488696801715368, + "test_left_shoulder_mae": 0.018824605271220207, + "test_left_shoulder_r2": 0.8913503885269165, + "test_left_elbow_mse": 0.0009546612272970378, + "test_left_elbow_rmse": 0.030897592580928338, + "test_left_elbow_mae": 0.023651018738746643, + "test_left_elbow_r2": 0.929541289806366, + "test_right_shoulder_mse": 0.0006181217613629997, + "test_right_shoulder_rmse": 0.024862054648861983, + "test_right_shoulder_mae": 0.01879579946398735, + "test_right_shoulder_r2": 0.9071531295776367, + "test_right_elbow_mse": 0.0011235896963626146, + "test_right_elbow_rmse": 0.03351998950421397, + "test_right_elbow_mae": 0.025160331279039383, + "test_right_elbow_r2": 0.928135097026825, + "test_left_hand_mse": 0.0013546826085075736, + "test_left_hand_rmse": 0.036806013211261736, + "test_left_hand_mae": 0.027777988463640213, + "test_left_hand_r2": 0.9482738375663757, + "test_right_hand_mse": 0.0017435638001188636, + "test_right_hand_rmse": 0.04175600316264553, + "test_right_hand_mae": 0.031958967447280884, + "test_right_hand_r2": 0.9380349516868591, + "test_left_hip_mse": 0.0006996557931415737, + "test_left_hip_rmse": 0.026451007412602902, + "test_left_hip_mae": 0.020167279988527298, + "test_left_hip_r2": 0.9206179976463318, + "test_right_hip_mse": 0.000734224624466151, + "test_right_hip_rmse": 0.02709657957134352, + "test_right_hip_mae": 0.02046043798327446, + "test_right_hip_r2": 0.9198757410049438, + "test_left_knee_mse": 0.0008288918761536479, + "test_left_knee_rmse": 0.028790482388345768, + "test_left_knee_mae": 0.02076132781803608, + "test_left_knee_r2": 0.9148035049438477, + "test_right_knee_mse": 0.0007784324116073549, + "test_right_knee_rmse": 0.02790040163881794, + "test_right_knee_mae": 0.02092660404741764, + "test_right_knee_r2": 0.9274083375930786, + "test_left_foot_mse": 0.0011156864929944277, + "test_left_foot_rmse": 0.03340189355402516, + "test_left_foot_mae": 0.02132447622716427, + "test_left_foot_r2": 0.760097861289978, + "test_right_foot_mse": 0.0010465980740264058, + "test_right_foot_rmse": 0.032351168047327225, + "test_right_foot_mae": 0.022040534764528275, + "test_right_foot_r2": 0.8006362318992615 + } +} \ No newline at end of file diff --git a/A9/cv_results_loss_optimizer/conv1d_v3_rmsprop_mse_fold1_best.h5 b/A9/cv_results_loss_optimizer/conv1d_v3_rmsprop_mse_fold1_best.h5 new file mode 100644 index 0000000000000000000000000000000000000000..30bcb71a528da17b56acc4a0985a293495a1ec51 --- /dev/null +++ b/A9/cv_results_loss_optimizer/conv1d_v3_rmsprop_mse_fold1_best.h5 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:56fb39b4b9dbf9c2c767a89dfafec2693162f3516de87228250856b3b5d63a3e +size 2658520 diff --git a/A9/cv_results_loss_optimizer/conv1d_v3_rmsprop_mse_fold2_best.h5 b/A9/cv_results_loss_optimizer/conv1d_v3_rmsprop_mse_fold2_best.h5 new file mode 100644 index 0000000000000000000000000000000000000000..6d0a5344894dfc73e9b99878df39d21e2681187d --- /dev/null +++ b/A9/cv_results_loss_optimizer/conv1d_v3_rmsprop_mse_fold2_best.h5 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:9a5e6144faa0397fa2f30f141c569d8d96a618ce3f4a72fdc544ca951cb9b82c +size 2658520 diff --git a/A9/cv_results_loss_optimizer/conv1d_v3_rmsprop_mse_fold3_best.h5 b/A9/cv_results_loss_optimizer/conv1d_v3_rmsprop_mse_fold3_best.h5 new file mode 100644 index 0000000000000000000000000000000000000000..a763e525b90abd181ecfcbc70b17273cdd4a314b --- /dev/null +++ b/A9/cv_results_loss_optimizer/conv1d_v3_rmsprop_mse_fold3_best.h5 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:3343e8e31c94e518f56207fcf723e1719019901a280072fb7a2aae06a7153143 +size 2658520 diff --git a/A9/cv_results_loss_optimizer/conv1d_v3_rmsprop_mse_fold4_best.h5 b/A9/cv_results_loss_optimizer/conv1d_v3_rmsprop_mse_fold4_best.h5 new file mode 100644 index 0000000000000000000000000000000000000000..68a019025c4805bb8b6af21997f973d710ad3a7e --- /dev/null +++ b/A9/cv_results_loss_optimizer/conv1d_v3_rmsprop_mse_fold4_best.h5 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:73295f77bd4bbffbfe7e28badee7b193d766ef4d08497d27d8b51d008fb820d7 +size 2658520 diff --git a/A9/cv_results_loss_optimizer/conv1d_v3_rmsprop_mse_fold5_best.h5 b/A9/cv_results_loss_optimizer/conv1d_v3_rmsprop_mse_fold5_best.h5 new file mode 100644 index 0000000000000000000000000000000000000000..fecdc1847a9ef9f57592b039cbc8e0284e3022fe --- /dev/null +++ b/A9/cv_results_loss_optimizer/conv1d_v3_rmsprop_mse_fold5_best.h5 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c816d55ed48acd655e7c1111fc6377bda4f5d0ebb61f9e1cc4604b84f84f1959 +size 2658520 diff --git a/A9/cv_results_loss_optimizer/conv1d_v3_rmsprop_mse_predictions.png b/A9/cv_results_loss_optimizer/conv1d_v3_rmsprop_mse_predictions.png new file mode 100644 index 0000000000000000000000000000000000000000..b1f973d9da6bb4f2d46785103d36060241c02131 --- /dev/null +++ b/A9/cv_results_loss_optimizer/conv1d_v3_rmsprop_mse_predictions.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:34d13c6705c6f9ac9b9507865f72ea215aff7887269fa2cdb982a8013b12794a +size 563802 diff --git a/A9/cv_results_loss_optimizer/conv1d_v3_rmsprop_mse_test_predictions.png b/A9/cv_results_loss_optimizer/conv1d_v3_rmsprop_mse_test_predictions.png new file mode 100644 index 0000000000000000000000000000000000000000..4323fd6d2174e7c8a8b7d5b366edad1192f8454e --- /dev/null +++ b/A9/cv_results_loss_optimizer/conv1d_v3_rmsprop_mse_test_predictions.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:78192c970279b025582c4dfe37986c1ad922dae54106e8231fae97eca3dfb83f +size 453171 diff --git a/A9/cv_results_loss_optimizer/conv1d_v3_rmsprop_mse_training_history.png b/A9/cv_results_loss_optimizer/conv1d_v3_rmsprop_mse_training_history.png new file mode 100644 index 0000000000000000000000000000000000000000..872bf5a54775011d8982393b006b3f635bac8a4b --- /dev/null +++ b/A9/cv_results_loss_optimizer/conv1d_v3_rmsprop_mse_training_history.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:d8eba0a537d1832195cd80868a47350158026cfcce97831787163b4e218e3864 +size 374244 diff --git a/A9/cv_results_loss_optimizer/conv1d_v3_sgd_mae_cv_results.json b/A9/cv_results_loss_optimizer/conv1d_v3_sgd_mae_cv_results.json new file mode 100644 index 0000000000000000000000000000000000000000..d7896143ffe791d4c9b2f05e3381566d8c61e284 --- /dev/null +++ b/A9/cv_results_loss_optimizer/conv1d_v3_sgd_mae_cv_results.json @@ -0,0 +1,1243 @@ +{ + "model_name": "conv1d_v3_sgd_mae", + "n_splits": 5, + "epochs": 100, + "batch_size": 32, + "patience": 10, + "aggregate_results": { + "n_folds": 5, + "train_metrics": { + "train_mse": { + "mean": 0.0009816524456255139, + "std": 5.182128516188738e-05 + }, + "train_rmse": { + "mean": 0.031320353448900065, + "std": 0.0008294006640265926 + }, + "train_mae": { + "mean": 0.0212290421128273, + "std": 0.0005848423459028629 + }, + "train_r2": { + "mean": 0.9535727739334107, + "std": 0.002263956241893429 + }, + "train_head_mse": { + "mean": 0.0008658226463012398, + "std": 5.542839284916688e-05 + }, + "train_head_rmse": { + "mean": 0.02940936254420277, + "std": 0.0009550084004252027 + }, + "train_head_mae": { + "mean": 0.02164476700127125, + "std": 0.0006899992888251819 + }, + "train_head_r2": { + "mean": 0.9130115270614624, + "std": 0.004851117503030859 + }, + "train_left_shoulder_mse": { + "mean": 0.0005536125507205724, + "std": 2.9733065638334908e-05 + }, + "train_left_shoulder_rmse": { + "mean": 0.02352054678472806, + "std": 0.0006296266099768571 + }, + "train_left_shoulder_mae": { + "mean": 0.01719004809856415, + "std": 0.00028974680792085924 + }, + "train_left_shoulder_r2": { + "mean": 0.900450325012207, + "std": 0.0044323305933709175 + }, + "train_left_elbow_mse": { + "mean": 0.0008982262457720936, + "std": 6.747200219082733e-05 + }, + "train_left_elbow_rmse": { + "mean": 0.029948964212096207, + "std": 0.00113392609753635 + }, + "train_left_elbow_mae": { + "mean": 0.022139040753245354, + "std": 0.0005680734462630507 + }, + "train_left_elbow_r2": { + "mean": 0.9349658489227295, + "std": 0.004569440096932496 + }, + "train_right_shoulder_mse": { + "mean": 0.0005286659172270447, + "std": 3.98875242180034e-05 + }, + "train_right_shoulder_rmse": { + "mean": 0.022975959507322136, + "std": 0.0008781810433722261 + }, + "train_right_shoulder_mae": { + "mean": 0.016737018525600434, + "std": 0.0004587836313021875 + }, + "train_right_shoulder_r2": { + "mean": 0.9190039873123169, + "std": 0.005500431937969111 + }, + "train_right_elbow_mse": { + "mean": 0.001068951957859099, + "std": 6.750071338962293e-05 + }, + "train_right_elbow_rmse": { + "mean": 0.03267833820872425, + "std": 0.0010383495439071073 + }, + "train_right_elbow_mae": { + "mean": 0.0229312714189291, + "std": 0.0007663019534834774 + }, + "train_right_elbow_r2": { + "mean": 0.9322673916816712, + "std": 0.0036998723756199677 + }, + "train_left_hand_mse": { + "mean": 0.0014045829884707928, + "std": 8.964990832786196e-05 + }, + "train_left_hand_rmse": { + "mean": 0.03745861211903454, + "std": 0.0011980678555540156 + }, + "train_left_hand_mae": { + "mean": 0.028269453346729277, + "std": 0.000677533196630763 + }, + "train_left_hand_r2": { + "mean": 0.9465750336647034, + "std": 0.0032638556821172624 + }, + "train_right_hand_mse": { + "mean": 0.001971399923786521, + "std": 0.00012584320135291386 + }, + "train_right_hand_rmse": { + "mean": 0.044377513159375104, + "std": 0.0014269721707203663 + }, + "train_right_hand_mae": { + "mean": 0.03056452013552189, + "std": 0.0007415377196765824 + }, + "train_right_hand_r2": { + "mean": 0.9292244791984559, + "std": 0.0043240454250346875 + }, + "train_left_hip_mse": { + "mean": 0.000678187794983387, + "std": 3.055466336534993e-05 + }, + "train_left_hip_rmse": { + "mean": 0.026035440421258806, + "std": 0.000586205641793242 + }, + "train_left_hip_mae": { + "mean": 0.018971035629510878, + "std": 0.0005359902694518125 + }, + "train_left_hip_r2": { + "mean": 0.9130264520645142, + "std": 0.003465748467292388 + }, + "train_right_hip_mse": { + "mean": 0.0006672225776128471, + "std": 2.830913801666252e-05 + }, + "train_right_hip_rmse": { + "mean": 0.025824846713659394, + "std": 0.0005476037151468887 + }, + "train_right_hip_mae": { + "mean": 0.01889536716043949, + "std": 0.0004175964094965894 + }, + "train_right_hip_r2": { + "mean": 0.9177745819091797, + "std": 0.003147345200306335 + }, + "train_left_knee_mse": { + "mean": 0.0008299559238366782, + "std": 4.4803133836811646e-05 + }, + "train_left_knee_rmse": { + "mean": 0.028798440202562604, + "std": 0.0007783095374620095 + }, + "train_left_knee_mae": { + "mean": 0.019815708324313164, + "std": 0.0005315716759385143 + }, + "train_left_knee_r2": { + "mean": 0.9124874353408814, + "std": 0.005061178195186833 + }, + "train_right_knee_mse": { + "mean": 0.0009362512733787298, + "std": 6.13604666340401e-05 + }, + "train_right_knee_rmse": { + "mean": 0.030581333692994282, + "std": 0.0010165150940652326 + }, + "train_right_knee_mae": { + "mean": 0.019945769757032394, + "std": 0.0006581417572811086 + }, + "train_right_knee_r2": { + "mean": 0.9129587769508362, + "std": 0.005626920310678423 + }, + "train_left_foot_mse": { + "mean": 0.0012180721620097756, + "std": 0.00012134665778442061 + }, + "train_left_foot_rmse": { + "mean": 0.03485745953283475, + "std": 0.0017405967156592622 + }, + "train_left_foot_mae": { + "mean": 0.01943829096853733, + "std": 0.0010578640601671126 + }, + "train_left_foot_r2": { + "mean": 0.7218832015991211, + "std": 0.021332297998329017 + }, + "train_right_foot_mse": { + "mean": 0.0011405303608626126, + "std": 0.000185385651858909 + }, + "train_right_foot_rmse": { + "mean": 0.033669106128591156, + "std": 0.002630903526220351 + }, + "train_right_foot_mae": { + "mean": 0.019435254484415056, + "std": 0.0014222422204123429 + }, + "train_right_foot_r2": { + "mean": 0.7452970981597901, + "std": 0.04349810837680747 + } + }, + "val_metrics": { + "val_mse": { + "mean": 0.0012695140205323697, + "std": 4.392572457121642e-05 + }, + "val_rmse": { + "mean": 0.03562485971824863, + "std": 0.0006191854225315301 + }, + "val_mae": { + "mean": 0.0239640474319458, + "std": 0.0003834885118341804 + }, + "val_r2": { + "mean": 0.939908790588379, + "std": 0.001838873987600906 + }, + "val_head_mse": { + "mean": 0.0011000689817592501, + "std": 8.724371109035158e-05 + }, + "val_head_rmse": { + "mean": 0.033140204162962664, + "std": 0.0013400932043713216 + }, + "val_head_mae": { + "mean": 0.02410258464515209, + "std": 0.0011348826046913834 + }, + "val_head_r2": { + "mean": 0.8890335083007812, + "std": 0.010779629532236502 + }, + "val_left_shoulder_mse": { + "mean": 0.0007156602456234396, + "std": 8.23109742226724e-05 + }, + "val_left_shoulder_rmse": { + "mean": 0.026706655835138045, + "std": 0.0015539561534761942 + }, + "val_left_shoulder_mae": { + "mean": 0.019291719421744347, + "std": 0.0008816467197095651 + }, + "val_left_shoulder_r2": { + "mean": 0.8710466027259827, + "std": 0.014181071827695613 + }, + "val_left_elbow_mse": { + "mean": 0.0011230704607442022, + "std": 8.042459225989693e-05 + }, + "val_left_elbow_rmse": { + "mean": 0.03349027522745569, + "std": 0.001213229546899261 + }, + "val_left_elbow_mae": { + "mean": 0.025061722472310068, + "std": 0.0007376540885005809 + }, + "val_left_elbow_r2": { + "mean": 0.9184705972671509, + "std": 0.006769921618537548 + }, + "val_right_shoulder_mse": { + "mean": 0.0006960909115150571, + "std": 6.079158957244627e-05 + }, + "val_right_shoulder_rmse": { + "mean": 0.026359279790091276, + "std": 0.0011310528116506026 + }, + "val_right_shoulder_mae": { + "mean": 0.01910279355943203, + "std": 0.0007745212850641678 + }, + "val_right_shoulder_r2": { + "mean": 0.8927608966827393, + "std": 0.01278763068408144 + }, + "val_right_elbow_mse": { + "mean": 0.001403364446014166, + "std": 0.00013280275331460092 + }, + "val_right_elbow_rmse": { + "mean": 0.03742011167769495, + "std": 0.0017605930941032581 + }, + "val_right_elbow_mae": { + "mean": 0.02666374184191227, + "std": 0.000599918176541976 + }, + "val_right_elbow_r2": { + "mean": 0.9108903765678406, + "std": 0.008175452342807494 + }, + "val_left_hand_mse": { + "mean": 0.00191392891574651, + "std": 0.000177828648447242 + }, + "val_left_hand_rmse": { + "mean": 0.04370334615164542, + "std": 0.0019865676167624244 + }, + "val_left_hand_mae": { + "mean": 0.032699509710073474, + "std": 0.0008634607948212756 + }, + "val_left_hand_r2": { + "mean": 0.926901388168335, + "std": 0.00833078258577589 + }, + "val_right_hand_mse": { + "mean": 0.002524533774703741, + "std": 0.00021755501614620612 + }, + "val_right_hand_rmse": { + "mean": 0.050196423132810586, + "std": 0.002202925186105972 + }, + "val_right_hand_mae": { + "mean": 0.035292649269104005, + "std": 0.0012635046994016908 + }, + "val_right_hand_r2": { + "mean": 0.9092629671096801, + "std": 0.006722101572541683 + }, + "val_left_hip_mse": { + "mean": 0.0008738719625398517, + "std": 0.0001461776131475648 + }, + "val_left_hip_rmse": { + "mean": 0.029461822100073476, + "std": 0.0024234275486330692 + }, + "val_left_hip_mae": { + "mean": 0.02097252234816551, + "std": 0.0008555331483840701 + }, + "val_left_hip_r2": { + "mean": 0.8874184250831604, + "std": 0.01915123657565372 + }, + "val_right_hip_mse": { + "mean": 0.0008608708507381379, + "std": 0.0001021691644642922 + }, + "val_right_hip_rmse": { + "mean": 0.029289961932642716, + "std": 0.001723073046181914 + }, + "val_right_hip_mae": { + "mean": 0.020927439257502557, + "std": 0.0006777248044924608 + }, + "val_right_hip_r2": { + "mean": 0.8932270050048828, + "std": 0.01480439046667738 + }, + "val_left_knee_mse": { + "mean": 0.0010506846942007543, + "std": 0.00013743216185422384 + }, + "val_left_knee_rmse": { + "mean": 0.03234711230508652, + "std": 0.0020854303447667604 + }, + "val_left_knee_mae": { + "mean": 0.02211601845920086, + "std": 0.0007835741499896832 + }, + "val_left_knee_r2": { + "mean": 0.8894660472869873, + "std": 0.009848147085121125 + }, + "val_right_knee_mse": { + "mean": 0.0011921534663997591, + "std": 0.00034852679036483645 + }, + "val_right_knee_rmse": { + "mean": 0.03416037308679212, + "std": 0.005022188464297928 + }, + "val_right_knee_mae": { + "mean": 0.022634769231081008, + "std": 0.0016438283014546854 + }, + "val_right_knee_r2": { + "mean": 0.8902383089065552, + "std": 0.027500870270625692 + }, + "val_left_foot_mse": { + "mean": 0.0014242215664125979, + "std": 0.0003399164746863615 + }, + "val_left_foot_rmse": { + "mean": 0.037437501402528044, + "std": 0.004759732676139297 + }, + "val_left_foot_mae": { + "mean": 0.020928798615932463, + "std": 0.001255892746067263 + }, + "val_left_foot_r2": { + "mean": 0.6787849307060242, + "std": 0.04394700117333159 + }, + "val_right_foot_mse": { + "mean": 0.0016251614782959224, + "std": 0.0002496670949963932 + }, + "val_right_foot_rmse": { + "mean": 0.04019434950191007, + "std": 0.003094470296225778 + }, + "val_right_foot_mae": { + "mean": 0.021738357096910476, + "std": 0.0013420973787201247 + }, + "val_right_foot_r2": { + "mean": 0.6383013129234314, + "std": 0.043273304852163304 + } + }, + "best_fold": "1", + "best_fold_metrics": { + "fold": 1, + "train_metrics": { + "train_mse": 0.0010491193970665336, + "train_rmse": 0.03239011264362218, + "train_mae": 0.021728452295064926, + "train_r2": 0.9507158398628235, + "train_head_mse": 0.0009115813300013542, + "train_head_rmse": 0.03019240517085968, + "train_head_mae": 0.02232111059129238, + "train_head_r2": 0.9097143411636353, + "train_left_shoulder_mse": 0.0006004967144690454, + "train_left_shoulder_rmse": 0.024505034471900777, + "train_left_shoulder_mae": 0.017516467720270157, + "train_left_shoulder_r2": 0.8937053680419922, + "train_left_elbow_mse": 0.0009123553172685206, + "train_left_elbow_rmse": 0.030205220033439926, + "train_left_elbow_mae": 0.02229217439889908, + "train_left_elbow_r2": 0.9338961243629456, + "train_right_shoulder_mse": 0.0005671100807376206, + "train_right_shoulder_rmse": 0.02381407316562248, + "train_right_shoulder_mae": 0.017003178596496582, + "train_right_shoulder_r2": 0.9141778349876404, + "train_right_elbow_mse": 0.0011459470260888338, + "train_right_elbow_rmse": 0.03385183933095562, + "train_right_elbow_mae": 0.023299096152186394, + "train_right_elbow_r2": 0.928013265132904, + "train_left_hand_mse": 0.0014412045711651444, + "train_left_hand_rmse": 0.03796320022291515, + "train_left_hand_mae": 0.02881404384970665, + "train_left_hand_r2": 0.9446902871131897, + "train_right_hand_mse": 0.0021300865337252617, + "train_right_hand_rmse": 0.046152860515089, + "train_right_hand_mae": 0.031201163306832314, + "train_right_hand_r2": 0.92377769947052, + "train_left_hip_mse": 0.000716088805347681, + "train_left_hip_rmse": 0.026759835674900565, + "train_left_hip_mae": 0.019543487578630447, + "train_left_hip_r2": 0.910422146320343, + "train_right_hip_mse": 0.0006946850917302072, + "train_right_hip_rmse": 0.02635687940045648, + "train_right_hip_mae": 0.0191897414624691, + "train_right_hip_r2": 0.9162916541099548, + "train_left_knee_mse": 0.0008263527415692806, + "train_left_knee_rmse": 0.02874635179582412, + "train_left_knee_mae": 0.019736966118216515, + "train_left_knee_r2": 0.9135642647743225, + "train_right_knee_mse": 0.0010034881997853518, + "train_right_knee_rmse": 0.03167788187024744, + "train_right_knee_mae": 0.019938161596655846, + "train_right_knee_r2": 0.9066911935806274, + "train_left_foot_mse": 0.0011924183927476406, + "train_left_foot_rmse": 0.034531411681940266, + "train_left_foot_mae": 0.019691254943609238, + "train_left_foot_r2": 0.7293270230293274, + "train_right_foot_mse": 0.0014967371243983507, + "train_right_foot_rmse": 0.03868768698692584, + "train_right_foot_mae": 0.021923039108514786, + "train_right_foot_r2": 0.6628140807151794 + }, + "val_metrics": { + "val_mse": 0.001196252298541367, + "val_rmse": 0.03458688043957372, + "val_mae": 0.023554855957627296, + "val_r2": 0.9417719841003418, + "val_head_mse": 0.0011163532035425305, + "val_head_rmse": 0.03341187219451389, + "val_head_mae": 0.023987948894500732, + "val_head_r2": 0.8803815841674805, + "val_left_shoulder_mse": 0.0007288336637429893, + "val_left_shoulder_rmse": 0.02699691952321578, + "val_left_shoulder_mae": 0.018724968656897545, + "val_left_shoulder_r2": 0.8594399094581604, + "val_left_elbow_mse": 0.0009880431462079287, + "val_left_elbow_rmse": 0.03143315361537764, + "val_left_elbow_mae": 0.023851662874221802, + "val_left_elbow_r2": 0.9283910989761353, + "val_right_shoulder_mse": 0.0006760975229553878, + "val_right_shoulder_rmse": 0.026001875373814633, + "val_right_shoulder_mae": 0.018915515393018723, + "val_right_shoulder_r2": 0.8906076550483704, + "val_right_elbow_mse": 0.0012139591854065657, + "val_right_elbow_rmse": 0.03484191707421631, + "val_right_elbow_mae": 0.025725964456796646, + "val_right_elbow_r2": 0.9200828075408936, + "val_left_hand_mse": 0.0017575733363628387, + "val_left_hand_rmse": 0.041923422288296534, + "val_left_hand_mae": 0.03224239498376846, + "val_left_hand_r2": 0.9352684617042542, + "val_right_hand_mse": 0.002157759154215455, + "val_right_hand_rmse": 0.04645168623651304, + "val_right_hand_mae": 0.033848490566015244, + "val_right_hand_r2": 0.9214270114898682, + "val_left_hip_mse": 0.0007608290761709213, + "val_left_hip_rmse": 0.027583130282310624, + "val_left_hip_mae": 0.019774219021201134, + "val_left_hip_r2": 0.8914890885353088, + "val_right_hip_mse": 0.0007761946762911975, + "val_right_hip_rmse": 0.02786027057103354, + "val_right_hip_mae": 0.01984744891524315, + "val_right_hip_r2": 0.8948253393173218, + "val_left_knee_mse": 0.0009606833336874843, + "val_left_knee_rmse": 0.030994892058006658, + "val_left_knee_mae": 0.021149655804038048, + "val_left_knee_r2": 0.8953889012336731, + "val_right_knee_mse": 0.0012086331844329834, + "val_right_knee_rmse": 0.034765402118097, + "val_right_knee_mae": 0.023065073415637016, + "val_right_knee_r2": 0.8876001238822937, + "val_left_foot_mse": 0.001353806583210826, + "val_left_foot_rmse": 0.03679411071368387, + "val_left_foot_mae": 0.02112625725567341, + "val_left_foot_r2": 0.6811556816101074, + "val_right_foot_mse": 0.0018525117775425315, + "val_right_foot_rmse": 0.04304081525183429, + "val_right_foot_mae": 0.0239535141736269, + "val_right_foot_r2": 0.6030257344245911 + }, + "best_epoch": 100 + } + }, + "fold_details": [ + { + "fold": 1, + "train_metrics": { + "train_mse": 0.0010491193970665336, + "train_rmse": 0.03239011264362218, + "train_mae": 0.021728452295064926, + "train_r2": 0.9507158398628235, + "train_head_mse": 0.0009115813300013542, + "train_head_rmse": 0.03019240517085968, + "train_head_mae": 0.02232111059129238, + "train_head_r2": 0.9097143411636353, + "train_left_shoulder_mse": 0.0006004967144690454, + "train_left_shoulder_rmse": 0.024505034471900777, + "train_left_shoulder_mae": 0.017516467720270157, + "train_left_shoulder_r2": 0.8937053680419922, + "train_left_elbow_mse": 0.0009123553172685206, + "train_left_elbow_rmse": 0.030205220033439926, + "train_left_elbow_mae": 0.02229217439889908, + "train_left_elbow_r2": 0.9338961243629456, + "train_right_shoulder_mse": 0.0005671100807376206, + "train_right_shoulder_rmse": 0.02381407316562248, + "train_right_shoulder_mae": 0.017003178596496582, + "train_right_shoulder_r2": 0.9141778349876404, + "train_right_elbow_mse": 0.0011459470260888338, + "train_right_elbow_rmse": 0.03385183933095562, + "train_right_elbow_mae": 0.023299096152186394, + "train_right_elbow_r2": 0.928013265132904, + "train_left_hand_mse": 0.0014412045711651444, + "train_left_hand_rmse": 0.03796320022291515, + "train_left_hand_mae": 0.02881404384970665, + "train_left_hand_r2": 0.9446902871131897, + "train_right_hand_mse": 0.0021300865337252617, + "train_right_hand_rmse": 0.046152860515089, + "train_right_hand_mae": 0.031201163306832314, + "train_right_hand_r2": 0.92377769947052, + "train_left_hip_mse": 0.000716088805347681, + "train_left_hip_rmse": 0.026759835674900565, + "train_left_hip_mae": 0.019543487578630447, + "train_left_hip_r2": 0.910422146320343, + "train_right_hip_mse": 0.0006946850917302072, + "train_right_hip_rmse": 0.02635687940045648, + "train_right_hip_mae": 0.0191897414624691, + "train_right_hip_r2": 0.9162916541099548, + "train_left_knee_mse": 0.0008263527415692806, + "train_left_knee_rmse": 0.02874635179582412, + "train_left_knee_mae": 0.019736966118216515, + "train_left_knee_r2": 0.9135642647743225, + "train_right_knee_mse": 0.0010034881997853518, + "train_right_knee_rmse": 0.03167788187024744, + "train_right_knee_mae": 0.019938161596655846, + "train_right_knee_r2": 0.9066911935806274, + "train_left_foot_mse": 0.0011924183927476406, + "train_left_foot_rmse": 0.034531411681940266, + "train_left_foot_mae": 0.019691254943609238, + "train_left_foot_r2": 0.7293270230293274, + "train_right_foot_mse": 0.0014967371243983507, + "train_right_foot_rmse": 0.03868768698692584, + "train_right_foot_mae": 0.021923039108514786, + "train_right_foot_r2": 0.6628140807151794 + }, + "val_metrics": { + "val_mse": 0.001196252298541367, + "val_rmse": 0.03458688043957372, + "val_mae": 0.023554855957627296, + "val_r2": 0.9417719841003418, + "val_head_mse": 0.0011163532035425305, + "val_head_rmse": 0.03341187219451389, + "val_head_mae": 0.023987948894500732, + "val_head_r2": 0.8803815841674805, + "val_left_shoulder_mse": 0.0007288336637429893, + "val_left_shoulder_rmse": 0.02699691952321578, + "val_left_shoulder_mae": 0.018724968656897545, + "val_left_shoulder_r2": 0.8594399094581604, + "val_left_elbow_mse": 0.0009880431462079287, + "val_left_elbow_rmse": 0.03143315361537764, + "val_left_elbow_mae": 0.023851662874221802, + "val_left_elbow_r2": 0.9283910989761353, + "val_right_shoulder_mse": 0.0006760975229553878, + "val_right_shoulder_rmse": 0.026001875373814633, + "val_right_shoulder_mae": 0.018915515393018723, + "val_right_shoulder_r2": 0.8906076550483704, + "val_right_elbow_mse": 0.0012139591854065657, + "val_right_elbow_rmse": 0.03484191707421631, + "val_right_elbow_mae": 0.025725964456796646, + "val_right_elbow_r2": 0.9200828075408936, + "val_left_hand_mse": 0.0017575733363628387, + "val_left_hand_rmse": 0.041923422288296534, + "val_left_hand_mae": 0.03224239498376846, + "val_left_hand_r2": 0.9352684617042542, + "val_right_hand_mse": 0.002157759154215455, + "val_right_hand_rmse": 0.04645168623651304, + "val_right_hand_mae": 0.033848490566015244, + "val_right_hand_r2": 0.9214270114898682, + "val_left_hip_mse": 0.0007608290761709213, + "val_left_hip_rmse": 0.027583130282310624, + "val_left_hip_mae": 0.019774219021201134, + "val_left_hip_r2": 0.8914890885353088, + "val_right_hip_mse": 0.0007761946762911975, + "val_right_hip_rmse": 0.02786027057103354, + "val_right_hip_mae": 0.01984744891524315, + "val_right_hip_r2": 0.8948253393173218, + "val_left_knee_mse": 0.0009606833336874843, + "val_left_knee_rmse": 0.030994892058006658, + "val_left_knee_mae": 0.021149655804038048, + "val_left_knee_r2": 0.8953889012336731, + "val_right_knee_mse": 0.0012086331844329834, + "val_right_knee_rmse": 0.034765402118097, + "val_right_knee_mae": 0.023065073415637016, + "val_right_knee_r2": 0.8876001238822937, + "val_left_foot_mse": 0.001353806583210826, + "val_left_foot_rmse": 0.03679411071368387, + "val_left_foot_mae": 0.02112625725567341, + "val_left_foot_r2": 0.6811556816101074, + "val_right_foot_mse": 0.0018525117775425315, + "val_right_foot_rmse": 0.04304081525183429, + "val_right_foot_mae": 0.0239535141736269, + "val_right_foot_r2": 0.6030257344245911 + }, + "best_epoch": 100 + }, + { + "fold": 2, + "train_metrics": { + "train_mse": 0.0009895284892991185, + "train_rmse": 0.031456771755841675, + "train_mae": 0.021367806941270828, + "train_r2": 0.9528217315673828, + "train_head_mse": 0.0008623775793239474, + "train_head_rmse": 0.02936626600921451, + "train_head_mae": 0.02145039290189743, + "train_head_r2": 0.9117954969406128, + "train_left_shoulder_mse": 0.0005522686988115311, + "train_left_shoulder_rmse": 0.0235003978436862, + "train_left_shoulder_mae": 0.01719602756202221, + "train_left_shoulder_r2": 0.9001592993736267, + "train_left_elbow_mse": 0.000997199327684939, + "train_left_elbow_rmse": 0.0315784630355079, + "train_left_elbow_mae": 0.02265734039247036, + "train_left_elbow_r2": 0.9282678365707397, + "train_right_shoulder_mse": 0.0005385574768297374, + "train_right_shoulder_rmse": 0.023206841164401016, + "train_right_shoulder_mae": 0.016633126884698868, + "train_right_shoulder_r2": 0.9160831570625305, + "train_right_elbow_mse": 0.001086019561626017, + "train_right_elbow_rmse": 0.03295481090259838, + "train_right_elbow_mae": 0.023132354021072388, + "train_right_elbow_r2": 0.9306054711341858, + "train_left_hand_mse": 0.0015371419722214341, + "train_left_hand_rmse": 0.03920640218410042, + "train_left_hand_mae": 0.029036877676844597, + "train_left_hand_r2": 0.9418015480041504, + "train_right_hand_mse": 0.002016392769291997, + "train_right_hand_rmse": 0.04490426226197238, + "train_right_hand_mae": 0.03135915845632553, + "train_right_hand_r2": 0.9272335767745972, + "train_left_hip_mse": 0.0006699544610455632, + "train_left_hip_rmse": 0.025883478534493066, + "train_left_hip_mae": 0.01916479878127575, + "train_left_hip_r2": 0.9113119840621948, + "train_right_hip_mse": 0.0006557254819199443, + "train_right_hip_rmse": 0.025607137323799867, + "train_right_hip_mae": 0.018894601613283157, + "train_right_hip_r2": 0.9160928130149841, + "train_left_knee_mse": 0.0008577902335673571, + "train_left_knee_rmse": 0.02928805615890814, + "train_left_knee_mae": 0.019839143380522728, + "train_left_knee_r2": 0.9077005982398987, + "train_right_knee_mse": 0.000828287738841027, + "train_right_knee_rmse": 0.028779988513566626, + "train_right_knee_mae": 0.019631966948509216, + "train_right_knee_r2": 0.9230810403823853, + "train_left_foot_mse": 0.0012848444748669863, + "train_left_foot_rmse": 0.03584472729519624, + "train_left_foot_mae": 0.019724778831005096, + "train_left_foot_r2": 0.7024859189987183, + "train_right_foot_mse": 0.0009773101191967726, + "train_right_foot_rmse": 0.03126195961862872, + "train_right_foot_mae": 0.01906089298427105, + "train_right_foot_r2": 0.7860575914382935 + }, + "val_metrics": { + "val_mse": 0.0012675278121605515, + "val_rmse": 0.0356023568343523, + "val_mae": 0.02428676187992096, + "val_r2": 0.9418087005615234, + "val_head_mse": 0.0011739565525203943, + "val_head_rmse": 0.03426304937568159, + "val_head_mae": 0.025566769763827324, + "val_head_r2": 0.8894437551498413, + "val_left_shoulder_mse": 0.0006495834677480161, + "val_left_shoulder_rmse": 0.025486927389311097, + "val_left_shoulder_mae": 0.01919413171708584, + "val_left_shoulder_r2": 0.885074257850647, + "val_left_elbow_mse": 0.001205658889375627, + "val_left_elbow_rmse": 0.034722599116074636, + "val_left_elbow_mae": 0.025608744472265244, + "val_left_elbow_r2": 0.9099559783935547, + "val_right_shoulder_mse": 0.0007106392295099795, + "val_right_shoulder_rmse": 0.026657817418348027, + "val_right_shoulder_mae": 0.01919449493288994, + "val_right_shoulder_r2": 0.8974752426147461, + "val_right_elbow_mse": 0.001391024561598897, + "val_right_elbow_rmse": 0.03729644167476164, + "val_right_elbow_mae": 0.027087567374110222, + "val_right_elbow_r2": 0.9142732620239258, + "val_left_hand_mse": 0.0022487512324005365, + "val_left_hand_rmse": 0.04742099990932853, + "val_left_hand_mae": 0.034232087433338165, + "val_left_hand_r2": 0.9125444293022156, + "val_right_hand_mse": 0.00271914922632277, + "val_right_hand_rmse": 0.05214546218342273, + "val_right_hand_mae": 0.036551203578710556, + "val_right_hand_r2": 0.9040395617485046, + "val_left_hip_mse": 0.0007428837125189602, + "val_left_hip_rmse": 0.027255893170449585, + "val_left_hip_mae": 0.02039889618754387, + "val_left_hip_r2": 0.9151846766471863, + "val_right_hip_mse": 0.000785564596299082, + "val_right_hip_rmse": 0.02802792529423257, + "val_right_hip_mae": 0.020653340965509415, + "val_right_hip_r2": 0.9155598282814026, + "val_left_knee_mse": 0.0012753376504406333, + "val_left_knee_rmse": 0.035711869881604255, + "val_left_knee_mae": 0.023327447474002838, + "val_left_knee_r2": 0.8752919435501099, + "val_right_knee_mse": 0.0008445209241472185, + "val_right_knee_rmse": 0.029060642184012702, + "val_right_knee_mae": 0.021498408168554306, + "val_right_knee_r2": 0.9210842847824097, + "val_left_foot_mse": 0.0014203087193891406, + "val_left_foot_rmse": 0.0376869834211912, + "val_left_foot_mae": 0.021719355136156082, + "val_left_foot_r2": 0.6898573637008667, + "val_right_foot_mse": 0.001310483319684863, + "val_right_foot_rmse": 0.036200598333243926, + "val_right_foot_mae": 0.020695460960268974, + "val_right_foot_r2": 0.683856725692749 + }, + "best_epoch": 100 + }, + { + "fold": 3, + "train_metrics": { + "train_mse": 0.0009062218596227467, + "train_rmse": 0.030103519057125973, + "train_mae": 0.020130189135670662, + "train_r2": 0.9566512107849121, + "train_head_mse": 0.0007668979233130813, + "train_head_rmse": 0.02769292189916191, + "train_head_mae": 0.020515654236078262, + "train_head_r2": 0.9220696091651917, + "train_left_shoulder_mse": 0.0005163403693586588, + "train_left_shoulder_rmse": 0.022723124110884462, + "train_left_shoulder_mae": 0.01668931171298027, + "train_left_shoulder_r2": 0.9052420854568481, + "train_left_elbow_mse": 0.0007861341000534594, + "train_left_elbow_rmse": 0.028038083031003733, + "train_left_elbow_mae": 0.021061135455965996, + "train_left_elbow_r2": 0.942512035369873, + "train_right_shoulder_mse": 0.0004623570421244949, + "train_right_shoulder_rmse": 0.02150248920763582, + "train_right_shoulder_mae": 0.01607375219464302, + "train_right_shoulder_r2": 0.9281628727912903, + "train_right_elbow_mse": 0.0009684793185442686, + "train_right_elbow_rmse": 0.031120400359639793, + "train_right_elbow_mae": 0.021447038277983665, + "train_right_elbow_r2": 0.9376568794250488, + "train_left_hand_mse": 0.0012720144586637616, + "train_left_hand_rmse": 0.03566531170007858, + "train_left_hand_mae": 0.027197400107979774, + "train_left_hand_r2": 0.9511274695396423, + "train_right_hand_mse": 0.001887091901153326, + "train_right_hand_rmse": 0.0434406710486075, + "train_right_hand_mae": 0.029270736500620842, + "train_right_hand_r2": 0.9310591816902161, + "train_left_hip_mse": 0.0006383312866091728, + "train_left_hip_rmse": 0.025265218910770847, + "train_left_hip_mae": 0.018064185976982117, + "train_left_hip_r2": 0.9177325963973999, + "train_right_hip_mse": 0.0006286130519583821, + "train_right_hip_rmse": 0.025072156906783712, + "train_right_hip_mae": 0.018142832443118095, + "train_right_hip_r2": 0.9224663972854614, + "train_left_knee_mse": 0.0007634454523213208, + "train_left_knee_rmse": 0.027630516685746592, + "train_left_knee_mae": 0.018895776942372322, + "train_left_knee_r2": 0.9189119338989258, + "train_right_knee_mse": 0.0009154707076959312, + "train_right_knee_rmse": 0.03025674648232905, + "train_right_knee_mae": 0.018916890025138855, + "train_right_knee_r2": 0.914404034614563, + "train_left_foot_mse": 0.0010378367733210325, + "train_left_foot_rmse": 0.032215474128453124, + "train_left_foot_mae": 0.017923887819051743, + "train_left_foot_r2": 0.7564507722854614, + "train_right_foot_mse": 0.001137872226536274, + "train_right_foot_rmse": 0.03373236171002964, + "train_right_foot_mae": 0.01749385893344879, + "train_right_foot_r2": 0.7440325021743774 + }, + "val_metrics": { + "val_mse": 0.0013216582592576742, + "val_rmse": 0.0363546181283434, + "val_mae": 0.023912468925118446, + "val_r2": 0.9401435852050781, + "val_head_mse": 0.0010805086931213737, + "val_head_rmse": 0.032871092058545506, + "val_head_mae": 0.02387438528239727, + "val_head_r2": 0.8959357738494873, + "val_left_shoulder_mse": 0.0008057650993578136, + "val_left_shoulder_rmse": 0.028386001820577225, + "val_left_shoulder_mae": 0.019976884126663208, + "val_left_shoulder_r2": 0.8657060265541077, + "val_left_elbow_mse": 0.0010952831944450736, + "val_left_elbow_rmse": 0.03309506299200945, + "val_left_elbow_mae": 0.024989934638142586, + "val_left_elbow_r2": 0.9235436916351318, + "val_right_shoulder_mse": 0.0006480159354396164, + "val_right_shoulder_rmse": 0.025456157122386254, + "val_right_shoulder_mae": 0.018660353496670723, + "val_right_shoulder_r2": 0.9057614803314209, + "val_right_elbow_mse": 0.0014033234911039472, + "val_right_elbow_rmse": 0.03746095955930583, + "val_right_elbow_mae": 0.026879465207457542, + "val_right_elbow_r2": 0.9161484837532043, + "val_left_hand_mse": 0.0017699366435408592, + "val_left_hand_rmse": 0.042070614965090056, + "val_left_hand_mae": 0.03170895203948021, + "val_left_hand_r2": 0.9351636171340942, + "val_right_hand_mse": 0.002679985947906971, + "val_right_hand_rmse": 0.051768580702072284, + "val_right_hand_mae": 0.03667253255844116, + "val_right_hand_r2": 0.9099435210227966, + "val_left_hip_mse": 0.001107106334529817, + "val_left_hip_rmse": 0.0332732074578003, + "val_left_hip_mae": 0.021375559270381927, + "val_left_hip_r2": 0.8606928586959839, + "val_right_hip_mse": 0.0009967380901798606, + "val_right_hip_rmse": 0.031571159151666583, + "val_right_hip_mae": 0.02109505422413349, + "val_right_hip_r2": 0.877571702003479, + "val_left_knee_mse": 0.0011428325669839978, + "val_left_knee_rmse": 0.03380580670512091, + "val_left_knee_mae": 0.022245369851589203, + "val_left_knee_r2": 0.8830887079238892, + "val_right_knee_mse": 0.001365385833196342, + "val_right_knee_rmse": 0.036951127630917326, + "val_right_knee_mae": 0.02201123908162117, + "val_right_knee_r2": 0.8758586645126343, + "val_left_foot_mse": 0.001670902594923973, + "val_left_foot_rmse": 0.04087667543873857, + "val_left_foot_mae": 0.020925214514136314, + "val_left_foot_r2": 0.6519865393638611, + "val_right_foot_mse": 0.001415771315805614, + "val_right_foot_rmse": 0.037626736714809775, + "val_right_foot_mae": 0.020447177812457085, + "val_right_foot_r2": 0.6945900321006775 + }, + "best_epoch": 100 + }, + { + "fold": 4, + "train_metrics": { + "train_mse": 0.000942176440730691, + "train_rmse": 0.030694892746688185, + "train_mae": 0.021206451579928398, + "train_r2": 0.9557459950447083, + "train_head_mse": 0.00086357252439484, + "train_head_rmse": 0.02938660450604731, + "train_head_mae": 0.021520577371120453, + "train_head_r2": 0.9132826924324036, + "train_left_shoulder_mse": 0.0005295166629366577, + "train_left_shoulder_rmse": 0.023011229061844083, + "train_left_shoulder_mae": 0.017118854448199272, + "train_left_shoulder_r2": 0.9052581787109375, + "train_left_elbow_mse": 0.0009077684953808784, + "train_left_elbow_rmse": 0.0301291967264459, + "train_left_elbow_mae": 0.02253844402730465, + "train_left_elbow_r2": 0.9342334866523743, + "train_right_shoulder_mse": 0.0005074788932688534, + "train_right_shoulder_rmse": 0.022527292186786528, + "train_right_shoulder_mae": 0.016537319868803024, + "train_right_shoulder_r2": 0.9224554896354675, + "train_right_elbow_mse": 0.001015381538309157, + "train_right_elbow_rmse": 0.03186505198974508, + "train_right_elbow_mae": 0.023125775158405304, + "train_right_elbow_r2": 0.935563862323761, + "train_left_hand_mse": 0.0013473676517605782, + "train_left_hand_rmse": 0.03670650694033114, + "train_left_hand_mae": 0.02780914306640625, + "train_left_hand_r2": 0.9490392208099365, + "train_right_hand_mse": 0.0017743385396897793, + "train_right_hand_rmse": 0.042122898044766335, + "train_right_hand_mae": 0.030374739319086075, + "train_right_hand_r2": 0.9365422129631042, + "train_left_hip_mse": 0.0006556849693879485, + "train_left_hip_rmse": 0.025606346271734054, + "train_left_hip_mae": 0.01869572512805462, + "train_left_hip_r2": 0.916581928730011, + "train_right_hip_mse": 0.0006523378542624414, + "train_right_hip_rmse": 0.02554090550983738, + "train_right_hip_mae": 0.018886437639594078, + "train_right_hip_r2": 0.9202761650085449, + "train_left_knee_mse": 0.000806759053375572, + "train_left_knee_rmse": 0.028403504244645095, + "train_left_knee_mae": 0.02009805478155613, + "train_left_knee_r2": 0.9165636897087097, + "train_right_knee_mse": 0.0009797564707696438, + "train_right_knee_rmse": 0.03130106181537048, + "train_right_knee_mae": 0.020841127261519432, + "train_right_knee_r2": 0.9107751250267029, + "train_left_foot_mse": 0.0011728550307452679, + "train_left_foot_rmse": 0.03424697111782687, + "train_left_foot_mae": 0.01876804605126381, + "train_left_foot_r2": 0.7245949506759644, + "train_right_foot_mse": 0.0010354772675782442, + "train_right_foot_rmse": 0.03217883260123407, + "train_right_foot_mae": 0.019369641318917274, + "train_right_foot_r2": 0.7727074027061462 + }, + "val_metrics": { + "val_mse": 0.0012556669535115361, + "val_rmse": 0.03543539125664533, + "val_mae": 0.02355792000889778, + "val_r2": 0.9388530850410461, + "val_head_mse": 0.0009434018284082413, + "val_head_rmse": 0.0307148470354036, + "val_head_mae": 0.02219989150762558, + "val_head_r2": 0.9048267006874084, + "val_left_shoulder_mse": 0.0005957556422799826, + "val_left_shoulder_rmse": 0.024408106077284705, + "val_left_shoulder_mae": 0.01803910918533802, + "val_left_shoulder_r2": 0.8904122710227966, + "val_left_elbow_mse": 0.0012037858832627535, + "val_left_elbow_rmse": 0.03469561763771836, + "val_left_elbow_mae": 0.02601310983300209, + "val_left_elbow_r2": 0.91283118724823, + "val_right_shoulder_mse": 0.0006388173205778003, + "val_right_shoulder_rmse": 0.025274835718116947, + "val_right_shoulder_mae": 0.01822824403643608, + "val_right_shoulder_r2": 0.9008018374443054, + "val_right_elbow_mse": 0.001630347571335733, + "val_right_elbow_rmse": 0.040377562721587504, + "val_right_elbow_mae": 0.02624301053583622, + "val_right_elbow_r2": 0.896895170211792, + "val_left_hand_mse": 0.0019075099844485521, + "val_left_hand_rmse": 0.04367504990779692, + "val_left_hand_mae": 0.03235270082950592, + "val_left_hand_r2": 0.9256361722946167, + "val_right_hand_mse": 0.00267421524040401, + "val_right_hand_rmse": 0.05171281505008222, + "val_right_hand_mae": 0.03377479314804077, + "val_right_hand_r2": 0.902198076248169, + "val_left_hip_mse": 0.0007733309757895768, + "val_left_hip_rmse": 0.02780882909778074, + "val_left_hip_mae": 0.021026359871029854, + "val_left_hip_r2": 0.897491991519928, + "val_right_hip_mse": 0.000771437247749418, + "val_right_hip_rmse": 0.027774759184364103, + "val_right_hip_mae": 0.02111874893307686, + "val_right_hip_r2": 0.9016430974006653, + "val_left_knee_mse": 0.0009660822688601911, + "val_left_knee_rmse": 0.03108186398625718, + "val_left_knee_mae": 0.02138950489461422, + "val_left_knee_r2": 0.8896642923355103, + "val_right_knee_mse": 0.0008000745438039303, + "val_right_knee_rmse": 0.028285588977497538, + "val_right_knee_mae": 0.020979898050427437, + "val_right_knee_r2": 0.9188254475593567, + "val_left_foot_mse": 0.001836561830714345, + "val_left_foot_rmse": 0.04285512607278556, + "val_left_foot_mae": 0.02227203920483589, + "val_left_foot_r2": 0.6194279193878174, + "val_right_foot_mse": 0.0015823494177311659, + "val_right_foot_rmse": 0.03977875585951836, + "val_right_foot_mae": 0.022615544497966766, + "val_right_foot_r2": 0.6231709122657776 + }, + "best_epoch": 100 + }, + { + "fold": 5, + "train_metrics": { + "train_mse": 0.0010212160414084792, + "train_rmse": 0.0319564710412223, + "train_mae": 0.02171231061220169, + "train_r2": 0.9519290924072266, + "train_head_mse": 0.0009246838744729757, + "train_head_rmse": 0.03040861513573046, + "train_head_mae": 0.022416099905967712, + "train_head_r2": 0.9081954956054688, + "train_left_shoulder_mse": 0.0005694403080269694, + "train_left_shoulder_rmse": 0.02386294843532478, + "train_left_shoulder_mae": 0.01742957904934883, + "train_left_shoulder_r2": 0.8978866934776306, + "train_left_elbow_mse": 0.0008876739884726703, + "train_left_elbow_rmse": 0.029793858234083585, + "train_left_elbow_mae": 0.022146109491586685, + "train_left_elbow_r2": 0.9359197616577148, + "train_right_shoulder_mse": 0.0005678260931745172, + "train_right_shoulder_rmse": 0.02382910181216483, + "train_right_shoulder_mae": 0.017437715083360672, + "train_right_shoulder_r2": 0.9141405820846558, + "train_right_elbow_mse": 0.0011289323447272182, + "train_right_elbow_rmse": 0.033599588460682346, + "train_right_elbow_mae": 0.02365209348499775, + "train_right_elbow_r2": 0.929497480392456, + "train_left_hand_mse": 0.0014251862885430455, + "train_left_hand_rmse": 0.0377516395477474, + "train_left_hand_mae": 0.02848980203270912, + "train_left_hand_r2": 0.9462166428565979, + "train_right_hand_mse": 0.002049089875072241, + "train_right_hand_rmse": 0.0452668739264403, + "train_right_hand_mae": 0.030616803094744682, + "train_right_hand_r2": 0.9275097250938416, + "train_left_hip_mse": 0.0007108794525265694, + "train_left_hip_rmse": 0.026662322714395482, + "train_left_hip_mae": 0.019386980682611465, + "train_left_hip_r2": 0.9090836048126221, + "train_right_hip_mse": 0.0007047514081932604, + "train_right_hip_rmse": 0.02654715442741953, + "train_right_hip_mae": 0.019363222643733025, + "train_right_hip_r2": 0.9137458801269531, + "train_left_knee_mse": 0.0008954321383498609, + "train_left_knee_rmse": 0.029923772127689065, + "train_left_knee_mae": 0.020508600398898125, + "train_left_knee_r2": 0.90569669008255, + "train_right_knee_mse": 0.0009542532498016953, + "train_right_knee_rmse": 0.03089098978345782, + "train_right_knee_mae": 0.020400702953338623, + "train_right_knee_r2": 0.9098424911499023, + "train_left_foot_mse": 0.001402406138367951, + "train_left_foot_rmse": 0.037448713440757224, + "train_left_foot_mae": 0.021083487197756767, + "train_left_foot_r2": 0.696557343006134, + "train_right_foot_mse": 0.0010552550666034222, + "train_right_foot_rmse": 0.03248468972613748, + "train_right_foot_mae": 0.01932884007692337, + "train_right_foot_r2": 0.7608739137649536 + }, + "val_metrics": { + "val_mse": 0.0013064647791907191, + "val_rmse": 0.03614505193232843, + "val_mae": 0.02450823038816452, + "val_r2": 0.936966598033905, + "val_head_mse": 0.001186124631203711, + "val_head_rmse": 0.03444016015066874, + "val_head_mae": 0.024883927777409554, + "val_head_r2": 0.8745797276496887, + "val_left_shoulder_mse": 0.0007983633549883962, + "val_left_shoulder_rmse": 0.028255324365301423, + "val_left_shoulder_mae": 0.02052350342273712, + "val_left_shoulder_r2": 0.8546005487442017, + "val_left_elbow_mse": 0.001122581190429628, + "val_left_elbow_rmse": 0.03350494277609839, + "val_left_elbow_mae": 0.02484516054391861, + "val_left_elbow_r2": 0.9176310300827026, + "val_right_shoulder_mse": 0.0008068845490925014, + "val_right_shoulder_rmse": 0.028405713317790515, + "val_right_shoulder_mae": 0.020515359938144684, + "val_right_shoulder_r2": 0.8691582679748535, + "val_right_elbow_mse": 0.0013781674206256866, + "val_right_elbow_rmse": 0.03712367735860345, + "val_right_elbow_mae": 0.027382701635360718, + "val_right_elbow_r2": 0.9070521593093872, + "val_left_hand_mse": 0.0018858733819797635, + "val_left_hand_rmse": 0.04342664368771507, + "val_left_hand_mae": 0.0329614132642746, + "val_left_hand_r2": 0.9258942604064941, + "val_right_hand_mse": 0.0023915593046694994, + "val_right_hand_rmse": 0.04890357149196262, + "val_right_hand_mae": 0.035616226494312286, + "val_right_hand_r2": 0.9087066650390625, + "val_left_hip_mse": 0.0009852097136899829, + "val_left_hip_rmse": 0.03138805049202615, + "val_left_hip_mae": 0.022287577390670776, + "val_left_hip_r2": 0.872233510017395, + "val_right_hip_mse": 0.0009744196431711316, + "val_right_hip_rmse": 0.031215695461916777, + "val_right_hip_mae": 0.021922603249549866, + "val_right_hip_r2": 0.8765350580215454, + "val_left_knee_mse": 0.0009084876510314643, + "val_left_knee_rmse": 0.030141128894443623, + "val_left_knee_mae": 0.022468114271759987, + "val_left_knee_r2": 0.9038963913917542, + "val_right_knee_mse": 0.0017421528464183211, + "val_right_knee_rmse": 0.041739104523436064, + "val_right_knee_mae": 0.025619227439165115, + "val_right_knee_r2": 0.8478230237960815, + "val_left_foot_mse": 0.0008395281038247049, + "val_left_foot_rmse": 0.02897461136624105, + "val_left_foot_mae": 0.018601126968860626, + "val_left_foot_r2": 0.7514971494674683, + "val_right_foot_mse": 0.001964691560715437, + "val_right_foot_rmse": 0.044324841350144015, + "val_right_foot_mae": 0.02098008804023266, + "val_right_foot_r2": 0.5868631601333618 + }, + "best_epoch": 100 + } + ], + "test_metrics": { + "test_mse": 0.0014062192058190703, + "test_rmse": 0.03749958940867314, + "test_mae": 0.024982891976833344, + "test_r2": 0.9340598583221436, + "test_head_mse": 0.0011636127019301057, + "test_head_rmse": 0.03411176779250975, + "test_head_mae": 0.02552611194550991, + "test_head_r2": 0.8857917785644531, + "test_left_shoulder_mse": 0.0008609719225205481, + "test_left_shoulder_rmse": 0.029342323059371903, + "test_left_shoulder_mae": 0.02031836286187172, + "test_left_shoulder_r2": 0.8489664793014526, + "test_left_elbow_mse": 0.00118702941108495, + "test_left_elbow_rmse": 0.034453293182001486, + "test_left_elbow_mae": 0.025187183171510696, + "test_left_elbow_r2": 0.9123913645744324, + "test_right_shoulder_mse": 0.0007992673781700432, + "test_right_shoulder_rmse": 0.028271317234434677, + "test_right_shoulder_mae": 0.019846301525831223, + "test_right_shoulder_r2": 0.8799436092376709, + "test_right_elbow_mse": 0.0017161739524453878, + "test_right_elbow_rmse": 0.04142672992700954, + "test_right_elbow_mae": 0.027927812188863754, + "test_right_elbow_r2": 0.8902333974838257, + "test_left_hand_mse": 0.0018681157380342484, + "test_left_hand_rmse": 0.043221704478586316, + "test_left_hand_mae": 0.032467618584632874, + "test_left_hand_r2": 0.928669273853302, + "test_right_hand_mse": 0.0029382703360170126, + "test_right_hand_rmse": 0.05420581459601002, + "test_right_hand_mae": 0.037380464375019073, + "test_right_hand_r2": 0.8955758810043335, + "test_left_hip_mse": 0.0009324443526566029, + "test_left_hip_rmse": 0.03053595180531635, + "test_left_hip_mae": 0.0218509491533041, + "test_left_hip_r2": 0.8942061066627502, + "test_right_hip_mse": 0.000941735808737576, + "test_right_hip_rmse": 0.03068771429640168, + "test_right_hip_mae": 0.021653885021805763, + "test_right_hip_r2": 0.8972305059432983, + "test_left_knee_mse": 0.0010538198985159397, + "test_left_knee_rmse": 0.03246259229507002, + "test_left_knee_mae": 0.022155458107590675, + "test_left_knee_r2": 0.8916845917701721, + "test_right_knee_mse": 0.0011967695318162441, + "test_right_knee_rmse": 0.034594356936012614, + "test_right_knee_mae": 0.022922644391655922, + "test_right_knee_r2": 0.8883968591690063, + "test_left_foot_mse": 0.0017824883107095957, + "test_left_foot_rmse": 0.04221952523074597, + "test_left_foot_mae": 0.022857986390590668, + "test_left_foot_r2": 0.6167178153991699, + "test_right_foot_mse": 0.0018401507986709476, + "test_right_foot_rmse": 0.042896978899113015, + "test_right_foot_mae": 0.024682819843292236, + "test_right_foot_r2": 0.6494743824005127 + } +} \ No newline at end of file diff --git a/A9/cv_results_loss_optimizer/conv1d_v3_sgd_mae_fold1_best.h5 b/A9/cv_results_loss_optimizer/conv1d_v3_sgd_mae_fold1_best.h5 new file mode 100644 index 0000000000000000000000000000000000000000..6658069248f7192a1c4a0bbc313a9f10f54d407f --- /dev/null +++ b/A9/cv_results_loss_optimizer/conv1d_v3_sgd_mae_fold1_best.h5 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c77e56d24e67b5fbe7468176ec61ef9d2adb265610c3aaed630fc8092b53ac72 +size 2658520 diff --git a/A9/cv_results_loss_optimizer/conv1d_v3_sgd_mae_fold2_best.h5 b/A9/cv_results_loss_optimizer/conv1d_v3_sgd_mae_fold2_best.h5 new file mode 100644 index 0000000000000000000000000000000000000000..881eff900e32f68c3db7be9aa3ff6998fc8b163c --- /dev/null +++ b/A9/cv_results_loss_optimizer/conv1d_v3_sgd_mae_fold2_best.h5 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:d36eaa6008f82af66eebf47e37600d6fe0df6e339c1c811e35fa80a398225037 +size 2658520 diff --git a/A9/cv_results_loss_optimizer/conv1d_v3_sgd_mae_fold3_best.h5 b/A9/cv_results_loss_optimizer/conv1d_v3_sgd_mae_fold3_best.h5 new file mode 100644 index 0000000000000000000000000000000000000000..034c598873ae75a6d062a7e4bca4952d6b47352a --- /dev/null +++ b/A9/cv_results_loss_optimizer/conv1d_v3_sgd_mae_fold3_best.h5 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:915aa63e32b8b60454f1dcc20afc8b93e9f5b9129d3557786ab81c2429b889ba +size 2658520 diff --git a/A9/cv_results_loss_optimizer/conv1d_v3_sgd_mae_fold4_best.h5 b/A9/cv_results_loss_optimizer/conv1d_v3_sgd_mae_fold4_best.h5 new file mode 100644 index 0000000000000000000000000000000000000000..c4feed7ffc7c99312ba33a67e7ae781e4c8c879d --- /dev/null +++ b/A9/cv_results_loss_optimizer/conv1d_v3_sgd_mae_fold4_best.h5 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:cdb48f59b2810d3fdef7271f47f5a6126fa4f8e344601b4e998b7d2246441964 +size 2658520 diff --git a/A9/cv_results_loss_optimizer/conv1d_v3_sgd_mae_fold5_best.h5 b/A9/cv_results_loss_optimizer/conv1d_v3_sgd_mae_fold5_best.h5 new file mode 100644 index 0000000000000000000000000000000000000000..2e96a01cd9742c652f09d40f74e15d8d0c27c5de --- /dev/null +++ b/A9/cv_results_loss_optimizer/conv1d_v3_sgd_mae_fold5_best.h5 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:0c718ec30ae7e0a30cf986da1ca20fb666c9d605673a4c6b61c17a1022660d54 +size 2658520 diff --git a/A9/cv_results_loss_optimizer/conv1d_v3_sgd_mae_predictions.png b/A9/cv_results_loss_optimizer/conv1d_v3_sgd_mae_predictions.png new file mode 100644 index 0000000000000000000000000000000000000000..f7f094b575c4b3b8b71c770648263fcb4a324869 --- /dev/null +++ b/A9/cv_results_loss_optimizer/conv1d_v3_sgd_mae_predictions.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:f0a239d699323946c4cc314a74e21487e249655c72ce823c3ee5977c087dcdbc +size 519704 diff --git a/A9/cv_results_loss_optimizer/conv1d_v3_sgd_mae_test_predictions.png b/A9/cv_results_loss_optimizer/conv1d_v3_sgd_mae_test_predictions.png new file mode 100644 index 0000000000000000000000000000000000000000..c75fa0988ea8e543cb9f9525ddb698562081c864 --- /dev/null +++ b/A9/cv_results_loss_optimizer/conv1d_v3_sgd_mae_test_predictions.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:3692f09a69d2094119dc82b9b9d0dba06f1f8aa8ea7d8efe71ef61439fc32958 +size 425813 diff --git a/A9/cv_results_loss_optimizer/conv1d_v3_sgd_mae_training_history.png b/A9/cv_results_loss_optimizer/conv1d_v3_sgd_mae_training_history.png new file mode 100644 index 0000000000000000000000000000000000000000..5846d9650a46666541f7a9dedfd3cd69c7875129 --- /dev/null +++ b/A9/cv_results_loss_optimizer/conv1d_v3_sgd_mae_training_history.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:08fd10753318017c9a5d7ca394c4f73c7935ed23b13abbb68ba1c1bb3b897138 +size 325543 diff --git a/A9/cv_results_loss_optimizer/conv1d_v3_sgd_mse_cv_results.json b/A9/cv_results_loss_optimizer/conv1d_v3_sgd_mse_cv_results.json new file mode 100644 index 0000000000000000000000000000000000000000..6e7cded856a2fee3ec99a23efa6930ae81436330 --- /dev/null +++ b/A9/cv_results_loss_optimizer/conv1d_v3_sgd_mse_cv_results.json @@ -0,0 +1,1243 @@ +{ + "model_name": "conv1d_v3_sgd_mse", + "n_splits": 5, + "epochs": 100, + "batch_size": 32, + "patience": 10, + "aggregate_results": { + "n_folds": 5, + "train_metrics": { + "train_mse": { + "mean": 0.0007953721564263105, + "std": 4.7562491537766323e-05 + }, + "train_rmse": { + "mean": 0.028189831812412518, + "std": 0.0008399635790948152 + }, + "train_mae": { + "mean": 0.02089722380042076, + "std": 0.0006003474174216383 + }, + "train_r2": { + "mean": 0.9623723506927491, + "std": 0.00228550775688238 + }, + "train_head_mse": { + "mean": 0.000849397957790643, + "std": 4.631269022725269e-05 + }, + "train_head_rmse": { + "mean": 0.029133743379755783, + "std": 0.0007892746500293635 + }, + "train_head_mae": { + "mean": 0.0222737442702055, + "std": 0.0005630178703415852 + }, + "train_head_r2": { + "mean": 0.9146293997764587, + "std": 0.0044156074927181755 + }, + "train_left_shoulder_mse": { + "mean": 0.0005429646815173327, + "std": 3.74194683563416e-05 + }, + "train_left_shoulder_rmse": { + "mean": 0.023287916894303363, + "std": 0.0007985037516210391 + }, + "train_left_shoulder_mae": { + "mean": 0.01773580051958561, + "std": 0.0005015349003722482 + }, + "train_left_shoulder_r2": { + "mean": 0.9022977232933045, + "std": 0.00704678144468564 + }, + "train_left_elbow_mse": { + "mean": 0.0008116692188195884, + "std": 4.1341421548260575e-05 + }, + "train_left_elbow_rmse": { + "mean": 0.02848058179290236, + "std": 0.0007250376248077254 + }, + "train_left_elbow_mae": { + "mean": 0.021923743560910224, + "std": 0.00047748330696343537 + }, + "train_left_elbow_r2": { + "mean": 0.9412059187889099, + "std": 0.0030931328894993762 + }, + "train_right_shoulder_mse": { + "mean": 0.000480666448129341, + "std": 2.986768444588266e-05 + }, + "train_right_shoulder_rmse": { + "mean": 0.021913192479576866, + "std": 0.0006916960910377512 + }, + "train_right_shoulder_mae": { + "mean": 0.016905703768134116, + "std": 0.000522775889977517 + }, + "train_right_shoulder_r2": { + "mean": 0.9262992024421692, + "std": 0.004847009112233317 + }, + "train_right_elbow_mse": { + "mean": 0.0007654812536202372, + "std": 4.647519869257831e-05 + }, + "train_right_elbow_rmse": { + "mean": 0.02765468808761603, + "std": 0.0008363494466459035 + }, + "train_right_elbow_mae": { + "mean": 0.021393443644046783, + "std": 0.0006561145277248676 + }, + "train_right_elbow_r2": { + "mean": 0.9514848470687867, + "std": 0.0027058218640102407 + }, + "train_left_hand_mse": { + "mean": 0.0011941765435039998, + "std": 4.639910059378129e-05 + }, + "train_left_hand_rmse": { + "mean": 0.03455028879800913, + "std": 0.0006738601324939749 + }, + "train_left_hand_mae": { + "mean": 0.027201150357723237, + "std": 0.000573115511034555 + }, + "train_left_hand_r2": { + "mean": 0.9545672535896301, + "std": 0.0018240569382460838 + }, + "train_right_hand_mse": { + "mean": 0.0013257317710667848, + "std": 6.664911785741063e-05 + }, + "train_right_hand_rmse": { + "mean": 0.03639900569504956, + "std": 0.0009187793415912595 + }, + "train_right_hand_mae": { + "mean": 0.027895206212997438, + "std": 0.0007385783361335934 + }, + "train_right_hand_r2": { + "mean": 0.9523991823196412, + "std": 0.002333432331780023 + }, + "train_left_hip_mse": { + "mean": 0.0006395495380274952, + "std": 3.188691251283014e-05 + }, + "train_left_hip_rmse": { + "mean": 0.02528137799588947, + "std": 0.0006336123865983004 + }, + "train_left_hip_mae": { + "mean": 0.01923542022705078, + "std": 0.00045358831147044264 + }, + "train_left_hip_r2": { + "mean": 0.9179353833198547, + "std": 0.004611130654817937 + }, + "train_right_hip_mse": { + "mean": 0.0006239432259462774, + "std": 2.6156600530794666e-05 + }, + "train_right_hip_rmse": { + "mean": 0.024973300767245795, + "std": 0.000526758706580204 + }, + "train_right_hip_mae": { + "mean": 0.01893010064959526, + "std": 0.00038917739735357484 + }, + "train_right_hip_r2": { + "mean": 0.9230729460716247, + "std": 0.0037121248810790815 + }, + "train_left_knee_mse": { + "mean": 0.0006713719107210636, + "std": 5.1324102822734964e-05 + }, + "train_left_knee_rmse": { + "mean": 0.025892337443721335, + "std": 0.0009791692507067744 + }, + "train_left_knee_mae": { + "mean": 0.019186773151159287, + "std": 0.0007536513531164381 + }, + "train_left_knee_r2": { + "mean": 0.9292174816131592, + "std": 0.005476676599707047 + }, + "train_right_knee_mse": { + "mean": 0.0006821546703577042, + "std": 4.994025439292733e-05 + }, + "train_right_knee_rmse": { + "mean": 0.02610041337287174, + "std": 0.00096077683304823 + }, + "train_right_knee_mae": { + "mean": 0.01935901679098606, + "std": 0.0006103223424153869 + }, + "train_right_knee_r2": { + "mean": 0.9365197896957398, + "std": 0.005365738220074637 + }, + "train_left_foot_mse": { + "mean": 0.0008892910205759108, + "std": 0.00011446641488408969 + }, + "train_left_foot_rmse": { + "mean": 0.02976230849015895, + "std": 0.001869763063204311 + }, + "train_left_foot_mae": { + "mean": 0.019575729593634607, + "std": 0.0011310527213152747 + }, + "train_left_foot_r2": { + "mean": 0.7970824360847473, + "std": 0.02089212806403463 + }, + "train_right_foot_mse": { + "mean": 0.0008634399506263435, + "std": 6.912592171165173e-05 + }, + "train_right_foot_rmse": { + "mean": 0.029360765092980266, + "std": 0.001177040263191929 + }, + "train_right_foot_mae": { + "mean": 0.020048070326447488, + "std": 0.0008710977356199867 + }, + "train_right_foot_r2": { + "mean": 0.8072531223297119, + "std": 0.01748796535822246 + } + }, + "val_metrics": { + "val_mse": { + "mean": 0.0010413570562377572, + "std": 8.872879907278316e-05 + }, + "val_rmse": { + "mean": 0.03224083495173871, + "std": 0.001373178011949172 + }, + "val_mae": { + "mean": 0.023373308032751082, + "std": 0.0010204718651930882 + }, + "val_r2": { + "mean": 0.9507483959197998, + "std": 0.0036322475275655857 + }, + "val_head_mse": { + "mean": 0.0010457638185471297, + "std": 0.00010320678838089403 + }, + "val_head_rmse": { + "mean": 0.03229812735482443, + "std": 0.001610834541059813 + }, + "val_head_mae": { + "mean": 0.024422919377684593, + "std": 0.001232094282782669 + }, + "val_head_r2": { + "mean": 0.8948119759559632, + "std": 0.00885792061250419 + }, + "val_left_shoulder_mse": { + "mean": 0.000671362888533622, + "std": 8.912951511272009e-05 + }, + "val_left_shoulder_rmse": { + "mean": 0.02585561595310133, + "std": 0.0016881979206698524 + }, + "val_left_shoulder_mae": { + "mean": 0.019546468183398245, + "std": 0.0010663458209019961 + }, + "val_left_shoulder_r2": { + "mean": 0.8795102834701538, + "std": 0.010706755509356443 + }, + "val_left_elbow_mse": { + "mean": 0.001057420764118433, + "std": 9.539347879654275e-05 + }, + "val_left_elbow_rmse": { + "mean": 0.0324846719019231, + "std": 0.0014720243010361187 + }, + "val_left_elbow_mae": { + "mean": 0.024876165390014648, + "std": 0.0007218376181017783 + }, + "val_left_elbow_r2": { + "mean": 0.9233672380447387, + "std": 0.0060621118741011094 + }, + "val_right_shoulder_mse": { + "mean": 0.0006144687766209244, + "std": 7.974309962062505e-05 + }, + "val_right_shoulder_rmse": { + "mean": 0.02473457833709383, + "std": 0.0016338332555965586 + }, + "val_right_shoulder_mae": { + "mean": 0.018848047778010368, + "std": 0.0009968912771609912 + }, + "val_right_shoulder_r2": { + "mean": 0.905785059928894, + "std": 0.011198249288826735 + }, + "val_right_elbow_mse": { + "mean": 0.0010620659799315036, + "std": 0.00011167865778599875 + }, + "val_right_elbow_rmse": { + "mean": 0.0325447188824183, + "std": 0.0017050668009991716 + }, + "val_right_elbow_mae": { + "mean": 0.02478046715259552, + "std": 0.001515048483802289 + }, + "val_right_elbow_r2": { + "mean": 0.9324532270431518, + "std": 0.008156937738744778 + }, + "val_left_hand_mse": { + "mean": 0.001659048697911203, + "std": 0.00011984074725391454 + }, + "val_left_hand_rmse": { + "mean": 0.04070414555395448, + "std": 0.0014903800299557758 + }, + "val_left_hand_mae": { + "mean": 0.03153003677725792, + "std": 0.0008557794618395918 + }, + "val_left_hand_r2": { + "mean": 0.9366336822509765, + "std": 0.006177470225018122 + }, + "val_right_hand_mse": { + "mean": 0.0018670403631404042, + "std": 0.00024890265526771174 + }, + "val_right_hand_rmse": { + "mean": 0.043111811730353757, + "std": 0.0029003538520215736 + }, + "val_right_hand_mae": { + "mean": 0.03226032592356205, + "std": 0.002237609153141495 + }, + "val_right_hand_r2": { + "mean": 0.9327909827232361, + "std": 0.009484434537002755 + }, + "val_left_hip_mse": { + "mean": 0.0007823848514817655, + "std": 9.056927358372038e-05 + }, + "val_left_hip_rmse": { + "mean": 0.027924349258858738, + "std": 0.001617272379978374 + }, + "val_left_hip_mae": { + "mean": 0.021068546921014786, + "std": 0.000942462627938359 + }, + "val_left_hip_r2": { + "mean": 0.8993941307067871, + "std": 0.011314667125876825 + }, + "val_right_hip_mse": { + "mean": 0.0007707600481808186, + "std": 7.554475114557655e-05 + }, + "val_right_hip_rmse": { + "mean": 0.027729102234797565, + "std": 0.0013626949155857151 + }, + "val_right_hip_mae": { + "mean": 0.020791907235980032, + "std": 0.0008298151342236782 + }, + "val_right_hip_r2": { + "mean": 0.9046743750572205, + "std": 0.00971747203722941 + }, + "val_left_knee_mse": { + "mean": 0.000864117476157844, + "std": 9.149413640629903e-05 + }, + "val_left_knee_rmse": { + "mean": 0.029354837324810264, + "std": 0.0015527400915061288 + }, + "val_left_knee_mae": { + "mean": 0.021293051540851593, + "std": 0.001541090779230436 + }, + "val_left_knee_r2": { + "mean": 0.9089268326759339, + "std": 0.007348315753075634 + }, + "val_right_knee_mse": { + "mean": 0.0009042118559591472, + "std": 0.00015059731674377876 + }, + "val_right_knee_rmse": { + "mean": 0.02997323059593552, + "std": 0.002411908705159041 + }, + "val_right_knee_mae": { + "mean": 0.021575798839330675, + "std": 0.0015839776168932664 + }, + "val_right_knee_r2": { + "mean": 0.9162698030471802, + "std": 0.010365951251429188 + }, + "val_left_foot_mse": { + "mean": 0.0011026691645383835, + "std": 0.0002562571201536626 + }, + "val_left_foot_rmse": { + "mean": 0.03298485980489785, + "std": 0.003829907073235268 + }, + "val_left_foot_mae": { + "mean": 0.02111378125846386, + "std": 0.0009509938643483045 + }, + "val_left_foot_r2": { + "mean": 0.7483095526695251, + "std": 0.04213070715810592 + }, + "val_right_foot_mse": { + "mean": 0.0011363269295543432, + "std": 0.0001245798352323765 + }, + "val_right_foot_rmse": { + "mean": 0.03365835694179241, + "std": 0.0018552459495274362 + }, + "val_right_foot_mae": { + "mean": 0.02174548842012882, + "std": 0.000739869378499388 + }, + "val_right_foot_r2": { + "mean": 0.7459302306175232, + "std": 0.02824230239148711 + } + }, + "best_fold": "1", + "best_fold_metrics": { + "fold": 1, + "train_metrics": { + "train_mse": 0.0007875558221712708, + "train_rmse": 0.028063424990034107, + "train_mae": 0.020895114168524742, + "train_r2": 0.9630032181739807, + "train_head_mse": 0.0008363300585187972, + "train_head_rmse": 0.028919371682642022, + "train_head_mae": 0.022171637043356895, + "train_head_r2": 0.9171674251556396, + "train_left_shoulder_mse": 0.000530226796399802, + "train_left_shoulder_rmse": 0.023026654042648097, + "train_left_shoulder_mae": 0.017664751037955284, + "train_left_shoulder_r2": 0.9061439037322998, + "train_left_elbow_mse": 0.0008158910204656422, + "train_left_elbow_rmse": 0.028563806127084014, + "train_left_elbow_mae": 0.02226332575082779, + "train_left_elbow_r2": 0.9408853650093079, + "train_right_shoulder_mse": 0.00047371184336952865, + "train_right_shoulder_rmse": 0.021764922314805736, + "train_right_shoulder_mae": 0.016861364245414734, + "train_right_shoulder_r2": 0.9283120036125183, + "train_right_elbow_mse": 0.0007711277576163411, + "train_right_elbow_rmse": 0.027769187197617815, + "train_right_elbow_mae": 0.02162449061870575, + "train_right_elbow_r2": 0.9515588879585266, + "train_left_hand_mse": 0.001212855102494359, + "train_left_hand_rmse": 0.0348260692943427, + "train_left_hand_mae": 0.027607906609773636, + "train_left_hand_r2": 0.9534537196159363, + "train_right_hand_mse": 0.0013154156040400267, + "train_right_hand_rmse": 0.03626865870197059, + "train_right_hand_mae": 0.028017915785312653, + "train_right_hand_r2": 0.9529296159744263, + "train_left_hip_mse": 0.0006341904518194497, + "train_left_hip_rmse": 0.025183138244060245, + "train_left_hip_mae": 0.019194476306438446, + "train_left_hip_r2": 0.9206670522689819, + "train_right_hip_mse": 0.0006173900910653174, + "train_right_hip_rmse": 0.024847335693496744, + "train_right_hip_mae": 0.018832609057426453, + "train_right_hip_r2": 0.9256055951118469, + "train_left_knee_mse": 0.0006684326799586415, + "train_left_knee_rmse": 0.025854065056749614, + "train_left_knee_mae": 0.01881587691605091, + "train_left_knee_r2": 0.9300825595855713, + "train_right_knee_mse": 0.0006968815578147769, + "train_right_knee_rmse": 0.02639851431074819, + "train_right_knee_mae": 0.01950703002512455, + "train_right_knee_r2": 0.9352008104324341, + "train_left_foot_mse": 0.0008430992602370679, + "train_left_foot_rmse": 0.02903617158368279, + "train_left_foot_mae": 0.01914321258664131, + "train_left_foot_r2": 0.8086206912994385, + "train_right_foot_mse": 0.0008226742502301931, + "train_right_foot_rmse": 0.02868229855207203, + "train_right_foot_mae": 0.019931910559535027, + "train_right_foot_r2": 0.8146674036979675 + }, + "val_metrics": { + "val_mse": 0.0009282092214561999, + "val_rmse": 0.030466526245310606, + "val_mae": 0.022135484963655472, + "val_r2": 0.9548190832138062, + "val_head_mse": 0.0009410002385266125, + "val_head_rmse": 0.03067572718822836, + "val_head_mae": 0.02249930240213871, + "val_head_r2": 0.8991708159446716, + "val_left_shoulder_mse": 0.0005685879732482135, + "val_left_shoulder_rmse": 0.023845082789711876, + "val_left_shoulder_mae": 0.017884189262986183, + "val_left_shoulder_r2": 0.8903442621231079, + "val_left_elbow_mse": 0.0009089696686714888, + "val_left_elbow_rmse": 0.03014912384583487, + "val_left_elbow_mae": 0.023672036826610565, + "val_left_elbow_r2": 0.9341219663619995, + "val_right_shoulder_mse": 0.0005209188093431294, + "val_right_shoulder_rmse": 0.022823645838102407, + "val_right_shoulder_mae": 0.01766647957265377, + "val_right_shoulder_r2": 0.9157155156135559, + "val_right_elbow_mse": 0.0009246893459931016, + "val_right_elbow_rmse": 0.030408705102208833, + "val_right_elbow_mae": 0.023431317880749702, + "val_right_elbow_r2": 0.9391260147094727, + "val_left_hand_mse": 0.0014533187495544553, + "val_left_hand_rmse": 0.03812241793950713, + "val_left_hand_mae": 0.03028097003698349, + "val_left_hand_r2": 0.9464741945266724, + "val_right_hand_mse": 0.0016344180330634117, + "val_right_hand_rmse": 0.040427936294886634, + "val_right_hand_mae": 0.03058927319943905, + "val_right_hand_r2": 0.9404840469360352, + "val_left_hip_mse": 0.0006999411270953715, + "val_left_hip_rmse": 0.026456400493932872, + "val_left_hip_mae": 0.019850026816129684, + "val_left_hip_r2": 0.9001730680465698, + "val_right_hip_mse": 0.0006910217343829572, + "val_right_hip_rmse": 0.026287292260386143, + "val_right_hip_mae": 0.01957397535443306, + "val_right_hip_r2": 0.9063662886619568, + "val_left_knee_mse": 0.0007541072554886341, + "val_left_knee_rmse": 0.027461013373301323, + "val_left_knee_mae": 0.01950930804014206, + "val_left_knee_r2": 0.9178834557533264, + "val_right_knee_mse": 0.0008735336014069617, + "val_right_knee_rmse": 0.029555601861693863, + "val_right_knee_mae": 0.021566761657595634, + "val_right_knee_r2": 0.9187635779380798, + "val_left_foot_mse": 0.001099838293157518, + "val_left_foot_rmse": 0.03316380999157844, + "val_left_foot_mae": 0.020674221217632294, + "val_left_foot_r2": 0.7409694790840149, + "val_right_foot_mse": 0.0009963749907910824, + "val_right_foot_rmse": 0.031565408135981424, + "val_right_foot_mae": 0.020563436672091484, + "val_right_foot_r2": 0.7864870429039001 + }, + "best_epoch": 100 + } + }, + "fold_details": [ + { + "fold": 1, + "train_metrics": { + "train_mse": 0.0007875558221712708, + "train_rmse": 0.028063424990034107, + "train_mae": 0.020895114168524742, + "train_r2": 0.9630032181739807, + "train_head_mse": 0.0008363300585187972, + "train_head_rmse": 0.028919371682642022, + "train_head_mae": 0.022171637043356895, + "train_head_r2": 0.9171674251556396, + "train_left_shoulder_mse": 0.000530226796399802, + "train_left_shoulder_rmse": 0.023026654042648097, + "train_left_shoulder_mae": 0.017664751037955284, + "train_left_shoulder_r2": 0.9061439037322998, + "train_left_elbow_mse": 0.0008158910204656422, + "train_left_elbow_rmse": 0.028563806127084014, + "train_left_elbow_mae": 0.02226332575082779, + "train_left_elbow_r2": 0.9408853650093079, + "train_right_shoulder_mse": 0.00047371184336952865, + "train_right_shoulder_rmse": 0.021764922314805736, + "train_right_shoulder_mae": 0.016861364245414734, + "train_right_shoulder_r2": 0.9283120036125183, + "train_right_elbow_mse": 0.0007711277576163411, + "train_right_elbow_rmse": 0.027769187197617815, + "train_right_elbow_mae": 0.02162449061870575, + "train_right_elbow_r2": 0.9515588879585266, + "train_left_hand_mse": 0.001212855102494359, + "train_left_hand_rmse": 0.0348260692943427, + "train_left_hand_mae": 0.027607906609773636, + "train_left_hand_r2": 0.9534537196159363, + "train_right_hand_mse": 0.0013154156040400267, + "train_right_hand_rmse": 0.03626865870197059, + "train_right_hand_mae": 0.028017915785312653, + "train_right_hand_r2": 0.9529296159744263, + "train_left_hip_mse": 0.0006341904518194497, + "train_left_hip_rmse": 0.025183138244060245, + "train_left_hip_mae": 0.019194476306438446, + "train_left_hip_r2": 0.9206670522689819, + "train_right_hip_mse": 0.0006173900910653174, + "train_right_hip_rmse": 0.024847335693496744, + "train_right_hip_mae": 0.018832609057426453, + "train_right_hip_r2": 0.9256055951118469, + "train_left_knee_mse": 0.0006684326799586415, + "train_left_knee_rmse": 0.025854065056749614, + "train_left_knee_mae": 0.01881587691605091, + "train_left_knee_r2": 0.9300825595855713, + "train_right_knee_mse": 0.0006968815578147769, + "train_right_knee_rmse": 0.02639851431074819, + "train_right_knee_mae": 0.01950703002512455, + "train_right_knee_r2": 0.9352008104324341, + "train_left_foot_mse": 0.0008430992602370679, + "train_left_foot_rmse": 0.02903617158368279, + "train_left_foot_mae": 0.01914321258664131, + "train_left_foot_r2": 0.8086206912994385, + "train_right_foot_mse": 0.0008226742502301931, + "train_right_foot_rmse": 0.02868229855207203, + "train_right_foot_mae": 0.019931910559535027, + "train_right_foot_r2": 0.8146674036979675 + }, + "val_metrics": { + "val_mse": 0.0009282092214561999, + "val_rmse": 0.030466526245310606, + "val_mae": 0.022135484963655472, + "val_r2": 0.9548190832138062, + "val_head_mse": 0.0009410002385266125, + "val_head_rmse": 0.03067572718822836, + "val_head_mae": 0.02249930240213871, + "val_head_r2": 0.8991708159446716, + "val_left_shoulder_mse": 0.0005685879732482135, + "val_left_shoulder_rmse": 0.023845082789711876, + "val_left_shoulder_mae": 0.017884189262986183, + "val_left_shoulder_r2": 0.8903442621231079, + "val_left_elbow_mse": 0.0009089696686714888, + "val_left_elbow_rmse": 0.03014912384583487, + "val_left_elbow_mae": 0.023672036826610565, + "val_left_elbow_r2": 0.9341219663619995, + "val_right_shoulder_mse": 0.0005209188093431294, + "val_right_shoulder_rmse": 0.022823645838102407, + "val_right_shoulder_mae": 0.01766647957265377, + "val_right_shoulder_r2": 0.9157155156135559, + "val_right_elbow_mse": 0.0009246893459931016, + "val_right_elbow_rmse": 0.030408705102208833, + "val_right_elbow_mae": 0.023431317880749702, + "val_right_elbow_r2": 0.9391260147094727, + "val_left_hand_mse": 0.0014533187495544553, + "val_left_hand_rmse": 0.03812241793950713, + "val_left_hand_mae": 0.03028097003698349, + "val_left_hand_r2": 0.9464741945266724, + "val_right_hand_mse": 0.0016344180330634117, + "val_right_hand_rmse": 0.040427936294886634, + "val_right_hand_mae": 0.03058927319943905, + "val_right_hand_r2": 0.9404840469360352, + "val_left_hip_mse": 0.0006999411270953715, + "val_left_hip_rmse": 0.026456400493932872, + "val_left_hip_mae": 0.019850026816129684, + "val_left_hip_r2": 0.9001730680465698, + "val_right_hip_mse": 0.0006910217343829572, + "val_right_hip_rmse": 0.026287292260386143, + "val_right_hip_mae": 0.01957397535443306, + "val_right_hip_r2": 0.9063662886619568, + "val_left_knee_mse": 0.0007541072554886341, + "val_left_knee_rmse": 0.027461013373301323, + "val_left_knee_mae": 0.01950930804014206, + "val_left_knee_r2": 0.9178834557533264, + "val_right_knee_mse": 0.0008735336014069617, + "val_right_knee_rmse": 0.029555601861693863, + "val_right_knee_mae": 0.021566761657595634, + "val_right_knee_r2": 0.9187635779380798, + "val_left_foot_mse": 0.001099838293157518, + "val_left_foot_rmse": 0.03316380999157844, + "val_left_foot_mae": 0.020674221217632294, + "val_left_foot_r2": 0.7409694790840149, + "val_right_foot_mse": 0.0009963749907910824, + "val_right_foot_rmse": 0.031565408135981424, + "val_right_foot_mae": 0.020563436672091484, + "val_right_foot_r2": 0.7864870429039001 + }, + "best_epoch": 100 + }, + { + "fold": 2, + "train_metrics": { + "train_mse": 0.0007718307315371931, + "train_rmse": 0.027781841759271342, + "train_mae": 0.02058352716267109, + "train_r2": 0.9632009863853455, + "train_head_mse": 0.0008361193467862904, + "train_head_rmse": 0.02891572836340614, + "train_head_mae": 0.022102804854512215, + "train_head_r2": 0.9144812226295471, + "train_left_shoulder_mse": 0.0005331310094334185, + "train_left_shoulder_rmse": 0.023089629911140164, + "train_left_shoulder_mae": 0.017740342766046524, + "train_left_shoulder_r2": 0.9036190509796143, + "train_left_elbow_mse": 0.0007731188088655472, + "train_left_elbow_rmse": 0.027805014095762425, + "train_left_elbow_mae": 0.021563084796071053, + "train_left_elbow_r2": 0.9443867802619934, + "train_right_shoulder_mse": 0.000488677469547838, + "train_right_shoulder_rmse": 0.022106050518983213, + "train_right_shoulder_mae": 0.017057202756404877, + "train_right_shoulder_r2": 0.9238553047180176, + "train_right_elbow_mse": 0.0007186686270870268, + "train_right_elbow_rmse": 0.026807995581300495, + "train_right_elbow_mae": 0.02076360210776329, + "train_right_elbow_r2": 0.9540784955024719, + "train_left_hand_mse": 0.001182462670840323, + "train_left_hand_rmse": 0.03438695495155573, + "train_left_hand_mae": 0.026793338358402252, + "train_left_hand_r2": 0.9552302360534668, + "train_right_hand_mse": 0.0013304174644872546, + "train_right_hand_rmse": 0.03647488813536314, + "train_right_hand_mae": 0.02781589701771736, + "train_right_hand_r2": 0.9519886374473572, + "train_left_hip_mse": 0.0006423701997846365, + "train_left_hip_rmse": 0.025345023175855186, + "train_left_hip_mae": 0.01931493729352951, + "train_left_hip_r2": 0.9149635434150696, + "train_right_hip_mse": 0.0006234141765162349, + "train_right_hip_rmse": 0.024968263386071426, + "train_right_hip_mae": 0.018955521285533905, + "train_right_hip_r2": 0.9202274084091187, + "train_left_knee_mse": 0.0006219662027433515, + "train_left_knee_rmse": 0.02493925024421046, + "train_left_knee_mae": 0.0186793003231287, + "train_left_knee_r2": 0.9330756068229675, + "train_right_knee_mse": 0.0006702675018459558, + "train_right_knee_rmse": 0.02588952494438544, + "train_right_knee_mae": 0.01881502754986286, + "train_right_knee_r2": 0.9377555847167969, + "train_left_foot_mse": 0.0007706634933128953, + "train_left_foot_rmse": 0.027760826596355075, + "train_left_foot_mae": 0.018662240356206894, + "train_left_foot_r2": 0.8215478658676147, + "train_right_foot_mse": 0.0008425220148637891, + "train_right_foot_rmse": 0.02902622977349606, + "train_right_foot_mae": 0.019322527572512627, + "train_right_foot_r2": 0.8155639171600342 + }, + "val_metrics": { + "val_mse": 0.00101874687243253, + "val_rmse": 0.031917814342973576, + "val_mae": 0.023579705506563187, + "val_r2": 0.953230082988739, + "val_head_mse": 0.0011149944039061666, + "val_head_rmse": 0.03339153191912834, + "val_head_mae": 0.025267411023378372, + "val_head_r2": 0.8949964642524719, + "val_left_shoulder_mse": 0.0006439197459258139, + "val_left_shoulder_rmse": 0.025375573804858362, + "val_left_shoulder_mae": 0.01957693323493004, + "val_left_shoulder_r2": 0.8860762715339661, + "val_left_elbow_mse": 0.0010280595161020756, + "val_left_elbow_rmse": 0.03206336719844121, + "val_left_elbow_mae": 0.025107353925704956, + "val_left_elbow_r2": 0.9232199192047119, + "val_right_shoulder_mse": 0.0006791979540139437, + "val_right_shoulder_rmse": 0.026061426553700848, + "val_right_shoulder_mae": 0.0193700660020113, + "val_right_shoulder_r2": 0.9020112752914429, + "val_right_elbow_mse": 0.0010433056158944964, + "val_right_elbow_rmse": 0.0323002417312084, + "val_right_elbow_mae": 0.024807926267385483, + "val_right_elbow_r2": 0.9357026815414429, + "val_left_hand_mse": 0.0017756626475602388, + "val_left_hand_rmse": 0.042138612311753205, + "val_left_hand_mae": 0.03224276751279831, + "val_left_hand_r2": 0.9309431910514832, + "val_right_hand_mse": 0.0019068048568442464, + "val_right_hand_rmse": 0.04366697673121241, + "val_right_hand_mae": 0.03274356946349144, + "val_right_hand_r2": 0.932707667350769, + "val_left_hip_mse": 0.0007649022154510021, + "val_left_hip_rmse": 0.027656865611471633, + "val_left_hip_mae": 0.02121901698410511, + "val_left_hip_r2": 0.9126707911491394, + "val_right_hip_mse": 0.0007858806638978422, + "val_right_hip_rmse": 0.028033563168064136, + "val_right_hip_mae": 0.02117357775568962, + "val_right_hip_r2": 0.9155258536338806, + "val_left_knee_mse": 0.0008813031017780304, + "val_left_knee_rmse": 0.029686749599409337, + "val_left_knee_mae": 0.02166443131864071, + "val_left_knee_r2": 0.9138223528862, + "val_right_knee_mse": 0.0007992945029400289, + "val_right_knee_rmse": 0.028271796952794297, + "val_right_knee_mae": 0.021233035251498222, + "val_right_knee_r2": 0.9253104329109192, + "val_left_foot_mse": 0.0008297227323055267, + "val_left_foot_rmse": 0.02880490812874651, + "val_left_foot_mae": 0.020763661712408066, + "val_left_foot_r2": 0.8188194036483765, + "val_right_foot_mse": 0.0009906620252877474, + "val_right_foot_rmse": 0.03147478395934986, + "val_right_foot_mae": 0.0213664211332798, + "val_right_foot_r2": 0.7610109448432922 + }, + "best_epoch": 100 + }, + { + "fold": 3, + "train_metrics": { + "train_mse": 0.0008147816406562924, + "train_rmse": 0.02854438019394172, + "train_mae": 0.02117432840168476, + "train_r2": 0.9610252380371094, + "train_head_mse": 0.0008573331870138645, + "train_head_rmse": 0.029280252509393843, + "train_head_mae": 0.022326752543449402, + "train_head_r2": 0.9128797650337219, + "train_left_shoulder_mse": 0.0005545366439037025, + "train_left_shoulder_rmse": 0.02354860173988474, + "train_left_shoulder_mae": 0.017710281535983086, + "train_left_shoulder_r2": 0.8982324004173279, + "train_left_elbow_mse": 0.0008382562082260847, + "train_left_elbow_rmse": 0.02895265459722277, + "train_left_elbow_mae": 0.022240668535232544, + "train_left_elbow_r2": 0.9387004971504211, + "train_right_shoulder_mse": 0.0005065648001618683, + "train_right_shoulder_rmse": 0.02250699447198289, + "train_right_shoulder_mae": 0.01714191399514675, + "train_right_shoulder_r2": 0.9212942123413086, + "train_right_elbow_mse": 0.0007847468950785697, + "train_right_elbow_rmse": 0.02801333423708377, + "train_right_elbow_mae": 0.021833889186382294, + "train_right_elbow_r2": 0.94948410987854, + "train_left_hand_mse": 0.001198413548991084, + "train_left_hand_rmse": 0.03461811013026396, + "train_left_hand_mae": 0.02745596133172512, + "train_left_hand_r2": 0.9539552927017212, + "train_right_hand_mse": 0.0013421850744634867, + "train_right_hand_rmse": 0.036635844121072014, + "train_right_hand_mae": 0.028027594089508057, + "train_right_hand_r2": 0.9509661793708801, + "train_left_hip_mse": 0.000649516296107322, + "train_left_hip_rmse": 0.025485609588693812, + "train_left_hip_mae": 0.019317714497447014, + "train_left_hip_r2": 0.9162910580635071, + "train_right_hip_mse": 0.0006432574591599405, + "train_right_hip_rmse": 0.025362520757210637, + "train_right_hip_mae": 0.019185760989785194, + "train_right_hip_r2": 0.920660138130188, + "train_left_knee_mse": 0.0006854099337942898, + "train_left_knee_rmse": 0.02618033486787917, + "train_left_knee_mae": 0.019594039767980576, + "train_left_knee_r2": 0.9272003769874573, + "train_right_knee_mse": 0.00068521813955158, + "train_right_knee_rmse": 0.026176671666802483, + "train_right_knee_mae": 0.019764253869652748, + "train_right_knee_r2": 0.9359325170516968, + "train_left_foot_mse": 0.0009128634701482952, + "train_left_foot_rmse": 0.03021363053570847, + "train_left_foot_mae": 0.019886324182152748, + "train_left_foot_r2": 0.785778284072876, + "train_right_foot_mse": 0.0009338599629700184, + "train_right_foot_rmse": 0.030559122418191566, + "train_right_foot_mae": 0.020781103521585464, + "train_right_foot_r2": 0.7899256348609924 + }, + "val_metrics": { + "val_mse": 0.0011517954990267754, + "val_rmse": 0.03393811277939266, + "val_mae": 0.02428419515490532, + "val_r2": 0.9478365182876587, + "val_head_mse": 0.001165734021924436, + "val_head_rmse": 0.034142847302538144, + "val_head_mae": 0.025580579414963722, + "val_head_r2": 0.887727677822113, + "val_left_shoulder_mse": 0.0008267351076938212, + "val_left_shoulder_rmse": 0.028753001716235144, + "val_left_shoulder_mae": 0.020735209807753563, + "val_left_shoulder_r2": 0.8622110486030579, + "val_left_elbow_mse": 0.0012026152107864618, + "val_left_elbow_rmse": 0.03467874292396513, + "val_left_elbow_mae": 0.0258388202637434, + "val_left_elbow_r2": 0.9160513281822205, + "val_right_shoulder_mse": 0.0006734383059665561, + "val_right_shoulder_rmse": 0.025950689893845906, + "val_right_shoulder_mae": 0.019532784819602966, + "val_right_shoulder_r2": 0.9020643830299377, + "val_right_elbow_mse": 0.001141247688792646, + "val_right_elbow_rmse": 0.03378235765592221, + "val_right_elbow_mae": 0.02616545744240284, + "val_right_elbow_r2": 0.9318080544471741, + "val_left_hand_mse": 0.0016194494673982263, + "val_left_hand_rmse": 0.040242383967630774, + "val_left_hand_mae": 0.03161788359284401, + "val_left_hand_r2": 0.9406762719154358, + "val_right_hand_mse": 0.00207453197799623, + "val_right_hand_rmse": 0.04554703039712062, + "val_right_hand_mae": 0.033806007355451584, + "val_right_hand_r2": 0.9302887916564941, + "val_left_hip_mse": 0.0008837055647745728, + "val_left_hip_rmse": 0.029727185618126935, + "val_left_hip_mae": 0.021637922152876854, + "val_left_hip_r2": 0.8888033628463745, + "val_right_hip_mse": 0.0008215990965254605, + "val_right_hip_rmse": 0.028663549963768627, + "val_right_hip_mae": 0.02104070410132408, + "val_right_hip_r2": 0.8990838527679443, + "val_left_knee_mse": 0.0010028282413259149, + "val_left_knee_rmse": 0.03166746344950784, + "val_left_knee_mae": 0.02250279113650322, + "val_left_knee_r2": 0.897411048412323, + "val_right_knee_mse": 0.000900746847037226, + "val_right_knee_rmse": 0.030012444869374204, + "val_right_knee_mae": 0.021542923524975777, + "val_right_knee_r2": 0.918103814125061, + "val_left_foot_mse": 0.0015036939876154065, + "val_left_foot_rmse": 0.03877749331268599, + "val_left_foot_mae": 0.022949200123548508, + "val_left_foot_r2": 0.6868125200271606, + "val_right_foot_mse": 0.0011570157948881388, + "val_right_foot_rmse": 0.034014934879963224, + "val_right_foot_mae": 0.022744260728359222, + "val_right_foot_r2": 0.7504087090492249 + }, + "best_epoch": 100 + }, + { + "fold": 4, + "train_metrics": { + "train_mse": 0.0007297657430171967, + "train_rmse": 0.027014176704412012, + "train_mae": 0.020014870911836624, + "train_r2": 0.965722918510437, + "train_head_mse": 0.0007873828872106969, + "train_head_rmse": 0.02806034367591917, + "train_head_mae": 0.021512476727366447, + "train_head_r2": 0.9209334254264832, + "train_left_shoulder_mse": 0.0004912468139082193, + "train_left_shoulder_rmse": 0.022164088384326104, + "train_left_shoulder_mae": 0.016991961747407913, + "train_left_shoulder_r2": 0.9121055006980896, + "train_left_elbow_mse": 0.0007593939080834389, + "train_left_elbow_rmse": 0.02755710267940806, + "train_left_elbow_mae": 0.021163301542401314, + "train_left_elbow_r2": 0.9449830055236816, + "train_right_shoulder_mse": 0.0004264837480150163, + "train_right_shoulder_rmse": 0.020651482949536972, + "train_right_shoulder_mae": 0.015950817614793777, + "train_right_shoulder_r2": 0.9348317980766296, + "train_right_elbow_mse": 0.0007133227190934122, + "train_right_elbow_rmse": 0.026708102124512932, + "train_right_elbow_mae": 0.020504433661699295, + "train_right_elbow_r2": 0.9547325372695923, + "train_left_hand_mse": 0.0011171698570251465, + "train_left_hand_rmse": 0.0334240909678206, + "train_left_hand_mae": 0.026295356452465057, + "train_left_hand_r2": 0.9577459096908569, + "train_right_hand_mse": 0.0012160302139818668, + "train_right_hand_rmse": 0.034871624768310794, + "train_right_hand_mae": 0.026651205494999886, + "train_right_hand_r2": 0.9565096497535706, + "train_left_hip_mse": 0.0005862356047146022, + "train_left_hip_rmse": 0.024212302755306076, + "train_left_hip_mae": 0.018465537577867508, + "train_left_hip_r2": 0.9254174828529358, + "train_right_hip_mse": 0.0005795534234493971, + "train_right_hip_rmse": 0.024073915831235207, + "train_right_hip_mae": 0.018261708319187164, + "train_right_hip_r2": 0.9291713833808899, + "train_left_knee_mse": 0.0006205593817867339, + "train_left_knee_rmse": 0.02491102932009703, + "train_left_knee_mae": 0.018381165340542793, + "train_left_knee_r2": 0.9358207583427429, + "train_right_knee_mse": 0.0006014690734446049, + "train_right_knee_rmse": 0.024524866430718942, + "train_right_knee_mae": 0.01852361112833023, + "train_right_knee_r2": 0.945225179195404, + "train_left_foot_mse": 0.0008206742932088673, + "train_left_foot_rmse": 0.028647413377281855, + "train_left_foot_mae": 0.01855357177555561, + "train_left_foot_r2": 0.8072925806045532, + "train_right_foot_mse": 0.0007674327353015542, + "train_right_foot_rmse": 0.02770257632967653, + "train_right_foot_mae": 0.01893816515803337, + "train_right_foot_r2": 0.8315445184707642 + }, + "val_metrics": { + "val_mse": 0.000971757632214576, + "val_rmse": 0.03117302731873464, + "val_mae": 0.022251412272453308, + "val_r2": 0.9526785612106323, + "val_head_mse": 0.0009047214989550412, + "val_head_rmse": 0.030078588712820972, + "val_head_mae": 0.02342616766691208, + "val_head_r2": 0.9087288975715637, + "val_left_shoulder_mse": 0.0006143580540083349, + "val_left_shoulder_rmse": 0.024786247275623126, + "val_left_shoulder_mae": 0.01892833039164543, + "val_left_shoulder_r2": 0.8869904279708862, + "val_left_elbow_mse": 0.001051521860063076, + "val_left_elbow_rmse": 0.03242717779984987, + "val_left_elbow_mae": 0.024585405364632607, + "val_left_elbow_r2": 0.9238569736480713, + "val_right_shoulder_mse": 0.0005130188656039536, + "val_right_shoulder_rmse": 0.02264991977036461, + "val_right_shoulder_mae": 0.017645670101046562, + "val_right_shoulder_r2": 0.9203363060951233, + "val_right_elbow_mse": 0.0009703408577479422, + "val_right_elbow_rmse": 0.031150294665507457, + "val_right_elbow_mae": 0.022782132029533386, + "val_right_elbow_r2": 0.9386346936225891, + "val_left_hand_mse": 0.0016684813890606165, + "val_left_hand_rmse": 0.040847048719101074, + "val_left_hand_mae": 0.030892495065927505, + "val_left_hand_r2": 0.9349546432495117, + "val_right_hand_mse": 0.0015349863097071648, + "val_right_hand_rmse": 0.039178901333589804, + "val_right_hand_mae": 0.028954461216926575, + "val_right_hand_r2": 0.9438621401786804, + "val_left_hip_mse": 0.0006729575106874108, + "val_left_hip_rmse": 0.025941424607900988, + "val_left_hip_mae": 0.020203160122036934, + "val_left_hip_r2": 0.9107968807220459, + "val_right_hip_mse": 0.0006794946966692805, + "val_right_hip_rmse": 0.026067119071145558, + "val_right_hip_mae": 0.02019696682691574, + "val_right_hip_r2": 0.9133656024932861, + "val_left_knee_mse": 0.000773480802308768, + "val_left_knee_rmse": 0.02781152283332878, + "val_left_knee_mae": 0.01951405219733715, + "val_left_knee_r2": 0.9116611480712891, + "val_right_knee_mse": 0.0007595366332679987, + "val_right_knee_rmse": 0.027559692183839767, + "val_right_knee_mae": 0.019282951951026917, + "val_right_knee_r2": 0.922938346862793, + "val_left_foot_mse": 0.001246402389369905, + "val_left_foot_rmse": 0.0353044245013271, + "val_left_foot_mae": 0.02097209542989731, + "val_left_foot_r2": 0.7417206764221191, + "val_right_foot_mse": 0.0012435478856787086, + "val_right_foot_rmse": 0.0352639743318689, + "val_right_foot_mae": 0.021884478628635406, + "val_right_foot_r2": 0.7038549184799194 + }, + "best_epoch": 100 + }, + { + "fold": 5, + "train_metrics": { + "train_mse": 0.0008729268447495997, + "train_rmse": 0.0295453354144034, + "train_mae": 0.02181827835738659, + "train_r2": 0.9589093923568726, + "train_head_mse": 0.0009298243094235659, + "train_head_rmse": 0.03049302066741775, + "train_head_mae": 0.02325505018234253, + "train_head_r2": 0.9076851606369019, + "train_left_shoulder_mse": 0.0006056821439415216, + "train_left_shoulder_rmse": 0.024610610393517703, + "train_left_shoulder_mae": 0.01857166551053524, + "train_left_shoulder_r2": 0.8913877606391907, + "train_left_elbow_mse": 0.0008716861484572291, + "train_left_elbow_rmse": 0.02952433146503455, + "train_left_elbow_mae": 0.022388337180018425, + "train_left_elbow_r2": 0.9370739459991455, + "train_right_shoulder_mse": 0.0005078943795524538, + "train_right_shoulder_rmse": 0.02253651214257552, + "train_right_shoulder_mae": 0.017517220228910446, + "train_right_shoulder_r2": 0.9232026934623718, + "train_right_elbow_mse": 0.0008395402692258358, + "train_right_elbow_rmse": 0.028974821297565165, + "train_right_elbow_mae": 0.02224080264568329, + "train_right_elbow_r2": 0.9475702047348022, + "train_left_hand_mse": 0.001259981538169086, + "train_left_hand_rmse": 0.03549621864606265, + "train_left_hand_mae": 0.027853189036250114, + "train_left_hand_r2": 0.9524511098861694, + "train_right_hand_mse": 0.0014246104983612895, + "train_right_hand_rmse": 0.03774401274853125, + "train_right_hand_mae": 0.028963418677449226, + "train_right_hand_r2": 0.9496018290519714, + "train_left_hip_mse": 0.0006854351377114654, + "train_left_hip_rmse": 0.02618081621553204, + "train_left_hip_mae": 0.019884435459971428, + "train_left_hip_r2": 0.9123377799987793, + "train_right_hip_mse": 0.0006561009795404971, + "train_right_hip_rmse": 0.025614468168214954, + "train_right_hip_mae": 0.019414903596043587, + "train_right_hip_r2": 0.9197002053260803, + "train_left_knee_mse": 0.0007604913553223014, + "train_left_knee_rmse": 0.027577007729670408, + "train_left_knee_mae": 0.020463483408093452, + "train_left_knee_r2": 0.9199081063270569, + "train_right_knee_mse": 0.0007569370791316032, + "train_right_knee_rmse": 0.02751248951170365, + "train_right_knee_mae": 0.020185161381959915, + "train_right_knee_r2": 0.9284848570823669, + "train_left_foot_mse": 0.0010991545859724283, + "train_left_foot_rmse": 0.033153500357766576, + "train_left_foot_mae": 0.021633299067616463, + "train_left_foot_r2": 0.7621727585792542, + "train_right_foot_mse": 0.0009507107897661626, + "train_right_foot_rmse": 0.030833598391465156, + "train_right_foot_mae": 0.021266644820570946, + "train_right_foot_r2": 0.7845641374588013 + }, + "val_metrics": { + "val_mse": 0.0011362760560587049, + "val_rmse": 0.03370869407228208, + "val_mae": 0.02461574226617813, + "val_r2": 0.9451777338981628, + "val_head_mse": 0.0011023689294233918, + "val_head_rmse": 0.03320194165140635, + "val_head_mae": 0.025341136381030083, + "val_head_r2": 0.8834360241889954, + "val_left_shoulder_mse": 0.0007032135617919266, + "val_left_shoulder_rmse": 0.02651817417907814, + "val_left_shoulder_mae": 0.020607678219676018, + "val_left_shoulder_r2": 0.871929407119751, + "val_left_elbow_mse": 0.0010959375649690628, + "val_left_elbow_rmse": 0.03310494774152442, + "val_left_elbow_mae": 0.025177210569381714, + "val_left_elbow_r2": 0.9195860028266907, + "val_right_shoulder_mse": 0.0006857699481770396, + "val_right_shoulder_rmse": 0.026187209629455362, + "val_right_shoulder_mae": 0.020025238394737244, + "val_right_shoulder_r2": 0.8887978196144104, + "val_right_elbow_mse": 0.0012307463912293315, + "val_right_elbow_rmse": 0.035081995257244585, + "val_right_elbow_mae": 0.02671550214290619, + "val_right_elbow_r2": 0.9169946908950806, + "val_left_hand_mse": 0.0017783312359824777, + "val_left_hand_rmse": 0.0421702648317802, + "val_left_hand_mae": 0.03261606767773628, + "val_left_hand_r2": 0.9301201105117798, + "val_right_hand_mse": 0.002184460638090968, + "val_right_hand_rmse": 0.046738213894959316, + "val_right_hand_mae": 0.0352083183825016, + "val_right_hand_r2": 0.9166122674942017, + "val_left_hip_mse": 0.0008904178394004703, + "val_left_hip_rmse": 0.02983986996286127, + "val_left_hip_mae": 0.022432608529925346, + "val_left_hip_r2": 0.8845265507698059, + "val_right_hip_mse": 0.0008758040494285524, + "val_right_hip_rmse": 0.029593986710623365, + "val_right_hip_mae": 0.021974312141537666, + "val_right_hip_r2": 0.8890302777290344, + "val_left_knee_mse": 0.0009088679798878729, + "val_left_knee_rmse": 0.030147437368504025, + "val_left_knee_mae": 0.023274675011634827, + "val_left_knee_r2": 0.9038561582565308, + "val_right_knee_mse": 0.0011879476951435208, + "val_right_knee_rmse": 0.034466617111975476, + "val_right_knee_mae": 0.024253321811556816, + "val_right_knee_r2": 0.8962328433990479, + "val_left_foot_mse": 0.0008336884202435613, + "val_left_foot_rmse": 0.028873663090151228, + "val_left_foot_mae": 0.020209727808833122, + "val_left_foot_r2": 0.7532256841659546, + "val_right_foot_mse": 0.001294033951126039, + "val_right_foot_rmse": 0.035972683401798634, + "val_right_foot_mae": 0.022168844938278198, + "val_right_foot_r2": 0.7278895378112793 + }, + "best_epoch": 96 + } + ], + "test_metrics": { + "test_mse": 0.0010199841344729066, + "test_rmse": 0.031937190459915325, + "test_mae": 0.02346094697713852, + "test_r2": 0.9521711468696594, + "test_head_mse": 0.0010701818391680717, + "test_head_rmse": 0.03271363384229994, + "test_head_mae": 0.024962028488516808, + "test_head_r2": 0.8949620127677917, + "test_left_shoulder_mse": 0.0007148993317969143, + "test_left_shoulder_rmse": 0.02673760145931034, + "test_left_shoulder_mae": 0.019924800843000412, + "test_left_shoulder_r2": 0.8745908737182617, + "test_left_elbow_mse": 0.0010018351022154093, + "test_left_elbow_rmse": 0.031651778815975085, + "test_left_elbow_mae": 0.024087904021143913, + "test_left_elbow_r2": 0.9260596036911011, + "test_right_shoulder_mse": 0.0006979801109991968, + "test_right_shoulder_rmse": 0.026419313219673154, + "test_right_shoulder_mae": 0.019715825095772743, + "test_right_shoulder_r2": 0.8951577544212341, + "test_right_elbow_mse": 0.0011033808114007115, + "test_right_elbow_rmse": 0.03321717645135889, + "test_right_elbow_mae": 0.025424422696232796, + "test_right_elbow_r2": 0.9294276833534241, + "test_left_hand_mse": 0.0014988387702032924, + "test_left_hand_rmse": 0.03871483914732557, + "test_left_hand_mae": 0.029591461643576622, + "test_left_hand_r2": 0.942769467830658, + "test_right_hand_mse": 0.0017961004050448537, + "test_right_hand_rmse": 0.04238042478603599, + "test_right_hand_mae": 0.033074818551540375, + "test_right_hand_r2": 0.93616783618927, + "test_left_hip_mse": 0.0007702053990215063, + "test_left_hip_rmse": 0.027752574637707153, + "test_left_hip_mae": 0.020965537056326866, + "test_left_hip_r2": 0.9126135110855103, + "test_right_hip_mse": 0.0008085676236078143, + "test_right_hip_rmse": 0.028435323518606472, + "test_right_hip_mae": 0.020978618413209915, + "test_right_hip_r2": 0.9117628335952759, + "test_left_knee_mse": 0.0009110781829804182, + "test_left_knee_rmse": 0.030184071676637964, + "test_left_knee_mae": 0.02164597064256668, + "test_left_knee_r2": 0.9063560962677002, + "test_right_knee_mse": 0.0009026473271660507, + "test_right_knee_rmse": 0.030044089721042486, + "test_right_knee_mae": 0.0221392884850502, + "test_right_knee_r2": 0.915824830532074, + "test_left_foot_mse": 0.000985367689281702, + "test_left_foot_rmse": 0.03139056688372643, + "test_left_foot_mae": 0.02102532796561718, + "test_left_foot_r2": 0.7881198525428772, + "test_right_foot_mse": 0.0009987117955461144, + "test_right_foot_rmse": 0.031602401736990096, + "test_right_foot_mae": 0.021456323564052582, + "test_right_foot_r2": 0.8097579479217529 + } +} \ No newline at end of file diff --git a/A9/cv_results_loss_optimizer/conv1d_v3_sgd_mse_fold1_best.h5 b/A9/cv_results_loss_optimizer/conv1d_v3_sgd_mse_fold1_best.h5 new file mode 100644 index 0000000000000000000000000000000000000000..83bedf7bc31102e738b266191d093db3f968619e --- /dev/null +++ b/A9/cv_results_loss_optimizer/conv1d_v3_sgd_mse_fold1_best.h5 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c1372e468394bddd1f836f7aa78cd8dfc619f685b5f5a411fc69cea8b9d8f94e +size 2658520 diff --git a/A9/cv_results_loss_optimizer/conv1d_v3_sgd_mse_fold2_best.h5 b/A9/cv_results_loss_optimizer/conv1d_v3_sgd_mse_fold2_best.h5 new file mode 100644 index 0000000000000000000000000000000000000000..5f0e5eb758cad34c3be320a055d56417ca18a11e --- /dev/null +++ b/A9/cv_results_loss_optimizer/conv1d_v3_sgd_mse_fold2_best.h5 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:18fd53bd83a3d2ee8f778b2cb1c09e3ec9a8e63f03391fc16d2a240c1192416d +size 2658520 diff --git a/A9/cv_results_loss_optimizer/conv1d_v3_sgd_mse_fold3_best.h5 b/A9/cv_results_loss_optimizer/conv1d_v3_sgd_mse_fold3_best.h5 new file mode 100644 index 0000000000000000000000000000000000000000..802bd3e3d3ab3b736197ce3320e2d3f318ec20ed --- /dev/null +++ b/A9/cv_results_loss_optimizer/conv1d_v3_sgd_mse_fold3_best.h5 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:0f0150eb636083bf90226a05d9a72b920e465c42b0b91217929fbc363b591e02 +size 2658520 diff --git a/A9/cv_results_loss_optimizer/conv1d_v3_sgd_mse_fold4_best.h5 b/A9/cv_results_loss_optimizer/conv1d_v3_sgd_mse_fold4_best.h5 new file mode 100644 index 0000000000000000000000000000000000000000..5c6b11982da097e0d3a6f43065f924a9500d1163 --- /dev/null +++ b/A9/cv_results_loss_optimizer/conv1d_v3_sgd_mse_fold4_best.h5 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e7cc0dffb64ba8017e47c868d272fe89132090e580b7a8499cd20c7354313f25 +size 2658520 diff --git a/A9/cv_results_loss_optimizer/conv1d_v3_sgd_mse_fold5_best.h5 b/A9/cv_results_loss_optimizer/conv1d_v3_sgd_mse_fold5_best.h5 new file mode 100644 index 0000000000000000000000000000000000000000..b36a16e66fdfab6a9a898c916869eb58535714ef --- /dev/null +++ b/A9/cv_results_loss_optimizer/conv1d_v3_sgd_mse_fold5_best.h5 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:700d59c202faafe3b8e0505b39b07c7223b77813bf24a6235b3e1b7eaf245c46 +size 2658520 diff --git a/A9/cv_results_loss_optimizer/conv1d_v3_sgd_mse_predictions.png b/A9/cv_results_loss_optimizer/conv1d_v3_sgd_mse_predictions.png new file mode 100644 index 0000000000000000000000000000000000000000..aeb383f111b9917a222bb0015f9bf3a29774fdb7 --- /dev/null +++ b/A9/cv_results_loss_optimizer/conv1d_v3_sgd_mse_predictions.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:49c21079a722b3c4fe5dbc8aa09ccbb2901c3260a1446b196e0bd0bc03933a9d +size 549392 diff --git a/A9/cv_results_loss_optimizer/conv1d_v3_sgd_mse_test_predictions.png b/A9/cv_results_loss_optimizer/conv1d_v3_sgd_mse_test_predictions.png new file mode 100644 index 0000000000000000000000000000000000000000..8520c916b18452ce64a15c484bc457c11f1c06ab --- /dev/null +++ b/A9/cv_results_loss_optimizer/conv1d_v3_sgd_mse_test_predictions.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:1ee0d3546e9edbb2841560447f5664e6fc4ad90c1f7ee0c59bf9cf796477bb67 +size 470751 diff --git a/A9/cv_results_loss_optimizer/conv1d_v3_sgd_mse_training_history.png b/A9/cv_results_loss_optimizer/conv1d_v3_sgd_mse_training_history.png new file mode 100644 index 0000000000000000000000000000000000000000..2a703bbbb71062069640e0fe4562be82e68864a0 --- /dev/null +++ b/A9/cv_results_loss_optimizer/conv1d_v3_sgd_mse_training_history.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:3ee74f39a1b3cfb0a0360a2b4d2aaf2c0b2a64a0866128ca75fa3808b26c5c03 +size 385089 diff --git a/A9/cv_results_loss_optimizer/summary.txt b/A9/cv_results_loss_optimizer/summary.txt new file mode 100644 index 0000000000000000000000000000000000000000..6c689f463b34ed2deed3a1b60ba9cdd77fbd1738 --- /dev/null +++ b/A9/cv_results_loss_optimizer/summary.txt @@ -0,0 +1,779 @@ +Cross-Validation Training Summary - Optimizer & Loss Function Variants +================================================== + +CONV1D_V3: +---------------------------------------- + + Optimizer: SGD, Loss: MSE + Best Fold: 1 + Val RMSE: 0.032241 ± 0.001373 + Val MAE: 0.023373 ± 0.001020 + Val R²: 0.950748 ± 0.003632 + Test RMSE: 0.031937 + Test MAE: 0.023461 + Test R²: 0.952171 + + Optimizer: SGD, Loss: MAE + Best Fold: 1 + Val RMSE: 0.035625 ± 0.000619 + Val MAE: 0.023964 ± 0.000383 + Val R²: 0.939909 ± 0.001839 + Test RMSE: 0.037500 + Test MAE: 0.024983 + Test R²: 0.934060 + + Optimizer: RMSPROP, Loss: MSE + Best Fold: 4 + Val RMSE: 0.032467 ± 0.000629 + Val MAE: 0.023508 ± 0.000743 + Val R²: 0.950038 ± 0.002828 + Test RMSE: 0.031119 + Test MAE: 0.022788 + Test R²: 0.954592 + + Optimizer: RMSPROP, Loss: MAE + Best Fold: 1 + Val RMSE: 0.034623 ± 0.001290 + Val MAE: 0.023639 ± 0.000801 + Val R²: 0.943196 ± 0.003858 + Test RMSE: 0.034465 + Test MAE: 0.023634 + Test R²: 0.944302 + + Optimizer: ADAM, Loss: MSE + Best Fold: 1 + Val RMSE: 0.032956 ± 0.000841 + Val MAE: 0.023640 ± 0.000489 + Val R²: 0.948571 ± 0.002204 + Test RMSE: 0.031707 + Test MAE: 0.023486 + Test R²: 0.952860 + + Optimizer: ADAM, Loss: MAE + Best Fold: 1 + Val RMSE: 0.035601 ± 0.000889 + Val MAE: 0.024090 ± 0.000520 + Val R²: 0.939982 ± 0.002526 + Test RMSE: 0.037298 + Test MAE: 0.025800 + Test R²: 0.934766 + +CONV1D_V3_SGD_MSE +------------------------------ +Best Fold: 1 + +Validation Metrics (mean ± std): + val_mse: 0.001041 ± 0.000089 + val_rmse: 0.032241 ± 0.001373 + val_mae: 0.023373 ± 0.001020 + val_r2: 0.950748 ± 0.003632 + val_head_mse: 0.001046 ± 0.000103 + val_head_rmse: 0.032298 ± 0.001611 + val_head_mae: 0.024423 ± 0.001232 + val_head_r2: 0.894812 ± 0.008858 + val_left_shoulder_mse: 0.000671 ± 0.000089 + val_left_shoulder_rmse: 0.025856 ± 0.001688 + val_left_shoulder_mae: 0.019546 ± 0.001066 + val_left_shoulder_r2: 0.879510 ± 0.010707 + val_left_elbow_mse: 0.001057 ± 0.000095 + val_left_elbow_rmse: 0.032485 ± 0.001472 + val_left_elbow_mae: 0.024876 ± 0.000722 + val_left_elbow_r2: 0.923367 ± 0.006062 + val_right_shoulder_mse: 0.000614 ± 0.000080 + val_right_shoulder_rmse: 0.024735 ± 0.001634 + val_right_shoulder_mae: 0.018848 ± 0.000997 + val_right_shoulder_r2: 0.905785 ± 0.011198 + val_right_elbow_mse: 0.001062 ± 0.000112 + val_right_elbow_rmse: 0.032545 ± 0.001705 + val_right_elbow_mae: 0.024780 ± 0.001515 + val_right_elbow_r2: 0.932453 ± 0.008157 + val_left_hand_mse: 0.001659 ± 0.000120 + val_left_hand_rmse: 0.040704 ± 0.001490 + val_left_hand_mae: 0.031530 ± 0.000856 + val_left_hand_r2: 0.936634 ± 0.006177 + val_right_hand_mse: 0.001867 ± 0.000249 + val_right_hand_rmse: 0.043112 ± 0.002900 + val_right_hand_mae: 0.032260 ± 0.002238 + val_right_hand_r2: 0.932791 ± 0.009484 + val_left_hip_mse: 0.000782 ± 0.000091 + val_left_hip_rmse: 0.027924 ± 0.001617 + val_left_hip_mae: 0.021069 ± 0.000942 + val_left_hip_r2: 0.899394 ± 0.011315 + val_right_hip_mse: 0.000771 ± 0.000076 + val_right_hip_rmse: 0.027729 ± 0.001363 + val_right_hip_mae: 0.020792 ± 0.000830 + val_right_hip_r2: 0.904674 ± 0.009717 + val_left_knee_mse: 0.000864 ± 0.000091 + val_left_knee_rmse: 0.029355 ± 0.001553 + val_left_knee_mae: 0.021293 ± 0.001541 + val_left_knee_r2: 0.908927 ± 0.007348 + val_right_knee_mse: 0.000904 ± 0.000151 + val_right_knee_rmse: 0.029973 ± 0.002412 + val_right_knee_mae: 0.021576 ± 0.001584 + val_right_knee_r2: 0.916270 ± 0.010366 + val_left_foot_mse: 0.001103 ± 0.000256 + val_left_foot_rmse: 0.032985 ± 0.003830 + val_left_foot_mae: 0.021114 ± 0.000951 + val_left_foot_r2: 0.748310 ± 0.042131 + val_right_foot_mse: 0.001136 ± 0.000125 + val_right_foot_rmse: 0.033658 ± 0.001855 + val_right_foot_mae: 0.021745 ± 0.000740 + val_right_foot_r2: 0.745930 ± 0.028242 + +Test Metrics: + test_mse: 0.001020 + test_rmse: 0.031937 + test_mae: 0.023461 + test_r2: 0.952171 + test_head_mse: 0.001070 + test_head_rmse: 0.032714 + test_head_mae: 0.024962 + test_head_r2: 0.894962 + test_left_shoulder_mse: 0.000715 + test_left_shoulder_rmse: 0.026738 + test_left_shoulder_mae: 0.019925 + test_left_shoulder_r2: 0.874591 + test_left_elbow_mse: 0.001002 + test_left_elbow_rmse: 0.031652 + test_left_elbow_mae: 0.024088 + test_left_elbow_r2: 0.926060 + test_right_shoulder_mse: 0.000698 + test_right_shoulder_rmse: 0.026419 + test_right_shoulder_mae: 0.019716 + test_right_shoulder_r2: 0.895158 + test_right_elbow_mse: 0.001103 + test_right_elbow_rmse: 0.033217 + test_right_elbow_mae: 0.025424 + test_right_elbow_r2: 0.929428 + test_left_hand_mse: 0.001499 + test_left_hand_rmse: 0.038715 + test_left_hand_mae: 0.029591 + test_left_hand_r2: 0.942769 + test_right_hand_mse: 0.001796 + test_right_hand_rmse: 0.042380 + test_right_hand_mae: 0.033075 + test_right_hand_r2: 0.936168 + test_left_hip_mse: 0.000770 + test_left_hip_rmse: 0.027753 + test_left_hip_mae: 0.020966 + test_left_hip_r2: 0.912614 + test_right_hip_mse: 0.000809 + test_right_hip_rmse: 0.028435 + test_right_hip_mae: 0.020979 + test_right_hip_r2: 0.911763 + test_left_knee_mse: 0.000911 + test_left_knee_rmse: 0.030184 + test_left_knee_mae: 0.021646 + test_left_knee_r2: 0.906356 + test_right_knee_mse: 0.000903 + test_right_knee_rmse: 0.030044 + test_right_knee_mae: 0.022139 + test_right_knee_r2: 0.915825 + test_left_foot_mse: 0.000985 + test_left_foot_rmse: 0.031391 + test_left_foot_mae: 0.021025 + test_left_foot_r2: 0.788120 + test_right_foot_mse: 0.000999 + test_right_foot_rmse: 0.031602 + test_right_foot_mae: 0.021456 + test_right_foot_r2: 0.809758 + +CONV1D_V3_SGD_MAE +------------------------------ +Best Fold: 1 + +Validation Metrics (mean ± std): + val_mse: 0.001270 ± 0.000044 + val_rmse: 0.035625 ± 0.000619 + val_mae: 0.023964 ± 0.000383 + val_r2: 0.939909 ± 0.001839 + val_head_mse: 0.001100 ± 0.000087 + val_head_rmse: 0.033140 ± 0.001340 + val_head_mae: 0.024103 ± 0.001135 + val_head_r2: 0.889034 ± 0.010780 + val_left_shoulder_mse: 0.000716 ± 0.000082 + val_left_shoulder_rmse: 0.026707 ± 0.001554 + val_left_shoulder_mae: 0.019292 ± 0.000882 + val_left_shoulder_r2: 0.871047 ± 0.014181 + val_left_elbow_mse: 0.001123 ± 0.000080 + val_left_elbow_rmse: 0.033490 ± 0.001213 + val_left_elbow_mae: 0.025062 ± 0.000738 + val_left_elbow_r2: 0.918471 ± 0.006770 + val_right_shoulder_mse: 0.000696 ± 0.000061 + val_right_shoulder_rmse: 0.026359 ± 0.001131 + val_right_shoulder_mae: 0.019103 ± 0.000775 + val_right_shoulder_r2: 0.892761 ± 0.012788 + val_right_elbow_mse: 0.001403 ± 0.000133 + val_right_elbow_rmse: 0.037420 ± 0.001761 + val_right_elbow_mae: 0.026664 ± 0.000600 + val_right_elbow_r2: 0.910890 ± 0.008175 + val_left_hand_mse: 0.001914 ± 0.000178 + val_left_hand_rmse: 0.043703 ± 0.001987 + val_left_hand_mae: 0.032700 ± 0.000863 + val_left_hand_r2: 0.926901 ± 0.008331 + val_right_hand_mse: 0.002525 ± 0.000218 + val_right_hand_rmse: 0.050196 ± 0.002203 + val_right_hand_mae: 0.035293 ± 0.001264 + val_right_hand_r2: 0.909263 ± 0.006722 + val_left_hip_mse: 0.000874 ± 0.000146 + val_left_hip_rmse: 0.029462 ± 0.002423 + val_left_hip_mae: 0.020973 ± 0.000856 + val_left_hip_r2: 0.887418 ± 0.019151 + val_right_hip_mse: 0.000861 ± 0.000102 + val_right_hip_rmse: 0.029290 ± 0.001723 + val_right_hip_mae: 0.020927 ± 0.000678 + val_right_hip_r2: 0.893227 ± 0.014804 + val_left_knee_mse: 0.001051 ± 0.000137 + val_left_knee_rmse: 0.032347 ± 0.002085 + val_left_knee_mae: 0.022116 ± 0.000784 + val_left_knee_r2: 0.889466 ± 0.009848 + val_right_knee_mse: 0.001192 ± 0.000349 + val_right_knee_rmse: 0.034160 ± 0.005022 + val_right_knee_mae: 0.022635 ± 0.001644 + val_right_knee_r2: 0.890238 ± 0.027501 + val_left_foot_mse: 0.001424 ± 0.000340 + val_left_foot_rmse: 0.037438 ± 0.004760 + val_left_foot_mae: 0.020929 ± 0.001256 + val_left_foot_r2: 0.678785 ± 0.043947 + val_right_foot_mse: 0.001625 ± 0.000250 + val_right_foot_rmse: 0.040194 ± 0.003094 + val_right_foot_mae: 0.021738 ± 0.001342 + val_right_foot_r2: 0.638301 ± 0.043273 + +Test Metrics: + test_mse: 0.001406 + test_rmse: 0.037500 + test_mae: 0.024983 + test_r2: 0.934060 + test_head_mse: 0.001164 + test_head_rmse: 0.034112 + test_head_mae: 0.025526 + test_head_r2: 0.885792 + test_left_shoulder_mse: 0.000861 + test_left_shoulder_rmse: 0.029342 + test_left_shoulder_mae: 0.020318 + test_left_shoulder_r2: 0.848966 + test_left_elbow_mse: 0.001187 + test_left_elbow_rmse: 0.034453 + test_left_elbow_mae: 0.025187 + test_left_elbow_r2: 0.912391 + test_right_shoulder_mse: 0.000799 + test_right_shoulder_rmse: 0.028271 + test_right_shoulder_mae: 0.019846 + test_right_shoulder_r2: 0.879944 + test_right_elbow_mse: 0.001716 + test_right_elbow_rmse: 0.041427 + test_right_elbow_mae: 0.027928 + test_right_elbow_r2: 0.890233 + test_left_hand_mse: 0.001868 + test_left_hand_rmse: 0.043222 + test_left_hand_mae: 0.032468 + test_left_hand_r2: 0.928669 + test_right_hand_mse: 0.002938 + test_right_hand_rmse: 0.054206 + test_right_hand_mae: 0.037380 + test_right_hand_r2: 0.895576 + test_left_hip_mse: 0.000932 + test_left_hip_rmse: 0.030536 + test_left_hip_mae: 0.021851 + test_left_hip_r2: 0.894206 + test_right_hip_mse: 0.000942 + test_right_hip_rmse: 0.030688 + test_right_hip_mae: 0.021654 + test_right_hip_r2: 0.897231 + test_left_knee_mse: 0.001054 + test_left_knee_rmse: 0.032463 + test_left_knee_mae: 0.022155 + test_left_knee_r2: 0.891685 + test_right_knee_mse: 0.001197 + test_right_knee_rmse: 0.034594 + test_right_knee_mae: 0.022923 + test_right_knee_r2: 0.888397 + test_left_foot_mse: 0.001782 + test_left_foot_rmse: 0.042220 + test_left_foot_mae: 0.022858 + test_left_foot_r2: 0.616718 + test_right_foot_mse: 0.001840 + test_right_foot_rmse: 0.042897 + test_right_foot_mae: 0.024683 + test_right_foot_r2: 0.649474 + +CONV1D_V3_RMSPROP_MSE +------------------------------ +Best Fold: 4 + +Validation Metrics (mean ± std): + val_mse: 0.001054 ± 0.000041 + val_rmse: 0.032467 ± 0.000629 + val_mae: 0.023508 ± 0.000743 + val_r2: 0.950038 ± 0.002828 + val_head_mse: 0.001063 ± 0.000078 + val_head_rmse: 0.032576 ± 0.001215 + val_head_mae: 0.024308 ± 0.000890 + val_head_r2: 0.892730 ± 0.010764 + val_left_shoulder_mse: 0.000699 ± 0.000039 + val_left_shoulder_rmse: 0.026428 ± 0.000738 + val_left_shoulder_mae: 0.019816 ± 0.000733 + val_left_shoulder_r2: 0.873859 ± 0.008982 + val_left_elbow_mse: 0.001094 ± 0.000043 + val_left_elbow_rmse: 0.033076 ± 0.000649 + val_left_elbow_mae: 0.025304 ± 0.000628 + val_left_elbow_r2: 0.920571 ± 0.004142 + val_right_shoulder_mse: 0.000632 ± 0.000071 + val_right_shoulder_rmse: 0.025090 ± 0.001458 + val_right_shoulder_mae: 0.019220 ± 0.000928 + val_right_shoulder_r2: 0.902821 ± 0.012510 + val_right_elbow_mse: 0.001077 ± 0.000071 + val_right_elbow_rmse: 0.032792 ± 0.001085 + val_right_elbow_mae: 0.025012 ± 0.001089 + val_right_elbow_r2: 0.931464 ± 0.006602 + val_left_hand_mse: 0.001704 ± 0.000090 + val_left_hand_rmse: 0.041266 ± 0.001084 + val_left_hand_mae: 0.031792 ± 0.000994 + val_left_hand_r2: 0.934955 ± 0.005021 + val_right_hand_mse: 0.001861 ± 0.000125 + val_right_hand_rmse: 0.043118 ± 0.001467 + val_right_hand_mae: 0.031991 ± 0.001409 + val_right_hand_r2: 0.932944 ± 0.005840 + val_left_hip_mse: 0.000795 ± 0.000074 + val_left_hip_rmse: 0.028166 ± 0.001283 + val_left_hip_mae: 0.021137 ± 0.000937 + val_left_hip_r2: 0.897507 ± 0.011738 + val_right_hip_mse: 0.000800 ± 0.000084 + val_right_hip_rmse: 0.028244 ± 0.001463 + val_right_hip_mae: 0.020961 ± 0.000858 + val_right_hip_r2: 0.900903 ± 0.012227 + val_left_knee_mse: 0.000861 ± 0.000052 + val_left_knee_rmse: 0.029328 ± 0.000891 + val_left_knee_mae: 0.021292 ± 0.000901 + val_left_knee_r2: 0.909106 ± 0.004623 + val_right_knee_mse: 0.000884 ± 0.000135 + val_right_knee_rmse: 0.029640 ± 0.002256 + val_right_knee_mae: 0.021653 ± 0.001681 + val_right_knee_r2: 0.917943 ± 0.010613 + val_left_foot_mse: 0.001083 ± 0.000264 + val_left_foot_rmse: 0.032651 ± 0.004157 + val_left_foot_mae: 0.020926 ± 0.000985 + val_left_foot_r2: 0.753628 ± 0.044716 + val_right_foot_mse: 0.001156 ± 0.000172 + val_right_foot_rmse: 0.033900 ± 0.002585 + val_right_foot_mae: 0.022197 ± 0.001155 + val_right_foot_r2: 0.741331 ± 0.040377 + +Test Metrics: + test_mse: 0.000968 + test_rmse: 0.031119 + test_mae: 0.022788 + test_r2: 0.954592 + test_head_mse: 0.000971 + test_head_rmse: 0.031165 + test_head_mae: 0.024398 + test_head_r2: 0.904674 + test_left_shoulder_mse: 0.000619 + test_left_shoulder_rmse: 0.024887 + test_left_shoulder_mae: 0.018825 + test_left_shoulder_r2: 0.891350 + test_left_elbow_mse: 0.000955 + test_left_elbow_rmse: 0.030898 + test_left_elbow_mae: 0.023651 + test_left_elbow_r2: 0.929541 + test_right_shoulder_mse: 0.000618 + test_right_shoulder_rmse: 0.024862 + test_right_shoulder_mae: 0.018796 + test_right_shoulder_r2: 0.907153 + test_right_elbow_mse: 0.001124 + test_right_elbow_rmse: 0.033520 + test_right_elbow_mae: 0.025160 + test_right_elbow_r2: 0.928135 + test_left_hand_mse: 0.001355 + test_left_hand_rmse: 0.036806 + test_left_hand_mae: 0.027778 + test_left_hand_r2: 0.948274 + test_right_hand_mse: 0.001744 + test_right_hand_rmse: 0.041756 + test_right_hand_mae: 0.031959 + test_right_hand_r2: 0.938035 + test_left_hip_mse: 0.000700 + test_left_hip_rmse: 0.026451 + test_left_hip_mae: 0.020167 + test_left_hip_r2: 0.920618 + test_right_hip_mse: 0.000734 + test_right_hip_rmse: 0.027097 + test_right_hip_mae: 0.020460 + test_right_hip_r2: 0.919876 + test_left_knee_mse: 0.000829 + test_left_knee_rmse: 0.028790 + test_left_knee_mae: 0.020761 + test_left_knee_r2: 0.914804 + test_right_knee_mse: 0.000778 + test_right_knee_rmse: 0.027900 + test_right_knee_mae: 0.020927 + test_right_knee_r2: 0.927408 + test_left_foot_mse: 0.001116 + test_left_foot_rmse: 0.033402 + test_left_foot_mae: 0.021324 + test_left_foot_r2: 0.760098 + test_right_foot_mse: 0.001047 + test_right_foot_rmse: 0.032351 + test_right_foot_mae: 0.022041 + test_right_foot_r2: 0.800636 + +CONV1D_V3_RMSPROP_MAE +------------------------------ +Best Fold: 1 + +Validation Metrics (mean ± std): + val_mse: 0.001200 ± 0.000087 + val_rmse: 0.034623 ± 0.001290 + val_mae: 0.023639 ± 0.000801 + val_r2: 0.943196 ± 0.003858 + val_head_mse: 0.001068 ± 0.000062 + val_head_rmse: 0.032674 ± 0.000946 + val_head_mae: 0.024026 ± 0.000834 + val_head_r2: 0.892340 ± 0.007037 + val_left_shoulder_mse: 0.000710 ± 0.000061 + val_left_shoulder_rmse: 0.026620 ± 0.001143 + val_left_shoulder_mae: 0.019455 ± 0.000834 + val_left_shoulder_r2: 0.872298 ± 0.006826 + val_left_elbow_mse: 0.001151 ± 0.000077 + val_left_elbow_rmse: 0.033910 ± 0.001150 + val_left_elbow_mae: 0.025299 ± 0.000929 + val_left_elbow_r2: 0.916504 ± 0.005497 + val_right_shoulder_mse: 0.000650 ± 0.000058 + val_right_shoulder_rmse: 0.025464 ± 0.001125 + val_right_shoulder_mae: 0.018787 ± 0.000715 + val_right_shoulder_r2: 0.900289 ± 0.007994 + val_right_elbow_mse: 0.001307 ± 0.000131 + val_right_elbow_rmse: 0.036105 ± 0.001847 + val_right_elbow_mae: 0.025787 ± 0.001238 + val_right_elbow_r2: 0.917106 ± 0.007075 + val_left_hand_mse: 0.001834 ± 0.000159 + val_left_hand_rmse: 0.042781 ± 0.001919 + val_left_hand_mae: 0.032390 ± 0.001303 + val_left_hand_r2: 0.929957 ± 0.007527 + val_right_hand_mse: 0.002357 ± 0.000272 + val_right_hand_rmse: 0.048460 ± 0.002857 + val_right_hand_mae: 0.034097 ± 0.001977 + val_right_hand_r2: 0.915279 ± 0.009244 + val_left_hip_mse: 0.000848 ± 0.000088 + val_left_hip_rmse: 0.029089 ± 0.001512 + val_left_hip_mae: 0.021178 ± 0.000974 + val_left_hip_r2: 0.890800 ± 0.011662 + val_right_hip_mse: 0.000844 ± 0.000071 + val_right_hip_rmse: 0.029021 ± 0.001236 + val_right_hip_mae: 0.021117 ± 0.000982 + val_right_hip_r2: 0.895723 ± 0.008268 + val_left_knee_mse: 0.000972 ± 0.000049 + val_left_knee_rmse: 0.031174 ± 0.000794 + val_left_knee_mae: 0.021557 ± 0.000859 + val_left_knee_r2: 0.897245 ± 0.005647 + val_right_knee_mse: 0.001100 ± 0.000320 + val_right_knee_rmse: 0.032825 ± 0.004694 + val_right_knee_mae: 0.021865 ± 0.001906 + val_right_knee_r2: 0.898813 ± 0.024851 + val_left_foot_mse: 0.001341 ± 0.000343 + val_left_foot_rmse: 0.036325 ± 0.004628 + val_left_foot_mae: 0.020554 ± 0.001215 + val_left_foot_r2: 0.695945 ± 0.051237 + val_right_foot_mse: 0.001424 ± 0.000354 + val_right_foot_rmse: 0.037481 ± 0.004375 + val_right_foot_mae: 0.021192 ± 0.000590 + val_right_foot_r2: 0.683612 ± 0.067415 + +Test Metrics: + test_mse: 0.001188 + test_rmse: 0.034465 + test_mae: 0.023634 + test_r2: 0.944302 + test_head_mse: 0.001063 + test_head_rmse: 0.032609 + test_head_mae: 0.024371 + test_head_r2: 0.895631 + test_left_shoulder_mse: 0.000725 + test_left_shoulder_rmse: 0.026927 + test_left_shoulder_mae: 0.019381 + test_left_shoulder_r2: 0.872806 + test_left_elbow_mse: 0.001046 + test_left_elbow_rmse: 0.032334 + test_left_elbow_mae: 0.023429 + test_left_elbow_r2: 0.922836 + test_right_shoulder_mse: 0.000729 + test_right_shoulder_rmse: 0.026996 + test_right_shoulder_mae: 0.019621 + test_right_shoulder_r2: 0.890527 + test_right_elbow_mse: 0.001342 + test_right_elbow_rmse: 0.036638 + test_right_elbow_mae: 0.025898 + test_right_elbow_r2: 0.914146 + test_left_hand_mse: 0.001592 + test_left_hand_rmse: 0.039905 + test_left_hand_mae: 0.029758 + test_left_hand_r2: 0.939196 + test_right_hand_mse: 0.002156 + test_right_hand_rmse: 0.046435 + test_right_hand_mae: 0.034087 + test_right_hand_r2: 0.923370 + test_left_hip_mse: 0.000897 + test_left_hip_rmse: 0.029942 + test_left_hip_mae: 0.021813 + test_left_hip_r2: 0.898281 + test_right_hip_mse: 0.000897 + test_right_hip_rmse: 0.029955 + test_right_hip_mae: 0.021621 + test_right_hip_r2: 0.902077 + test_left_knee_mse: 0.001016 + test_left_knee_rmse: 0.031882 + test_left_knee_mae: 0.021726 + test_left_knee_r2: 0.895525 + test_right_knee_mse: 0.000978 + test_right_knee_rmse: 0.031271 + test_right_knee_mae: 0.021896 + test_right_knee_r2: 0.908810 + test_left_foot_mse: 0.001650 + test_left_foot_rmse: 0.040625 + test_left_foot_mae: 0.021258 + test_left_foot_r2: 0.645123 + test_right_foot_mse: 0.001349 + test_right_foot_rmse: 0.036732 + test_right_foot_mae: 0.022384 + test_right_foot_r2: 0.742987 + +CONV1D_V3_ADAM_MSE +------------------------------ +Best Fold: 1 + +Validation Metrics (mean ± std): + val_mse: 0.001087 ± 0.000056 + val_rmse: 0.032956 ± 0.000841 + val_mae: 0.023640 ± 0.000489 + val_r2: 0.948571 ± 0.002204 + val_head_mse: 0.001048 ± 0.000067 + val_head_rmse: 0.032359 ± 0.001034 + val_head_mae: 0.024258 ± 0.000878 + val_head_r2: 0.894584 ± 0.003348 + val_left_shoulder_mse: 0.000677 ± 0.000069 + val_left_shoulder_rmse: 0.025995 ± 0.001284 + val_left_shoulder_mae: 0.019635 ± 0.000641 + val_left_shoulder_r2: 0.878256 ± 0.007028 + val_left_elbow_mse: 0.001088 ± 0.000121 + val_left_elbow_rmse: 0.032941 ± 0.001826 + val_left_elbow_mae: 0.025225 ± 0.001215 + val_left_elbow_r2: 0.921138 ± 0.007851 + val_right_shoulder_mse: 0.000636 ± 0.000033 + val_right_shoulder_rmse: 0.025211 ± 0.000656 + val_right_shoulder_mae: 0.019013 ± 0.000455 + val_right_shoulder_r2: 0.902391 ± 0.003163 + val_right_elbow_mse: 0.001127 ± 0.000081 + val_right_elbow_rmse: 0.033550 ± 0.001204 + val_right_elbow_mae: 0.025310 ± 0.000966 + val_right_elbow_r2: 0.928556 ± 0.002307 + val_left_hand_mse: 0.001691 ± 0.000133 + val_left_hand_rmse: 0.041086 ± 0.001649 + val_left_hand_mae: 0.031744 ± 0.001123 + val_left_hand_r2: 0.935481 ± 0.005928 + val_right_hand_mse: 0.001940 ± 0.000182 + val_right_hand_rmse: 0.043995 ± 0.002095 + val_right_hand_mae: 0.032763 ± 0.001364 + val_right_hand_r2: 0.930288 ± 0.005564 + val_left_hip_mse: 0.000816 ± 0.000069 + val_left_hip_rmse: 0.028537 ± 0.001213 + val_left_hip_mae: 0.021349 ± 0.000486 + val_left_hip_r2: 0.894853 ± 0.010953 + val_right_hip_mse: 0.000804 ± 0.000069 + val_right_hip_rmse: 0.028325 ± 0.001205 + val_right_hip_mae: 0.020991 ± 0.000617 + val_right_hip_r2: 0.900422 ± 0.010651 + val_left_knee_mse: 0.000905 ± 0.000072 + val_left_knee_rmse: 0.030066 ± 0.001205 + val_left_knee_mae: 0.021556 ± 0.000459 + val_left_knee_r2: 0.904266 ± 0.008531 + val_right_knee_mse: 0.000914 ± 0.000117 + val_right_knee_rmse: 0.030165 ± 0.001925 + val_right_knee_mae: 0.021659 ± 0.001163 + val_right_knee_r2: 0.915041 ± 0.009656 + val_left_foot_mse: 0.001194 ± 0.000369 + val_left_foot_rmse: 0.034089 ± 0.005620 + val_left_foot_mae: 0.021269 ± 0.001436 + val_left_foot_r2: 0.731854 ± 0.061287 + val_right_foot_mse: 0.001288 ± 0.000223 + val_right_foot_rmse: 0.035762 ± 0.003027 + val_right_foot_mae: 0.022544 ± 0.000788 + val_right_foot_r2: 0.711032 ± 0.058026 + +Test Metrics: + test_mse: 0.001005 + test_rmse: 0.031707 + test_mae: 0.023486 + test_r2: 0.952860 + test_head_mse: 0.001046 + test_head_rmse: 0.032339 + test_head_mae: 0.024738 + test_head_r2: 0.897356 + test_left_shoulder_mse: 0.000656 + test_left_shoulder_rmse: 0.025622 + test_left_shoulder_mae: 0.019473 + test_left_shoulder_r2: 0.884837 + test_left_elbow_mse: 0.000941 + test_left_elbow_rmse: 0.030681 + test_left_elbow_mae: 0.023489 + test_left_elbow_r2: 0.930527 + test_right_shoulder_mse: 0.000705 + test_right_shoulder_rmse: 0.026545 + test_right_shoulder_mae: 0.019809 + test_right_shoulder_r2: 0.894157 + test_right_elbow_mse: 0.001110 + test_right_elbow_rmse: 0.033321 + test_right_elbow_mae: 0.025394 + test_right_elbow_r2: 0.928984 + test_left_hand_mse: 0.001396 + test_left_hand_rmse: 0.037370 + test_left_hand_mae: 0.028908 + test_left_hand_r2: 0.946678 + test_right_hand_mse: 0.001822 + test_right_hand_rmse: 0.042679 + test_right_hand_mae: 0.032805 + test_right_hand_r2: 0.935265 + test_left_hip_mse: 0.000846 + test_left_hip_rmse: 0.029091 + test_left_hip_mae: 0.022084 + test_left_hip_r2: 0.903983 + test_right_hip_mse: 0.000878 + test_right_hip_rmse: 0.029638 + test_right_hip_mae: 0.021979 + test_right_hip_r2: 0.904139 + test_left_knee_mse: 0.000869 + test_left_knee_rmse: 0.029484 + test_left_knee_mae: 0.021938 + test_left_knee_r2: 0.910647 + test_right_knee_mse: 0.000883 + test_right_knee_rmse: 0.029720 + test_right_knee_mae: 0.022207 + test_right_knee_r2: 0.917631 + test_left_foot_mse: 0.000837 + test_left_foot_rmse: 0.028928 + test_left_foot_mae: 0.020607 + test_left_foot_r2: 0.820062 + test_right_foot_mse: 0.001078 + test_right_foot_rmse: 0.032837 + test_right_foot_mae: 0.021891 + test_right_foot_r2: 0.794606 + +CONV1D_V3_ADAM_MAE +------------------------------ +Best Fold: 1 + +Validation Metrics (mean ± std): + val_mse: 0.001268 ± 0.000064 + val_rmse: 0.035601 ± 0.000889 + val_mae: 0.024090 ± 0.000520 + val_r2: 0.939982 ± 0.002526 + val_head_mse: 0.001086 ± 0.000106 + val_head_rmse: 0.032922 ± 0.001571 + val_head_mae: 0.024148 ± 0.000801 + val_head_r2: 0.890787 ± 0.007882 + val_left_shoulder_mse: 0.000700 ± 0.000045 + val_left_shoulder_rmse: 0.026453 ± 0.000842 + val_left_shoulder_mae: 0.019395 ± 0.000327 + val_left_shoulder_r2: 0.873916 ± 0.003248 + val_left_elbow_mse: 0.001158 ± 0.000063 + val_left_elbow_rmse: 0.034020 ± 0.000924 + val_left_elbow_mae: 0.025217 ± 0.000078 + val_left_elbow_r2: 0.915946 ± 0.005490 + val_right_shoulder_mse: 0.000686 ± 0.000085 + val_right_shoulder_rmse: 0.026136 ± 0.001610 + val_right_shoulder_mae: 0.019132 ± 0.000864 + val_right_shoulder_r2: 0.894700 ± 0.012625 + val_right_elbow_mse: 0.001354 ± 0.000115 + val_right_elbow_rmse: 0.036763 ± 0.001574 + val_right_elbow_mae: 0.026058 ± 0.001300 + val_right_elbow_r2: 0.913871 ± 0.008335 + val_left_hand_mse: 0.001945 ± 0.000236 + val_left_hand_rmse: 0.044021 ± 0.002619 + val_left_hand_mae: 0.033262 ± 0.001029 + val_left_hand_r2: 0.925762 ± 0.010007 + val_right_hand_mse: 0.002553 ± 0.000260 + val_right_hand_rmse: 0.050462 ± 0.002543 + val_right_hand_mae: 0.035610 ± 0.001238 + val_right_hand_r2: 0.908006 ± 0.010369 + val_left_hip_mse: 0.000894 ± 0.000113 + val_left_hip_rmse: 0.029841 ± 0.001887 + val_left_hip_mae: 0.021671 ± 0.000833 + val_left_hip_r2: 0.885617 ± 0.008023 + val_right_hip_mse: 0.000904 ± 0.000149 + val_right_hip_rmse: 0.029973 ± 0.002383 + val_right_hip_mae: 0.021577 ± 0.001021 + val_right_hip_r2: 0.889172 ± 0.009690 + val_left_knee_mse: 0.001074 ± 0.000166 + val_left_knee_rmse: 0.032683 ± 0.002492 + val_left_knee_mae: 0.022078 ± 0.000931 + val_left_knee_r2: 0.886947 ± 0.013710 + val_right_knee_mse: 0.001168 ± 0.000314 + val_right_knee_rmse: 0.033881 ± 0.004509 + val_right_knee_mae: 0.022207 ± 0.001592 + val_right_knee_r2: 0.892272 ± 0.024400 + val_left_foot_mse: 0.001454 ± 0.000381 + val_left_foot_rmse: 0.037750 ± 0.005392 + val_left_foot_mae: 0.021218 ± 0.001852 + val_left_foot_r2: 0.673416 ± 0.053396 + val_right_foot_mse: 0.001510 ± 0.000263 + val_right_foot_rmse: 0.038713 ± 0.003318 + val_right_foot_mae: 0.021598 ± 0.001085 + val_right_foot_r2: 0.664323 ± 0.046299 + +Test Metrics: + test_mse: 0.001391 + test_rmse: 0.037298 + test_mae: 0.025800 + test_r2: 0.934766 + test_head_mse: 0.001113 + test_head_rmse: 0.033360 + test_head_mae: 0.025347 + test_head_r2: 0.890772 + test_left_shoulder_mse: 0.000833 + test_left_shoulder_rmse: 0.028859 + test_left_shoulder_mae: 0.020718 + test_left_shoulder_r2: 0.853901 + test_left_elbow_mse: 0.001253 + test_left_elbow_rmse: 0.035399 + test_left_elbow_mae: 0.025991 + test_left_elbow_r2: 0.907515 + test_right_shoulder_mse: 0.000860 + test_right_shoulder_rmse: 0.029319 + test_right_shoulder_mae: 0.020807 + test_right_shoulder_r2: 0.870880 + test_right_elbow_mse: 0.001664 + test_right_elbow_rmse: 0.040791 + test_right_elbow_mae: 0.029100 + test_right_elbow_r2: 0.893575 + test_left_hand_mse: 0.001941 + test_left_hand_rmse: 0.044058 + test_left_hand_mae: 0.033780 + test_left_hand_r2: 0.925881 + test_right_hand_mse: 0.002751 + test_right_hand_rmse: 0.052448 + test_right_hand_mae: 0.038741 + test_right_hand_r2: 0.902239 + test_left_hip_mse: 0.001044 + test_left_hip_rmse: 0.032308 + test_left_hip_mae: 0.023333 + test_left_hip_r2: 0.881568 + test_right_hip_mse: 0.001096 + test_right_hip_rmse: 0.033106 + test_right_hip_mae: 0.023453 + test_right_hip_r2: 0.880399 + test_left_knee_mse: 0.001222 + test_left_knee_rmse: 0.034957 + test_left_knee_mae: 0.023554 + test_left_knee_r2: 0.874396 + test_right_knee_mse: 0.001066 + test_right_knee_rmse: 0.032651 + test_right_knee_mae: 0.023204 + test_right_knee_r2: 0.900581 + test_left_foot_mse: 0.001649 + test_left_foot_rmse: 0.040612 + test_left_foot_mae: 0.023026 + test_left_foot_r2: 0.645357 + test_right_foot_mse: 0.001594 + test_right_foot_rmse: 0.039921 + test_right_foot_mae: 0.024339 + test_right_foot_r2: 0.696419 diff --git a/A9/cv_results_optimizer/conv1d_v3_adam_cv_results.json b/A9/cv_results_optimizer/conv1d_v3_adam_cv_results.json new file mode 100644 index 0000000000000000000000000000000000000000..f5781500d7243bca32e9621f2db33178863dee71 --- /dev/null +++ b/A9/cv_results_optimizer/conv1d_v3_adam_cv_results.json @@ -0,0 +1,1243 @@ +{ + "model_name": "conv1d_v3_adam", + "n_splits": 5, + "epochs": 100, + "batch_size": 32, + "patience": 10, + "aggregate_results": { + "n_folds": 5, + "train_metrics": { + "train_mse": { + "mean": 0.0008022687514312565, + "std": 4.931776852749633e-05 + }, + "train_rmse": { + "mean": 0.02831077167680991, + "std": 0.0008769027852586857 + }, + "train_mae": { + "mean": 0.020969561859965326, + "std": 0.0006555657441224822 + }, + "train_r2": { + "mean": 0.9620649933815002, + "std": 0.002063107071330804 + }, + "train_head_mse": { + "mean": 0.0008610238204710185, + "std": 5.511879669767857e-05 + }, + "train_head_rmse": { + "mean": 0.029327870066104534, + "std": 0.0009486083789996706 + }, + "train_head_mae": { + "mean": 0.022207555174827576, + "std": 0.000752687485107249 + }, + "train_head_r2": { + "mean": 0.9135042667388916, + "std": 0.004630315390734466 + }, + "train_left_shoulder_mse": { + "mean": 0.0005409527686424554, + "std": 2.2906219421601745e-05 + }, + "train_left_shoulder_rmse": { + "mean": 0.023253171426674775, + "std": 0.0004927344560015893 + }, + "train_left_shoulder_mae": { + "mean": 0.017779596149921417, + "std": 0.00043552849230882616 + }, + "train_left_shoulder_r2": { + "mean": 0.9027207016944885, + "std": 0.0031320155846084924 + }, + "train_left_elbow_mse": { + "mean": 0.0008295898558571934, + "std": 6.229436748417185e-05 + }, + "train_left_elbow_rmse": { + "mean": 0.028782109241005088, + "std": 0.0010862980696121369 + }, + "train_left_elbow_mae": { + "mean": 0.022192852571606636, + "std": 0.0007829420630691152 + }, + "train_left_elbow_r2": { + "mean": 0.9399189352989197, + "std": 0.004436156187505968 + }, + "train_right_shoulder_mse": { + "mean": 0.0004836769658140838, + "std": 2.599648406090751e-05 + }, + "train_right_shoulder_rmse": { + "mean": 0.021984515885986676, + "std": 0.0005983536353055934 + }, + "train_right_shoulder_mae": { + "mean": 0.017034673318266868, + "std": 0.0005164778179215691 + }, + "train_right_shoulder_r2": { + "mean": 0.9258910059928894, + "std": 0.0032724726813268723 + }, + "train_right_elbow_mse": { + "mean": 0.0007903240621089935, + "std": 6.12962341420705e-05 + }, + "train_right_elbow_rmse": { + "mean": 0.028091379821119228, + "std": 0.0010947336911820427 + }, + "train_right_elbow_mae": { + "mean": 0.02164313942193985, + "std": 0.0008465498368682851 + }, + "train_right_elbow_r2": { + "mean": 0.9499314904212952, + "std": 0.003415261756352033 + }, + "train_left_hand_mse": { + "mean": 0.0012106648180633783, + "std": 8.935007267757482e-05 + }, + "train_left_hand_rmse": { + "mean": 0.03477025835131921, + "std": 0.001301519206886792 + }, + "train_left_hand_mae": { + "mean": 0.027260135486721994, + "std": 0.0010428121752570027 + }, + "train_left_hand_r2": { + "mean": 0.953951382637024, + "std": 0.0032975210837977442 + }, + "train_right_hand_mse": { + "mean": 0.0013144704746082426, + "std": 8.192705435082773e-05 + }, + "train_right_hand_rmse": { + "mean": 0.036237759469469284, + "std": 0.0011380963233100702 + }, + "train_right_hand_mae": { + "mean": 0.027648439258337022, + "std": 0.0009466874283275161 + }, + "train_right_hand_r2": { + "mean": 0.9528268575668335, + "std": 0.0025157840254366338 + }, + "train_left_hip_mse": { + "mean": 0.0006334094214253128, + "std": 3.4893328179793954e-05 + }, + "train_left_hip_rmse": { + "mean": 0.025157724409851163, + "std": 0.0007059206352642512 + }, + "train_left_hip_mae": { + "mean": 0.01919970028102398, + "std": 0.0005101738216889089 + }, + "train_left_hip_r2": { + "mean": 0.9187523245811462, + "std": 0.004487562460105181 + }, + "train_right_hip_mse": { + "mean": 0.0006238817237317562, + "std": 3.358972564348308e-05 + }, + "train_right_hip_rmse": { + "mean": 0.024968298054719056, + "std": 0.0006825071299767005 + }, + "train_right_hip_mae": { + "mean": 0.018917780369520187, + "std": 0.0004909443720670777 + }, + "train_right_hip_r2": { + "mean": 0.9230892419815063, + "std": 0.004406189061936453 + }, + "train_left_knee_mse": { + "mean": 0.0006855013431049884, + "std": 4.1509901950933854e-05 + }, + "train_left_knee_rmse": { + "mean": 0.02616999120020127, + "std": 0.0007955524409970993 + }, + "train_left_knee_mae": { + "mean": 0.01933443695306778, + "std": 0.0006589001261561441 + }, + "train_left_knee_r2": { + "mean": 0.9277706742286682, + "std": 0.0037446069801626287 + }, + "train_right_knee_mse": { + "mean": 0.0006661620573140681, + "std": 5.878841547449146e-05 + }, + "train_right_knee_rmse": { + "mean": 0.025785168001047, + "std": 0.001134534473627817 + }, + "train_right_knee_mae": { + "mean": 0.019333491101861, + "std": 0.0007033885209822197 + }, + "train_right_knee_r2": { + "mean": 0.9380330204963684, + "std": 0.005863021250961256 + }, + "train_left_foot_mse": { + "mean": 0.0009011184563860297, + "std": 8.69134186829372e-05 + }, + "train_left_foot_rmse": { + "mean": 0.029983977931456796, + "std": 0.0014420554052953388 + }, + "train_left_foot_mae": { + "mean": 0.019522371515631676, + "std": 0.0008865570402314158 + }, + "train_left_foot_r2": { + "mean": 0.7940123081207275, + "std": 0.0179394377935322 + }, + "train_right_foot_mse": { + "mean": 0.0008887180709280073, + "std": 6.924550177114311e-05 + }, + "train_right_foot_rmse": { + "mean": 0.029788535350796997, + "std": 0.001166718724596938 + }, + "train_right_foot_mae": { + "mean": 0.02053012400865555, + "std": 0.0007130652208698141 + }, + "train_right_foot_r2": { + "mean": 0.80175621509552, + "std": 0.016055554084124315 + } + }, + "val_metrics": { + "val_mse": { + "mean": 0.0010614968137815594, + "std": 8.001605953628374e-05 + }, + "val_rmse": { + "mean": 0.03255770982938862, + "std": 0.001221615506978951 + }, + "val_mae": { + "mean": 0.023489999771118163, + "std": 0.0006254969363884396 + }, + "val_r2": { + "mean": 0.9496784448623657, + "std": 0.004671084543969865 + }, + "val_head_mse": { + "mean": 0.0010572605300694705, + "std": 4.9567424057057754e-05 + }, + "val_head_rmse": { + "mean": 0.03250629399145759, + "std": 0.0007754875952575391 + }, + "val_head_mae": { + "mean": 0.024242159724235535, + "std": 0.0008884707703334784 + }, + "val_head_r2": { + "mean": 0.8934983015060425, + "std": 0.00539030358471048 + }, + "val_left_shoulder_mse": { + "mean": 0.0006873111124150455, + "std": 5.454962834475188e-05 + }, + "val_left_shoulder_rmse": { + "mean": 0.026196279785224607, + "std": 0.0010324910795150722 + }, + "val_left_shoulder_mae": { + "mean": 0.0198053564876318, + "std": 0.0007772292952355029 + }, + "val_left_shoulder_r2": { + "mean": 0.8763010859489441, + "std": 0.006259105336798023 + }, + "val_left_elbow_mse": { + "mean": 0.0010923350579105317, + "std": 0.00011649712719341095 + }, + "val_left_elbow_rmse": { + "mean": 0.03300277614786249, + "std": 0.001775337839570423 + }, + "val_left_elbow_mae": { + "mean": 0.02522401660680771, + "std": 0.0011847401224464181 + }, + "val_left_elbow_r2": { + "mean": 0.9208132982254028, + "std": 0.008026993433377025 + }, + "val_right_shoulder_mse": { + "mean": 0.0006204377627000212, + "std": 3.536158677831358e-05 + }, + "val_right_shoulder_rmse": { + "mean": 0.02489852323632204, + "std": 0.0007080277892518462 + }, + "val_right_shoulder_mae": { + "mean": 0.019003575667738914, + "std": 0.0005652377078576536 + }, + "val_right_shoulder_r2": { + "mean": 0.9046044111251831, + "std": 0.007138037783929853 + }, + "val_right_elbow_mse": { + "mean": 0.0010873436229303479, + "std": 7.714751401196565e-05 + }, + "val_right_elbow_rmse": { + "mean": 0.03295420757519278, + "std": 0.00116782962004521 + }, + "val_right_elbow_mae": { + "mean": 0.024839970469474792, + "std": 0.000611443348396553 + }, + "val_right_elbow_r2": { + "mean": 0.9307620167732239, + "std": 0.006868311240250766 + }, + "val_left_hand_mse": { + "mean": 0.001702110469341278, + "std": 0.00019971570400354624 + }, + "val_left_hand_rmse": { + "mean": 0.041185533330693656, + "std": 0.0024212215114690146 + }, + "val_left_hand_mae": { + "mean": 0.03163791373372078, + "std": 0.0012252524588020012 + }, + "val_left_hand_r2": { + "mean": 0.9349326610565185, + "std": 0.009125030299377668 + }, + "val_right_hand_mse": { + "mean": 0.001837017317302525, + "std": 0.00011438146346696293 + }, + "val_right_hand_rmse": { + "mean": 0.04283997667198235, + "std": 0.0013242794442764791 + }, + "val_right_hand_mae": { + "mean": 0.03180103413760662, + "std": 0.0008661251827479349 + }, + "val_right_hand_r2": { + "mean": 0.9337132096290588, + "std": 0.006597392427329876 + }, + "val_left_hip_mse": { + "mean": 0.0007935149478726089, + "std": 7.022537678994167e-05 + }, + "val_left_hip_rmse": { + "mean": 0.028142469621236132, + "std": 0.0012314022455763578 + }, + "val_left_hip_mae": { + "mean": 0.021356090158224105, + "std": 0.0008488489646342701 + }, + "val_left_hip_r2": { + "mean": 0.8979122281074524, + "std": 0.009255684989519846 + }, + "val_right_hip_mse": { + "mean": 0.0007861085934564472, + "std": 6.906163579993184e-05 + }, + "val_right_hip_rmse": { + "mean": 0.02801126379429759, + "std": 0.001215604418682307 + }, + "val_right_hip_mae": { + "mean": 0.021144339814782143, + "std": 0.0008486414973177481 + }, + "val_right_hip_r2": { + "mean": 0.9026969790458679, + "std": 0.009752024683352645 + }, + "val_left_knee_mse": { + "mean": 0.0008782385382801294, + "std": 5.677623833086246e-05 + }, + "val_left_knee_rmse": { + "mean": 0.02961942253584745, + "std": 0.0009635076144284237 + }, + "val_left_knee_mae": { + "mean": 0.021396001800894737, + "std": 0.0007133559927025497 + }, + "val_left_knee_r2": { + "mean": 0.9069435358047485, + "std": 0.009306786633009793 + }, + "val_right_knee_mse": { + "mean": 0.0009043014026246965, + "std": 0.00016761746937427494 + }, + "val_right_knee_rmse": { + "mean": 0.029952818160178318, + "std": 0.0026702222544178764 + }, + "val_right_knee_mae": { + "mean": 0.021673257276415826, + "std": 0.0015506152213261002 + }, + "val_right_knee_r2": { + "mean": 0.9160343289375306, + "std": 0.01354794935182328 + }, + "val_left_foot_mse": { + "mean": 0.0011391793377697468, + "std": 0.0003579998873510199 + }, + "val_left_foot_rmse": { + "mean": 0.03332381791891315, + "std": 0.0053574711456870925 + }, + "val_left_foot_mae": { + "mean": 0.0209686029702425, + "std": 0.0016670426357411755 + }, + "val_left_foot_r2": { + "mean": 0.742630934715271, + "std": 0.060461738229602685 + }, + "val_right_foot_mse": { + "mean": 0.0012143000727519392, + "std": 0.0002588213634944722 + }, + "val_right_foot_rmse": { + "mean": 0.0346565689512104, + "std": 0.003636248242617392 + }, + "val_right_foot_mae": { + "mean": 0.022277678176760674, + "std": 0.001069855193130743 + }, + "val_right_foot_r2": { + "mean": 0.7273741006851197, + "std": 0.06469133968640077 + } + }, + "best_fold": "1", + "best_fold_metrics": { + "fold": 1, + "train_metrics": { + "train_mse": 0.0008225670317187905, + "train_rmse": 0.028680429420055595, + "train_mae": 0.02142604999244213, + "train_r2": 0.9613585472106934, + "train_head_mse": 0.0009150367695838213, + "train_head_rmse": 0.030249574700875075, + "train_head_mae": 0.022983316332101822, + "train_head_r2": 0.909372091293335, + "train_left_shoulder_mse": 0.0005690310499630868, + "train_left_shoulder_rmse": 0.023854371715957785, + "train_left_shoulder_mae": 0.018318556249141693, + "train_left_shoulder_r2": 0.8992751240730286, + "train_left_elbow_mse": 0.0008461661054752767, + "train_left_elbow_rmse": 0.02908893441629096, + "train_left_elbow_mae": 0.02233702689409256, + "train_left_elbow_r2": 0.9386917948722839, + "train_right_shoulder_mse": 0.0005110198981128633, + "train_right_shoulder_rmse": 0.022605749226974614, + "train_right_shoulder_mae": 0.01757517084479332, + "train_right_shoulder_r2": 0.922666072845459, + "train_right_elbow_mse": 0.0008002259419299662, + "train_right_elbow_rmse": 0.02828826509225983, + "train_right_elbow_mae": 0.021982571110129356, + "train_right_elbow_r2": 0.9497309923171997, + "train_left_hand_mse": 0.0012923228787258267, + "train_left_hand_rmse": 0.035948892593873134, + "train_left_hand_mae": 0.028157377615571022, + "train_left_hand_r2": 0.9504039287567139, + "train_right_hand_mse": 0.0013607594883069396, + "train_right_hand_rmse": 0.036888473651086995, + "train_right_hand_mae": 0.02822374738752842, + "train_right_hand_r2": 0.9513070583343506, + "train_left_hip_mse": 0.0006482442258857191, + "train_left_hip_rmse": 0.02546064072025131, + "train_left_hip_mae": 0.01962846890091896, + "train_left_hip_r2": 0.9189090728759766, + "train_right_hip_mse": 0.0006288366857916117, + "train_right_hip_rmse": 0.025076616314638856, + "train_right_hip_mae": 0.01918206363916397, + "train_right_hip_r2": 0.9242262840270996, + "train_left_knee_mse": 0.0007065677782520652, + "train_left_knee_rmse": 0.026581342672108668, + "train_left_knee_mae": 0.019755028188228607, + "train_left_knee_r2": 0.9260936975479126, + "train_right_knee_mse": 0.0007124234107322991, + "train_right_knee_rmse": 0.026691260943093324, + "train_right_knee_mae": 0.020235450938344002, + "train_right_knee_r2": 0.9337556958198547, + "train_left_foot_mse": 0.0008249497623182833, + "train_left_foot_rmse": 0.02872193869358897, + "train_left_foot_mae": 0.01937454752624035, + "train_left_foot_r2": 0.8127405643463135, + "train_right_foot_mse": 0.0008777876500971615, + "train_right_foot_rmse": 0.029627481332323232, + "train_right_foot_mae": 0.020785316824913025, + "train_right_foot_r2": 0.802251398563385 + }, + "val_metrics": { + "val_mse": 0.0009656725451350212, + "val_rmse": 0.031075272245549533, + "val_mae": 0.022590914741158485, + "val_r2": 0.9529955387115479, + "val_head_mse": 0.0009593348950147629, + "val_head_rmse": 0.03097313182445009, + "val_head_mae": 0.022545555606484413, + "val_head_r2": 0.8972062468528748, + "val_left_shoulder_mse": 0.0006244576652534306, + "val_left_shoulder_rmse": 0.024989150951031342, + "val_left_shoulder_mae": 0.01856689713895321, + "val_left_shoulder_r2": 0.8795694708824158, + "val_left_elbow_mse": 0.0009137760498560965, + "val_left_elbow_rmse": 0.03022872888257289, + "val_left_elbow_mae": 0.023121671751141548, + "val_left_elbow_r2": 0.9337736368179321, + "val_right_shoulder_mse": 0.0005740191554650664, + "val_right_shoulder_rmse": 0.02395869686491873, + "val_right_shoulder_mae": 0.018575776368379593, + "val_right_shoulder_r2": 0.9071239233016968, + "val_right_elbow_mse": 0.0010274734813719988, + "val_right_elbow_rmse": 0.032054227199731375, + "val_right_elbow_mae": 0.02416996657848358, + "val_right_elbow_r2": 0.932359516620636, + "val_left_hand_mse": 0.0014064706629142165, + "val_left_hand_rmse": 0.037502942056780245, + "val_left_hand_mae": 0.02969232201576233, + "val_left_hand_r2": 0.9481995701789856, + "val_right_hand_mse": 0.0017203738680109382, + "val_right_hand_rmse": 0.04147738984086315, + "val_right_hand_mae": 0.031216906383633614, + "val_right_hand_r2": 0.9373540282249451, + "val_left_hip_mse": 0.0007021109340712428, + "val_left_hip_rmse": 0.026497375984637476, + "val_left_hip_mae": 0.02009912207722664, + "val_left_hip_r2": 0.899863600730896, + "val_right_hip_mse": 0.0007073612068779767, + "val_right_hip_rmse": 0.0265962630246803, + "val_right_hip_mae": 0.020049534738063812, + "val_right_hip_r2": 0.9041523337364197, + "val_left_knee_mse": 0.0008148963679559529, + "val_left_knee_rmse": 0.028546389753451362, + "val_left_knee_mae": 0.020890112966299057, + "val_left_knee_r2": 0.9112640023231506, + "val_right_knee_mse": 0.0008859908557496965, + "val_right_knee_rmse": 0.029765598528329586, + "val_right_knee_mae": 0.02234547585248947, + "val_right_knee_r2": 0.9176050424575806, + "val_left_foot_mse": 0.0011383057571947575, + "val_left_foot_rmse": 0.033738787132835076, + "val_left_foot_mae": 0.020821522921323776, + "val_left_foot_r2": 0.7319097518920898, + "val_right_foot_mse": 0.001079171895980835, + "val_right_foot_rmse": 0.03285075183280947, + "val_right_foot_mae": 0.021587016060948372, + "val_right_foot_r2": 0.7687445282936096 + }, + "best_epoch": 100 + } + }, + "fold_details": [ + { + "fold": 1, + "train_metrics": { + "train_mse": 0.0008225670317187905, + "train_rmse": 0.028680429420055595, + "train_mae": 0.02142604999244213, + "train_r2": 0.9613585472106934, + "train_head_mse": 0.0009150367695838213, + "train_head_rmse": 0.030249574700875075, + "train_head_mae": 0.022983316332101822, + "train_head_r2": 0.909372091293335, + "train_left_shoulder_mse": 0.0005690310499630868, + "train_left_shoulder_rmse": 0.023854371715957785, + "train_left_shoulder_mae": 0.018318556249141693, + "train_left_shoulder_r2": 0.8992751240730286, + "train_left_elbow_mse": 0.0008461661054752767, + "train_left_elbow_rmse": 0.02908893441629096, + "train_left_elbow_mae": 0.02233702689409256, + "train_left_elbow_r2": 0.9386917948722839, + "train_right_shoulder_mse": 0.0005110198981128633, + "train_right_shoulder_rmse": 0.022605749226974614, + "train_right_shoulder_mae": 0.01757517084479332, + "train_right_shoulder_r2": 0.922666072845459, + "train_right_elbow_mse": 0.0008002259419299662, + "train_right_elbow_rmse": 0.02828826509225983, + "train_right_elbow_mae": 0.021982571110129356, + "train_right_elbow_r2": 0.9497309923171997, + "train_left_hand_mse": 0.0012923228787258267, + "train_left_hand_rmse": 0.035948892593873134, + "train_left_hand_mae": 0.028157377615571022, + "train_left_hand_r2": 0.9504039287567139, + "train_right_hand_mse": 0.0013607594883069396, + "train_right_hand_rmse": 0.036888473651086995, + "train_right_hand_mae": 0.02822374738752842, + "train_right_hand_r2": 0.9513070583343506, + "train_left_hip_mse": 0.0006482442258857191, + "train_left_hip_rmse": 0.02546064072025131, + "train_left_hip_mae": 0.01962846890091896, + "train_left_hip_r2": 0.9189090728759766, + "train_right_hip_mse": 0.0006288366857916117, + "train_right_hip_rmse": 0.025076616314638856, + "train_right_hip_mae": 0.01918206363916397, + "train_right_hip_r2": 0.9242262840270996, + "train_left_knee_mse": 0.0007065677782520652, + "train_left_knee_rmse": 0.026581342672108668, + "train_left_knee_mae": 0.019755028188228607, + "train_left_knee_r2": 0.9260936975479126, + "train_right_knee_mse": 0.0007124234107322991, + "train_right_knee_rmse": 0.026691260943093324, + "train_right_knee_mae": 0.020235450938344002, + "train_right_knee_r2": 0.9337556958198547, + "train_left_foot_mse": 0.0008249497623182833, + "train_left_foot_rmse": 0.02872193869358897, + "train_left_foot_mae": 0.01937454752624035, + "train_left_foot_r2": 0.8127405643463135, + "train_right_foot_mse": 0.0008777876500971615, + "train_right_foot_rmse": 0.029627481332323232, + "train_right_foot_mae": 0.020785316824913025, + "train_right_foot_r2": 0.802251398563385 + }, + "val_metrics": { + "val_mse": 0.0009656725451350212, + "val_rmse": 0.031075272245549533, + "val_mae": 0.022590914741158485, + "val_r2": 0.9529955387115479, + "val_head_mse": 0.0009593348950147629, + "val_head_rmse": 0.03097313182445009, + "val_head_mae": 0.022545555606484413, + "val_head_r2": 0.8972062468528748, + "val_left_shoulder_mse": 0.0006244576652534306, + "val_left_shoulder_rmse": 0.024989150951031342, + "val_left_shoulder_mae": 0.01856689713895321, + "val_left_shoulder_r2": 0.8795694708824158, + "val_left_elbow_mse": 0.0009137760498560965, + "val_left_elbow_rmse": 0.03022872888257289, + "val_left_elbow_mae": 0.023121671751141548, + "val_left_elbow_r2": 0.9337736368179321, + "val_right_shoulder_mse": 0.0005740191554650664, + "val_right_shoulder_rmse": 0.02395869686491873, + "val_right_shoulder_mae": 0.018575776368379593, + "val_right_shoulder_r2": 0.9071239233016968, + "val_right_elbow_mse": 0.0010274734813719988, + "val_right_elbow_rmse": 0.032054227199731375, + "val_right_elbow_mae": 0.02416996657848358, + "val_right_elbow_r2": 0.932359516620636, + "val_left_hand_mse": 0.0014064706629142165, + "val_left_hand_rmse": 0.037502942056780245, + "val_left_hand_mae": 0.02969232201576233, + "val_left_hand_r2": 0.9481995701789856, + "val_right_hand_mse": 0.0017203738680109382, + "val_right_hand_rmse": 0.04147738984086315, + "val_right_hand_mae": 0.031216906383633614, + "val_right_hand_r2": 0.9373540282249451, + "val_left_hip_mse": 0.0007021109340712428, + "val_left_hip_rmse": 0.026497375984637476, + "val_left_hip_mae": 0.02009912207722664, + "val_left_hip_r2": 0.899863600730896, + "val_right_hip_mse": 0.0007073612068779767, + "val_right_hip_rmse": 0.0265962630246803, + "val_right_hip_mae": 0.020049534738063812, + "val_right_hip_r2": 0.9041523337364197, + "val_left_knee_mse": 0.0008148963679559529, + "val_left_knee_rmse": 0.028546389753451362, + "val_left_knee_mae": 0.020890112966299057, + "val_left_knee_r2": 0.9112640023231506, + "val_right_knee_mse": 0.0008859908557496965, + "val_right_knee_rmse": 0.029765598528329586, + "val_right_knee_mae": 0.02234547585248947, + "val_right_knee_r2": 0.9176050424575806, + "val_left_foot_mse": 0.0011383057571947575, + "val_left_foot_rmse": 0.033738787132835076, + "val_left_foot_mae": 0.020821522921323776, + "val_left_foot_r2": 0.7319097518920898, + "val_right_foot_mse": 0.001079171895980835, + "val_right_foot_rmse": 0.03285075183280947, + "val_right_foot_mae": 0.021587016060948372, + "val_right_foot_r2": 0.7687445282936096 + }, + "best_epoch": 100 + }, + { + "fold": 2, + "train_metrics": { + "train_mse": 0.0007643048302270472, + "train_rmse": 0.027646063557531065, + "train_mae": 0.020431796088814735, + "train_r2": 0.9635598063468933, + "train_head_mse": 0.0008213966502808034, + "train_head_rmse": 0.028660018323106554, + "train_head_mae": 0.021732987836003304, + "train_head_r2": 0.9159870743751526, + "train_left_shoulder_mse": 0.0005208700895309448, + "train_left_shoulder_rmse": 0.022822578503117145, + "train_left_shoulder_mae": 0.017431112006306648, + "train_left_shoulder_r2": 0.9058356285095215, + "train_left_elbow_mse": 0.0007591649773530662, + "train_left_elbow_rmse": 0.027552948614496166, + "train_left_elbow_mae": 0.021275216713547707, + "train_left_elbow_r2": 0.945390522480011, + "train_right_shoulder_mse": 0.0004776327987201512, + "train_right_shoulder_rmse": 0.021854811797866188, + "train_right_shoulder_mae": 0.016950242221355438, + "train_right_shoulder_r2": 0.9255762696266174, + "train_right_elbow_mse": 0.0007423469214700162, + "train_right_elbow_rmse": 0.027246044143508546, + "train_right_elbow_mae": 0.02072831615805626, + "train_right_elbow_r2": 0.9525654911994934, + "train_left_hand_mse": 0.0011518102837726474, + "train_left_hand_rmse": 0.033938330597904305, + "train_left_hand_mae": 0.026330407708883286, + "train_left_hand_r2": 0.9563907980918884, + "train_right_hand_mse": 0.0012374233920127153, + "train_right_hand_rmse": 0.03517702932330579, + "train_right_hand_mae": 0.02657577395439148, + "train_right_hand_r2": 0.955344557762146, + "train_left_hip_mse": 0.0006462499150075018, + "train_left_hip_rmse": 0.025421445966103146, + "train_left_hip_mae": 0.01924406923353672, + "train_left_hip_r2": 0.91444993019104, + "train_right_hip_mse": 0.0006360953557305038, + "train_right_hip_rmse": 0.02522093090531164, + "train_right_hip_mae": 0.01898357644677162, + "train_right_hip_r2": 0.9186046719551086, + "train_left_knee_mse": 0.0006378580001182854, + "train_left_knee_rmse": 0.025255850809629943, + "train_left_knee_mae": 0.018698057159781456, + "train_left_knee_r2": 0.9313656091690063, + "train_right_knee_mse": 0.0006324833957478404, + "train_right_knee_rmse": 0.025149222567463998, + "train_right_knee_mae": 0.018794510513544083, + "train_right_knee_r2": 0.9412643909454346, + "train_left_foot_mse": 0.0007986208656802773, + "train_left_foot_rmse": 0.028259880850426057, + "train_left_foot_mae": 0.018404385074973106, + "train_left_foot_r2": 0.8150741457939148, + "train_right_foot_mse": 0.000874010322149843, + "train_right_foot_rmse": 0.02956366557363689, + "train_right_foot_mae": 0.02046467550098896, + "train_right_foot_r2": 0.8086708784103394 + }, + "val_metrics": { + "val_mse": 0.000998190836980939, + "val_rmse": 0.03159415827302476, + "val_mae": 0.023212747648358345, + "val_r2": 0.9541738033294678, + "val_head_mse": 0.0010753999231383204, + "val_head_rmse": 0.03279329082508067, + "val_head_mae": 0.024518122896552086, + "val_head_r2": 0.8987252116203308, + "val_left_shoulder_mse": 0.0006387707544490695, + "val_left_shoulder_rmse": 0.025273914505851078, + "val_left_shoulder_mae": 0.019448839128017426, + "val_left_shoulder_r2": 0.8869872689247131, + "val_left_elbow_mse": 0.0010259050177410245, + "val_left_elbow_rmse": 0.03202975207117633, + "val_left_elbow_mae": 0.02475181594491005, + "val_left_elbow_r2": 0.9233808517456055, + "val_right_shoulder_mse": 0.000650093425065279, + "val_right_shoulder_rmse": 0.0254969297184049, + "val_right_shoulder_mae": 0.019147511571645737, + "val_right_shoulder_r2": 0.9062102437019348, + "val_right_elbow_mse": 0.0010683524888008833, + "val_right_elbow_rmse": 0.03268566182289848, + "val_right_elbow_mae": 0.024696599692106247, + "val_right_elbow_r2": 0.9341591000556946, + "val_left_hand_mse": 0.0017463546246290207, + "val_left_hand_rmse": 0.04178940804353444, + "val_left_hand_mae": 0.03138191998004913, + "val_left_hand_r2": 0.932083010673523, + "val_right_hand_mse": 0.0018025303725153208, + "val_right_hand_rmse": 0.04245621712441325, + "val_right_hand_mae": 0.03134032338857651, + "val_right_hand_r2": 0.9363875985145569, + "val_left_hip_mse": 0.0007981706876307726, + "val_left_hip_rmse": 0.02825191476043301, + "val_left_hip_mae": 0.02183166891336441, + "val_left_hip_r2": 0.9088725447654724, + "val_right_hip_mse": 0.0008062078850343823, + "val_right_hip_rmse": 0.028393800116123632, + "val_right_hip_mae": 0.02166750468313694, + "val_right_hip_r2": 0.9133408665657043, + "val_left_knee_mse": 0.0008050826145336032, + "val_left_knee_rmse": 0.028373977770725117, + "val_left_knee_mae": 0.02089545503258705, + "val_left_knee_r2": 0.9212754964828491, + "val_right_knee_mse": 0.0007719772984273732, + "val_right_knee_rmse": 0.027784479452157694, + "val_right_knee_mae": 0.020664354786276817, + "val_right_knee_r2": 0.9278630614280701, + "val_left_foot_mse": 0.0007692646468058228, + "val_left_foot_rmse": 0.027735620541206985, + "val_left_foot_mae": 0.019719887524843216, + "val_left_foot_r2": 0.8320212364196777, + "val_right_foot_mse": 0.0010183717822656035, + "val_right_foot_rmse": 0.031911937927139485, + "val_right_foot_mae": 0.021701714023947716, + "val_right_foot_r2": 0.7543262243270874 + }, + "best_epoch": 100 + }, + { + "fold": 3, + "train_metrics": { + "train_mse": 0.0007264706655405462, + "train_rmse": 0.026953119773795133, + "train_mae": 0.019953226670622826, + "train_r2": 0.9652495384216309, + "train_head_mse": 0.0007726949406787753, + "train_head_rmse": 0.027797390896966846, + "train_head_mae": 0.021021805703639984, + "train_head_r2": 0.9214805364608765, + "train_left_shoulder_mse": 0.0005107567994855344, + "train_left_shoulder_rmse": 0.022599929192046918, + "train_left_shoulder_mae": 0.017223816365003586, + "train_left_shoulder_r2": 0.9062668085098267, + "train_left_elbow_mse": 0.000753789849113673, + "train_left_elbow_rmse": 0.027455233546879055, + "train_left_elbow_mae": 0.021310575306415558, + "train_left_elbow_r2": 0.9448773264884949, + "train_right_shoulder_mse": 0.00043691706378012896, + "train_right_shoulder_rmse": 0.02090256117752389, + "train_right_shoulder_mae": 0.016140904277563095, + "train_right_shoulder_r2": 0.9321154952049255, + "train_right_elbow_mse": 0.0007021812489256263, + "train_right_elbow_rmse": 0.026498702778166825, + "train_right_elbow_mae": 0.02055516466498375, + "train_right_elbow_r2": 0.9547990560531616, + "train_left_hand_mse": 0.0010628241579979658, + "train_left_hand_rmse": 0.03260098400352305, + "train_left_hand_mae": 0.025700503960251808, + "train_left_hand_r2": 0.9591648578643799, + "train_right_hand_mse": 0.0011956270318478346, + "train_right_hand_rmse": 0.03457784018483275, + "train_right_hand_mae": 0.02645607851445675, + "train_right_hand_r2": 0.9563203454017639, + "train_left_hip_mse": 0.0005659887683577836, + "train_left_hip_rmse": 0.02379051845500185, + "train_left_hip_mae": 0.018259786069393158, + "train_left_hip_r2": 0.9270560145378113, + "train_right_hip_mse": 0.000560771964956075, + "train_right_hip_rmse": 0.023680624251824, + "train_right_hip_mae": 0.01798544079065323, + "train_right_hip_r2": 0.9308339357376099, + "train_left_knee_mse": 0.0006349936593323946, + "train_left_knee_rmse": 0.025199080525534948, + "train_left_knee_mae": 0.01839490234851837, + "train_left_knee_r2": 0.9325552582740784, + "train_right_knee_mse": 0.0005892842309549451, + "train_right_knee_rmse": 0.024275177258980932, + "train_right_knee_mae": 0.01836562715470791, + "train_right_knee_r2": 0.9449023008346558, + "train_left_foot_mse": 0.0008794412133283913, + "train_left_foot_rmse": 0.02965537410535216, + "train_left_foot_mae": 0.018778936937451363, + "train_left_foot_r2": 0.7936214804649353, + "train_right_foot_mse": 0.0007788478396832943, + "train_right_foot_rmse": 0.02790784548623011, + "train_right_foot_mae": 0.01919839158654213, + "train_right_foot_r2": 0.8247960805892944 + }, + "val_metrics": { + "val_mse": 0.0010395110584795475, + "val_rmse": 0.03224144938552775, + "val_mae": 0.02327403426170349, + "val_r2": 0.9529217481613159, + "val_head_mse": 0.0010731646325439215, + "val_head_rmse": 0.03275919157341831, + "val_head_mae": 0.024475181475281715, + "val_head_r2": 0.8966430425643921, + "val_left_shoulder_mse": 0.0007772698299959302, + "val_left_shoulder_rmse": 0.027879559357994348, + "val_left_shoulder_mae": 0.020617196336388588, + "val_left_shoulder_r2": 0.8704552054405212, + "val_left_elbow_mse": 0.0011561352293938398, + "val_left_elbow_rmse": 0.03400198860940107, + "val_left_elbow_mae": 0.026002665981650352, + "val_left_elbow_r2": 0.9192959070205688, + "val_right_shoulder_mse": 0.0006024945178069174, + "val_right_shoulder_rmse": 0.024545763744624395, + "val_right_shoulder_mae": 0.01851518824696541, + "val_right_shoulder_r2": 0.9123814702033997, + "val_right_elbow_mse": 0.0009890402434393764, + "val_right_elbow_rmse": 0.03144901021398569, + "val_right_elbow_mae": 0.02468448504805565, + "val_right_elbow_r2": 0.9409027695655823, + "val_left_hand_mse": 0.0016096290200948715, + "val_left_hand_rmse": 0.04012018220415844, + "val_left_hand_mae": 0.031457189470529556, + "val_left_hand_r2": 0.9410360455513, + "val_right_hand_mse": 0.001732240547426045, + "val_right_hand_rmse": 0.04162019398592521, + "val_right_hand_mae": 0.031495314091444016, + "val_right_hand_r2": 0.9417909383773804, + "val_left_hip_mse": 0.0007793009281158447, + "val_left_hip_rmse": 0.027915961887705835, + "val_left_hip_mae": 0.02074461430311203, + "val_left_hip_r2": 0.9019405841827393, + "val_right_hip_mse": 0.0007353043765760958, + "val_right_hip_rmse": 0.027116496391976892, + "val_right_hip_mae": 0.02021082304418087, + "val_right_hip_r2": 0.909683346748352, + "val_left_knee_mse": 0.0009274788899347186, + "val_left_knee_rmse": 0.030454538084409007, + "val_left_knee_mae": 0.02091609127819538, + "val_left_knee_r2": 0.9051192998886108, + "val_right_knee_mse": 0.0007585317362099886, + "val_right_knee_rmse": 0.027541454867344765, + "val_right_knee_mae": 0.02029244415462017, + "val_right_knee_r2": 0.9310340285301208, + "val_left_foot_mse": 0.0014270098181441426, + "val_left_foot_rmse": 0.03777578348815736, + "val_left_foot_mae": 0.021796507760882378, + "val_left_foot_r2": 0.7027842402458191, + "val_right_foot_mse": 0.0009460448054596782, + "val_right_foot_rmse": 0.030757841365409214, + "val_right_foot_mae": 0.02135474421083927, + "val_right_foot_r2": 0.7959193587303162 + }, + "best_epoch": 100 + }, + { + "fold": 4, + "train_metrics": { + "train_mse": 0.0008391302544623613, + "train_rmse": 0.02896774507037718, + "train_mae": 0.02142908051609993, + "train_r2": 0.9605860710144043, + "train_head_mse": 0.0008888598531484604, + "train_head_rmse": 0.029813752751850286, + "train_head_mae": 0.022326992824673653, + "train_head_r2": 0.9107434153556824, + "train_left_shoulder_mse": 0.0005403579561971128, + "train_left_shoulder_rmse": 0.02324560079234591, + "train_left_shoulder_mae": 0.01767861843109131, + "train_left_shoulder_r2": 0.903318464756012, + "train_left_elbow_mse": 0.0008913744240999222, + "train_left_elbow_rmse": 0.029855894294090777, + "train_left_elbow_mae": 0.02285793237388134, + "train_left_elbow_r2": 0.9354212284088135, + "train_right_shoulder_mse": 0.0004899154300801456, + "train_right_shoulder_rmse": 0.02213403329897526, + "train_right_shoulder_mae": 0.016989076510071754, + "train_right_shoulder_r2": 0.9251392483711243, + "train_right_elbow_mse": 0.0008353827288374305, + "train_right_elbow_rmse": 0.028902988233700515, + "train_right_elbow_mae": 0.022295888513326645, + "train_right_elbow_r2": 0.9469866156578064, + "train_left_hand_mse": 0.0012806425802409649, + "train_left_hand_rmse": 0.03578606684508602, + "train_left_hand_mae": 0.02824055776000023, + "train_left_hand_r2": 0.9515629410743713, + "train_right_hand_mse": 0.0013984149554744363, + "train_right_hand_rmse": 0.03739538682076221, + "train_right_hand_mae": 0.028767339885234833, + "train_right_hand_r2": 0.9499868154525757, + "train_left_hip_mse": 0.0006668353453278542, + "train_left_hip_rmse": 0.025823155216352905, + "train_left_hip_mae": 0.019681278616189957, + "train_left_hip_r2": 0.9151633381843567, + "train_right_hip_mse": 0.0006615064339712262, + "train_right_hip_rmse": 0.025719767377859897, + "train_right_hip_mae": 0.019419336691498756, + "train_right_hip_r2": 0.9191556572914124, + "train_left_knee_mse": 0.000710266234818846, + "train_left_knee_rmse": 0.026650820528059657, + "train_left_knee_mae": 0.019772887229919434, + "train_left_knee_r2": 0.926543116569519, + "train_right_knee_mse": 0.0006434056558646262, + "train_right_knee_rmse": 0.025365442157877442, + "train_right_knee_mae": 0.019282350316643715, + "train_right_knee_r2": 0.9414060711860657, + "train_left_foot_mse": 0.0009798004757612944, + "train_left_foot_rmse": 0.03130176473876983, + "train_left_foot_mae": 0.020318495109677315, + "train_left_foot_r2": 0.7699272036552429, + "train_right_foot_mse": 0.0009219311759807169, + "train_right_foot_rmse": 0.03036331958104576, + "train_right_foot_mae": 0.020947294309735298, + "train_right_foot_r2": 0.7976313233375549 + }, + "val_metrics": { + "val_mse": 0.0011831838637590408, + "val_rmse": 0.0343974397849468, + "val_mae": 0.02402014657855034, + "val_r2": 0.9423828125, + "val_head_mse": 0.0010831693653017282, + "val_head_rmse": 0.032911538482752946, + "val_head_mae": 0.024494823068380356, + "val_head_r2": 0.8907265067100525, + "val_left_shoulder_mse": 0.0006868643104098737, + "val_left_shoulder_rmse": 0.026208096275957812, + "val_left_shoulder_mae": 0.019753187894821167, + "val_left_shoulder_r2": 0.8736530542373657, + "val_left_elbow_mse": 0.0012573488056659698, + "val_left_elbow_rmse": 0.03545911456404361, + "val_left_elbow_mae": 0.026401519775390625, + "val_left_elbow_r2": 0.9089525938034058, + "val_right_shoulder_mse": 0.0006039246218279004, + "val_right_shoulder_rmse": 0.024574877859877563, + "val_right_shoulder_mae": 0.018735090270638466, + "val_right_shoulder_r2": 0.9062200784683228, + "val_right_elbow_mse": 0.0011910320026800036, + "val_right_elbow_rmse": 0.034511331511258785, + "val_right_elbow_mae": 0.02465122379362583, + "val_right_elbow_r2": 0.9246779680252075, + "val_left_hand_mse": 0.002019638894125819, + "val_left_hand_rmse": 0.044940392678812, + "val_left_hand_mae": 0.03345556557178497, + "val_left_hand_r2": 0.9212648272514343, + "val_right_hand_mse": 0.001906242105178535, + "val_right_hand_rmse": 0.04366053258010643, + "val_right_hand_mae": 0.03142942860722542, + "val_right_hand_r2": 0.9302845001220703, + "val_left_hip_mse": 0.0007698917179368436, + "val_left_hip_rmse": 0.027746922675079548, + "val_left_hip_mae": 0.021576792001724243, + "val_left_hip_r2": 0.8979478478431702, + "val_right_hip_mse": 0.0007749885553494096, + "val_right_hip_rmse": 0.027838616261398653, + "val_right_hip_mae": 0.021647458896040916, + "val_right_hip_r2": 0.9011903405189514, + "val_left_knee_mse": 0.000938024721108377, + "val_left_knee_rmse": 0.030627189245968637, + "val_left_knee_mae": 0.02154630608856678, + "val_left_knee_r2": 0.89286869764328, + "val_right_knee_mse": 0.0008830093429423869, + "val_right_knee_rmse": 0.029715473123313834, + "val_right_knee_mae": 0.020638499408960342, + "val_right_knee_r2": 0.9104109406471252, + "val_left_foot_mse": 0.0016376563580706716, + "val_left_foot_rmse": 0.04046796706125317, + "val_left_foot_mae": 0.02364707924425602, + "val_left_foot_r2": 0.6606450080871582, + "val_right_foot_mse": 0.0016295993700623512, + "val_right_foot_rmse": 0.040368296595996604, + "val_right_foot_mae": 0.024284956976771355, + "val_right_foot_r2": 0.611918568611145 + }, + "best_epoch": 100 + }, + { + "fold": 5, + "train_metrics": { + "train_mse": 0.0008588709752075374, + "train_rmse": 0.029306500562290567, + "train_mae": 0.021607656031847, + "train_r2": 0.9595710039138794, + "train_head_mse": 0.0009071308886632323, + "train_head_rmse": 0.030118613657723895, + "train_head_mae": 0.022972673177719116, + "train_head_r2": 0.9099382162094116, + "train_left_shoulder_mse": 0.0005637479480355978, + "train_left_shoulder_rmse": 0.023743376929906114, + "train_left_shoulder_mae": 0.01824587769806385, + "train_left_shoulder_r2": 0.898907482624054, + "train_left_elbow_mse": 0.0008974539232440293, + "train_left_elbow_rmse": 0.02995753533326848, + "train_left_elbow_mae": 0.023183511570096016, + "train_left_elbow_r2": 0.9352138042449951, + "train_right_shoulder_mse": 0.00050289963837713, + "train_right_shoulder_rmse": 0.022425423928593414, + "train_right_shoulder_mae": 0.017517972737550735, + "train_right_shoulder_r2": 0.9239579439163208, + "train_right_elbow_mse": 0.0008714834693819284, + "train_right_elbow_rmse": 0.029520898857960415, + "train_right_elbow_mae": 0.02265375666320324, + "train_right_elbow_r2": 0.9455752968788147, + "train_left_hand_mse": 0.0012657241895794868, + "train_left_hand_rmse": 0.03557701771620953, + "train_left_hand_mae": 0.027871830388903618, + "train_left_hand_r2": 0.9522343873977661, + "train_right_hand_mse": 0.0013801275053992867, + "train_right_hand_rmse": 0.03715006736735866, + "train_right_hand_mae": 0.028219256550073624, + "train_right_hand_r2": 0.9511755108833313, + "train_left_hip_mse": 0.0006397288525477052, + "train_left_hip_rmse": 0.025292861691546594, + "train_left_hip_mae": 0.0191848985850811, + "train_left_hip_r2": 0.9181832671165466, + "train_right_hip_mse": 0.0006321981782093644, + "train_right_hip_rmse": 0.025143551423960863, + "train_right_hip_mae": 0.01901848427951336, + "train_right_hip_r2": 0.9226256608963013, + "train_left_knee_mse": 0.0007378210430033505, + "train_left_knee_rmse": 0.027162861465673135, + "train_left_knee_mae": 0.02005130983889103, + "train_left_knee_r2": 0.9222956895828247, + "train_right_knee_mse": 0.0007532135932706296, + "train_right_knee_rmse": 0.027444737077819303, + "train_right_knee_mae": 0.019989516586065292, + "train_right_knee_r2": 0.9288366436958313, + "train_left_foot_mse": 0.0010227799648419023, + "train_left_foot_rmse": 0.031980931269146966, + "train_left_foot_mae": 0.020735492929816246, + "train_left_foot_r2": 0.7786981463432312, + "train_right_foot_mse": 0.000991013366729021, + "train_right_foot_rmse": 0.03148036478074898, + "train_right_foot_mae": 0.021254941821098328, + "train_right_foot_r2": 0.7754313945770264 + }, + "val_metrics": { + "val_mse": 0.0011209257645532489, + "val_rmse": 0.03348022945789424, + "val_mae": 0.02435215562582016, + "val_r2": 0.9459183216094971, + "val_head_mse": 0.001095233834348619, + "val_head_rmse": 0.033094317251585945, + "val_head_mae": 0.025177115574479103, + "val_head_r2": 0.8841904997825623, + "val_left_shoulder_mse": 0.0007091930019669235, + "val_left_shoulder_rmse": 0.026630677835288448, + "val_left_shoulder_mae": 0.0206406619399786, + "val_left_shoulder_r2": 0.8708404302597046, + "val_left_elbow_mse": 0.0011085101868957281, + "val_left_elbow_rmse": 0.03329429661211854, + "val_left_elbow_mae": 0.02584240958094597, + "val_left_elbow_r2": 0.918663501739502, + "val_right_shoulder_mse": 0.0006716570933349431, + "val_right_shoulder_rmse": 0.0259163479937846, + "val_right_shoulder_mae": 0.02004431188106537, + "val_right_shoulder_r2": 0.8910863399505615, + "val_right_elbow_mse": 0.0011608198983594775, + "val_right_elbow_rmse": 0.03407080712808955, + "val_right_elbow_mae": 0.025997577235102654, + "val_right_elbow_r2": 0.921710729598999, + "val_left_hand_mse": 0.0017284591449424624, + "val_left_hand_rmse": 0.041574741670183143, + "val_left_hand_mae": 0.032202571630477905, + "val_left_hand_r2": 0.9320798516273499, + "val_right_hand_mse": 0.0020236996933817863, + "val_right_hand_rmse": 0.0449855498286037, + "val_right_hand_mae": 0.03352319821715355, + "val_right_hand_r2": 0.9227489829063416, + "val_left_hip_mse": 0.0009181004716083407, + "val_left_hip_rmse": 0.030300172798324778, + "val_left_hip_mae": 0.022528253495693207, + "val_left_hip_r2": 0.8809365630149841, + "val_right_hip_mse": 0.0009066809434443712, + "val_right_hip_rmse": 0.030111143177308484, + "val_right_hip_mae": 0.022146377712488174, + "val_right_hip_r2": 0.8851180076599121, + "val_left_knee_mse": 0.0009057100978679955, + "val_left_knee_rmse": 0.030095017824683132, + "val_left_knee_mae": 0.022732043638825417, + "val_left_knee_r2": 0.904190182685852, + "val_right_knee_mse": 0.0012219977797940373, + "val_right_knee_rmse": 0.0349570848297457, + "val_right_knee_mae": 0.024425512179732323, + "val_right_knee_r2": 0.8932585716247559, + "val_left_foot_mse": 0.0007236601086333394, + "val_left_foot_rmse": 0.026900931371113147, + "val_left_foot_mae": 0.018858017399907112, + "val_left_foot_r2": 0.7857944369316101, + "val_right_foot_mse": 0.0013983125099912286, + "val_right_foot_rmse": 0.037394017034697254, + "val_right_foot_mae": 0.022459959611296654, + "val_right_foot_r2": 0.7059618234634399 + }, + "best_epoch": 90 + } + ], + "test_metrics": { + "test_mse": 0.000991992885246873, + "test_rmse": 0.031495918549025885, + "test_mae": 0.023255417123436928, + "test_r2": 0.9534837007522583, + "test_head_mse": 0.0010284307645633817, + "test_head_rmse": 0.032069155968989606, + "test_head_mae": 0.024457450956106186, + "test_head_r2": 0.8990598917007446, + "test_left_shoulder_mse": 0.0006820292910560966, + "test_left_shoulder_rmse": 0.026115690514633087, + "test_left_shoulder_mae": 0.019676726311445236, + "test_left_shoulder_r2": 0.880357027053833, + "test_left_elbow_mse": 0.0009444120223633945, + "test_left_elbow_rmse": 0.030731287352849287, + "test_left_elbow_mae": 0.02330195903778076, + "test_left_elbow_r2": 0.9302977323532104, + "test_right_shoulder_mse": 0.0006621659267693758, + "test_right_shoulder_rmse": 0.025732584922027863, + "test_right_shoulder_mae": 0.019634297117590904, + "test_right_shoulder_r2": 0.900537371635437, + "test_right_elbow_mse": 0.0010623643174767494, + "test_right_elbow_rmse": 0.03259393068466504, + "test_right_elbow_mae": 0.025116708129644394, + "test_right_elbow_r2": 0.9320510625839233, + "test_left_hand_mse": 0.001478076446801424, + "test_left_hand_rmse": 0.038445759802628744, + "test_left_hand_mae": 0.029303425922989845, + "test_left_hand_r2": 0.9435622692108154, + "test_right_hand_mse": 0.0018633577274158597, + "test_right_hand_rmse": 0.043166627473267584, + "test_right_hand_mae": 0.03278062492609024, + "test_right_hand_r2": 0.9337775111198425, + "test_left_hip_mse": 0.0007481245556846261, + "test_left_hip_rmse": 0.027351865671003617, + "test_left_hip_mae": 0.02081463672220707, + "test_left_hip_r2": 0.9151188135147095, + "test_right_hip_mse": 0.0007708412013016641, + "test_right_hip_rmse": 0.02776402710886272, + "test_right_hip_mae": 0.0206670593470335, + "test_right_hip_r2": 0.9158798456192017, + "test_left_knee_mse": 0.000813115038909018, + "test_left_knee_rmse": 0.028515172082753035, + "test_left_knee_mae": 0.021313384175300598, + "test_left_knee_r2": 0.9164251089096069, + "test_right_knee_mse": 0.0008224439807236195, + "test_right_knee_rmse": 0.02867828413144028, + "test_right_knee_mae": 0.02204538881778717, + "test_right_knee_r2": 0.9233040809631348, + "test_left_foot_mse": 0.0009163641952909529, + "test_left_foot_rmse": 0.03027150797847628, + "test_left_foot_mae": 0.0207055751234293, + "test_left_foot_r2": 0.8029574155807495, + "test_right_foot_mse": 0.0011041822144761682, + "test_right_foot_rmse": 0.033229237344184835, + "test_right_foot_mae": 0.022503167390823364, + "test_right_foot_r2": 0.7896671891212463 + } +} \ No newline at end of file diff --git a/A9/cv_results_optimizer/conv1d_v3_adam_fold1_best.h5 b/A9/cv_results_optimizer/conv1d_v3_adam_fold1_best.h5 new file mode 100644 index 0000000000000000000000000000000000000000..49df53924c8d52e779e09c6bf51c27ef65ca0598 --- /dev/null +++ b/A9/cv_results_optimizer/conv1d_v3_adam_fold1_best.h5 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:9316dfba38f1e7537d84a614955aa3d7a8e66b10ad0d6aebed34121b05a58748 +size 2658520 diff --git a/A9/cv_results_optimizer/conv1d_v3_adam_fold2_best.h5 b/A9/cv_results_optimizer/conv1d_v3_adam_fold2_best.h5 new file mode 100644 index 0000000000000000000000000000000000000000..6d6c5872a2484ee0b5074eac8ea5a28ac232fa62 --- /dev/null +++ b/A9/cv_results_optimizer/conv1d_v3_adam_fold2_best.h5 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:3d4fc0c302c529e745573b34f78349059eb8454cd381cf4c11558a15bec24ad7 +size 2658520 diff --git a/A9/cv_results_optimizer/conv1d_v3_adam_fold3_best.h5 b/A9/cv_results_optimizer/conv1d_v3_adam_fold3_best.h5 new file mode 100644 index 0000000000000000000000000000000000000000..78491db039202929205eff563dec582183242f2b --- /dev/null +++ b/A9/cv_results_optimizer/conv1d_v3_adam_fold3_best.h5 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:29fd5d5dd19d5fcee480ebd6090666f817b6169c6ac29c0b3bada7e579a201be +size 2658520 diff --git a/A9/cv_results_optimizer/conv1d_v3_adam_fold4_best.h5 b/A9/cv_results_optimizer/conv1d_v3_adam_fold4_best.h5 new file mode 100644 index 0000000000000000000000000000000000000000..56b540c835402021982ef3bd8ef89e83b289c053 --- /dev/null +++ b/A9/cv_results_optimizer/conv1d_v3_adam_fold4_best.h5 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e2bb24af5efc46db29967a8df96dc964f78339951ae8f2b7a2406752896d85ad +size 2658520 diff --git a/A9/cv_results_optimizer/conv1d_v3_adam_fold5_best.h5 b/A9/cv_results_optimizer/conv1d_v3_adam_fold5_best.h5 new file mode 100644 index 0000000000000000000000000000000000000000..2a1fd876c782a51bdc5bf1f172f15caf788b75b2 --- /dev/null +++ b/A9/cv_results_optimizer/conv1d_v3_adam_fold5_best.h5 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:bc6f37c15d9b0a64d10f138512e7e9c3b5e2b343baa61f987782d64e34a91182 +size 2658520 diff --git a/A9/cv_results_optimizer/conv1d_v3_adam_predictions.png b/A9/cv_results_optimizer/conv1d_v3_adam_predictions.png new file mode 100644 index 0000000000000000000000000000000000000000..640ffaf41ee334615a26fa9211472f6e9fa0b63b --- /dev/null +++ b/A9/cv_results_optimizer/conv1d_v3_adam_predictions.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:ac71c911b49648d3cff6c284ed4e20bfacfc3c8086ae244f920d0e76ace3cbf6 +size 555988 diff --git a/A9/cv_results_optimizer/conv1d_v3_adam_test_predictions.png b/A9/cv_results_optimizer/conv1d_v3_adam_test_predictions.png new file mode 100644 index 0000000000000000000000000000000000000000..f2b7f115662444f3d6da1f5ed0fb2456ffcc9bbb --- /dev/null +++ b/A9/cv_results_optimizer/conv1d_v3_adam_test_predictions.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:9ebd41d0e874f119d5e00c8398cd64b252b85c370dae6d21ebbc003edf05074e +size 460655 diff --git a/A9/cv_results_optimizer/conv1d_v3_adam_training_history.png b/A9/cv_results_optimizer/conv1d_v3_adam_training_history.png new file mode 100644 index 0000000000000000000000000000000000000000..cd0a9689c14de6796642bccdec3f3797d9d45fd8 --- /dev/null +++ b/A9/cv_results_optimizer/conv1d_v3_adam_training_history.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:f82b0a0dee403f21b5508770f61ec5aff5bf643a88921787c3a4508591433794 +size 377884 diff --git a/A9/cv_results_optimizer/conv1d_v3_rmsprop_cv_results.json b/A9/cv_results_optimizer/conv1d_v3_rmsprop_cv_results.json new file mode 100644 index 0000000000000000000000000000000000000000..9c22f2444cef5aa03adf781f0193148e66aecccb --- /dev/null +++ b/A9/cv_results_optimizer/conv1d_v3_rmsprop_cv_results.json @@ -0,0 +1,1243 @@ +{ + "model_name": "conv1d_v3_rmsprop", + "n_splits": 5, + "epochs": 100, + "batch_size": 32, + "patience": 10, + "aggregate_results": { + "n_folds": 5, + "train_metrics": { + "train_mse": { + "mean": 0.0008649110794067382, + "std": 8.824325398450086e-05 + }, + "train_rmse": { + "mean": 0.02937155767552967, + "std": 0.0014908652252207714 + }, + "train_mae": { + "mean": 0.021635980531573295, + "std": 0.0009593901497881564 + }, + "train_r2": { + "mean": 0.9590752124786377, + "std": 0.004254520833003684 + }, + "train_head_mse": { + "mean": 0.000905988912563771, + "std": 8.67112109487168e-05 + }, + "train_head_rmse": { + "mean": 0.030065423401202506, + "std": 0.0014350011394405765 + }, + "train_head_mae": { + "mean": 0.022776353731751443, + "std": 0.000954506294170352 + }, + "train_head_r2": { + "mean": 0.9088895201683045, + "std": 0.009123007983270249 + }, + "train_left_shoulder_mse": { + "mean": 0.0005727501935325563, + "std": 6.0391202041254114e-05 + }, + "train_left_shoulder_rmse": { + "mean": 0.023899560452425896, + "std": 0.0012494813777709486 + }, + "train_left_shoulder_mae": { + "mean": 0.018135455250740052, + "std": 0.0008055140195580671 + }, + "train_left_shoulder_r2": { + "mean": 0.8969090342521667, + "std": 0.011291304016830879 + }, + "train_left_elbow_mse": { + "mean": 0.0008840391063131392, + "std": 8.889444329821693e-05 + }, + "train_left_elbow_rmse": { + "mean": 0.02969601203742107, + "std": 0.0014785044424965015 + }, + "train_left_elbow_mae": { + "mean": 0.022739601135253907, + "std": 0.0011030161887621143 + }, + "train_left_elbow_r2": { + "mean": 0.9359802842140198, + "std": 0.006318980183906632 + }, + "train_right_shoulder_mse": { + "mean": 0.0005016362702008337, + "std": 3.348763229316499e-05 + }, + "train_right_shoulder_rmse": { + "mean": 0.022384663698499858, + "std": 0.0007504007635233563 + }, + "train_right_shoulder_mae": { + "mean": 0.017278416827321053, + "std": 0.0006035912929122334 + }, + "train_right_shoulder_r2": { + "mean": 0.9230553030967712, + "std": 0.0057982225387353415 + }, + "train_right_elbow_mse": { + "mean": 0.0008239429909735918, + "std": 8.431300539223756e-05 + }, + "train_right_elbow_rmse": { + "mean": 0.028666533767396436, + "std": 0.0014740531660397248 + }, + "train_right_elbow_mae": { + "mean": 0.02204204834997654, + "std": 0.0009580927651639716 + }, + "train_right_elbow_r2": { + "mean": 0.9477605819702148, + "std": 0.005369117053482486 + }, + "train_left_hand_mse": { + "mean": 0.0013151478953659535, + "std": 0.00012788711406386648 + }, + "train_left_hand_rmse": { + "mean": 0.03622214662597174, + "std": 0.0017618141708355393 + }, + "train_left_hand_mae": { + "mean": 0.02831711806356907, + "std": 0.001297288379064068 + }, + "train_left_hand_r2": { + "mean": 0.9499896407127381, + "std": 0.004628633155688908 + }, + "train_right_hand_mse": { + "mean": 0.0014226161874830723, + "std": 0.0001518865509477222 + }, + "train_right_hand_rmse": { + "mean": 0.03766445425095094, + "std": 0.0020012679633411843 + }, + "train_right_hand_mae": { + "mean": 0.028527287021279334, + "std": 0.0013297548879075845 + }, + "train_right_hand_r2": { + "mean": 0.9489306211471558, + "std": 0.005300094455826433 + }, + "train_left_hip_mse": { + "mean": 0.0006846916163340211, + "std": 5.7760508598921336e-05 + }, + "train_left_hip_rmse": { + "mean": 0.026143614023999878, + "std": 0.0010968418747193105 + }, + "train_left_hip_mae": { + "mean": 0.019952809438109398, + "std": 0.0007142865460425917 + }, + "train_left_hip_r2": { + "mean": 0.9120981097221375, + "std": 0.008191734112215592 + }, + "train_right_hip_mse": { + "mean": 0.0006752572720870375, + "std": 4.891809541199625e-05 + }, + "train_right_hip_rmse": { + "mean": 0.025968696910850853, + "std": 0.0009402408411693483 + }, + "train_right_hip_mae": { + "mean": 0.019704819098114966, + "std": 0.0006765069942508985 + }, + "train_right_hip_r2": { + "mean": 0.9166973948478698, + "std": 0.006931467008823075 + }, + "train_left_knee_mse": { + "mean": 0.0007321279379539191, + "std": 7.143224223446483e-05 + }, + "train_left_knee_rmse": { + "mean": 0.02702719781441152, + "std": 0.0012878339390575014 + }, + "train_left_knee_mae": { + "mean": 0.020038470253348352, + "std": 0.0008930885609104884 + }, + "train_left_knee_r2": { + "mean": 0.9227863669395446, + "std": 0.007818437994368436 + }, + "train_right_knee_mse": { + "mean": 0.0007379862130619585, + "std": 8.005808795698299e-05 + }, + "train_right_knee_rmse": { + "mean": 0.027126851918346906, + "std": 0.0014560625199616594 + }, + "train_right_knee_mae": { + "mean": 0.02005225084722042, + "std": 0.0007032744285370174 + }, + "train_right_knee_r2": { + "mean": 0.9313061833381653, + "std": 0.008184592321134344 + }, + "train_left_foot_mse": { + "mean": 0.0010293212486431002, + "std": 0.0001755111922928035 + }, + "train_left_foot_rmse": { + "mean": 0.03197051756040513, + "std": 0.0026846332268910637 + }, + "train_left_foot_mae": { + "mean": 0.0206429872661829, + "std": 0.001581341827529442 + }, + "train_left_foot_r2": { + "mean": 0.7652531266212463, + "std": 0.03448629493533994 + }, + "train_right_foot_mse": { + "mean": 0.0009583379724062979, + "std": 0.00012284044043664772 + }, + "train_right_foot_rmse": { + "mean": 0.03089462468182578, + "std": 0.0019647234348424652 + }, + "train_right_foot_mae": { + "mean": 0.021060135588049887, + "std": 0.0012638721279391234 + }, + "train_right_foot_r2": { + "mean": 0.7860891103744507, + "std": 0.028887518988564967 + } + }, + "val_metrics": { + "val_mse": { + "mean": 0.0011252352967858314, + "std": 0.0001163449479631884 + }, + "val_rmse": { + "mean": 0.033497893261198924, + "std": 0.0017681752874518421 + }, + "val_mae": { + "mean": 0.02413139268755913, + "std": 0.0012988234217981521 + }, + "val_r2": { + "mean": 0.9468027114868164, + "std": 0.0048638207842168334 + }, + "val_head_mse": { + "mean": 0.0010982222156599164, + "std": 0.00012869060131023114 + }, + "val_head_rmse": { + "mean": 0.033082643215868325, + "std": 0.001939312638920776 + }, + "val_head_mae": { + "mean": 0.025050128623843194, + "std": 0.0013610592256827515 + }, + "val_head_r2": { + "mean": 0.8897714734077453, + "std": 0.008997149047614837 + }, + "val_left_shoulder_mse": { + "mean": 0.0007041914272122085, + "std": 0.0001079559358340065 + }, + "val_left_shoulder_rmse": { + "mean": 0.02645963141892895, + "std": 0.002019735721978598 + }, + "val_left_shoulder_mae": { + "mean": 0.01988995634019375, + "std": 0.0011601629917151952 + }, + "val_left_shoulder_r2": { + "mean": 0.8737460136413574, + "std": 0.013880020455915772 + }, + "val_left_elbow_mse": { + "mean": 0.00115490083117038, + "std": 0.00013989328364650733 + }, + "val_left_elbow_rmse": { + "mean": 0.03391585123206231, + "std": 0.0021484567426405723 + }, + "val_left_elbow_mae": { + "mean": 0.025739220529794694, + "std": 0.001324849102822408 + }, + "val_left_elbow_r2": { + "mean": 0.9162502884864807, + "std": 0.009959369573324637 + }, + "val_right_shoulder_mse": { + "mean": 0.0006381653365679085, + "std": 7.499665604882143e-05 + }, + "val_right_shoulder_rmse": { + "mean": 0.02521756436707846, + "std": 0.0014965907791254578 + }, + "val_right_shoulder_mae": { + "mean": 0.01906120702624321, + "std": 0.0010421736465311626 + }, + "val_right_shoulder_r2": { + "mean": 0.9022559523582458, + "std": 0.008957135222230682 + }, + "val_right_elbow_mse": { + "mean": 0.0011429140926338733, + "std": 0.0001245924586483504 + }, + "val_right_elbow_rmse": { + "mean": 0.03375569146005759, + "std": 0.0018620920190121593 + }, + "val_right_elbow_mae": { + "mean": 0.02527901194989681, + "std": 0.0016973496368740762 + }, + "val_right_elbow_r2": { + "mean": 0.9273767590522766, + "std": 0.008384430263266369 + }, + "val_left_hand_mse": { + "mean": 0.0018367601558566093, + "std": 0.00024268382211872953 + }, + "val_left_hand_rmse": { + "mean": 0.04275824189563361, + "std": 0.0029142590569632316 + }, + "val_left_hand_mae": { + "mean": 0.03257917836308479, + "std": 0.0016980199276724845 + }, + "val_left_hand_r2": { + "mean": 0.9297553062438965, + "std": 0.010842212668292906 + }, + "val_right_hand_mse": { + "mean": 0.0020049342419952154, + "std": 0.00030760377316783606 + }, + "val_right_hand_rmse": { + "mean": 0.044642920173230106, + "std": 0.003455997743318155 + }, + "val_right_hand_mae": { + "mean": 0.03304735645651817, + "std": 0.0024717061421582133 + }, + "val_right_hand_r2": { + "mean": 0.9277131915092468, + "std": 0.012355909149715696 + }, + "val_left_hip_mse": { + "mean": 0.0008464130689390004, + "std": 0.00013080907753372108 + }, + "val_left_hip_rmse": { + "mean": 0.02900451414665914, + "std": 0.0022696317003536625 + }, + "val_left_hip_mae": { + "mean": 0.022026238217949867, + "std": 0.0015061825472129401 + }, + "val_left_hip_r2": { + "mean": 0.8914692997932434, + "std": 0.014882420831830414 + }, + "val_right_hip_mse": { + "mean": 0.0008410505135543645, + "std": 0.00011907282423314304 + }, + "val_right_hip_rmse": { + "mean": 0.02892669104820563, + "std": 0.0020729347688783297 + }, + "val_right_hip_mae": { + "mean": 0.021862692758440973, + "std": 0.0013751676574380056 + }, + "val_right_hip_r2": { + "mean": 0.8962132334709167, + "std": 0.013786001926846862 + }, + "val_left_knee_mse": { + "mean": 0.0009022137150168419, + "std": 0.00010826735122685515 + }, + "val_left_knee_rmse": { + "mean": 0.029982204679133734, + "std": 0.0018113855457558285 + }, + "val_left_knee_mae": { + "mean": 0.021775781735777856, + "std": 0.0015132187234003585 + }, + "val_left_knee_r2": { + "mean": 0.9049453258514404, + "std": 0.00909221463384313 + }, + "val_right_knee_mse": { + "mean": 0.0009799935156479477, + "std": 0.00019997158113997277 + }, + "val_right_knee_rmse": { + "mean": 0.031151617585714008, + "std": 0.0030935800363626038 + }, + "val_right_knee_mae": { + "mean": 0.022425953671336173, + "std": 0.0017554894001924712 + }, + "val_right_knee_r2": { + "mean": 0.9093814015388488, + "std": 0.014644321380921387 + }, + "val_left_foot_mse": { + "mean": 0.001227499800734222, + "std": 0.000253064164316046 + }, + "val_left_foot_rmse": { + "mean": 0.03485745519541672, + "std": 0.003529535101644114 + }, + "val_left_foot_mae": { + "mean": 0.021984563395380973, + "std": 0.0011249945043919028 + }, + "val_left_foot_r2": { + "mean": 0.7198350548744201, + "std": 0.03741644167630292 + }, + "val_right_foot_mse": { + "mean": 0.001250799559056759, + "std": 0.0001889850592454509 + }, + "val_right_foot_rmse": { + "mean": 0.03527220005408782, + "std": 0.002582917420508358 + }, + "val_right_foot_mae": { + "mean": 0.022986817732453348, + "std": 0.0008602749858762264 + }, + "val_right_foot_r2": { + "mean": 0.720678174495697, + "std": 0.038396833505447946 + } + }, + "best_fold": "1", + "best_fold_metrics": { + "fold": 1, + "train_metrics": { + "train_mse": 0.0007644317811354995, + "train_rmse": 0.027648359465536097, + "train_mae": 0.02060619555413723, + "train_r2": 0.9640895128250122, + "train_head_mse": 0.0007996394415386021, + "train_head_rmse": 0.028277896695804695, + "train_head_mae": 0.02181337960064411, + "train_head_r2": 0.9208014011383057, + "train_left_shoulder_mse": 0.0005024113925173879, + "train_left_shoulder_rmse": 0.022414535295593077, + "train_left_shoulder_mae": 0.017242593690752983, + "train_left_shoulder_r2": 0.911067545413971, + "train_left_elbow_mse": 0.0007704654708504677, + "train_left_elbow_rmse": 0.027757259786413854, + "train_left_elbow_mae": 0.02142667956650257, + "train_left_elbow_r2": 0.9441766142845154, + "train_right_shoulder_mse": 0.0004583777626976371, + "train_right_shoulder_rmse": 0.02140975858569258, + "train_right_shoulder_mae": 0.016558000817894936, + "train_right_shoulder_r2": 0.9306325316429138, + "train_right_elbow_mse": 0.0007271860959008336, + "train_right_elbow_rmse": 0.026966388262072354, + "train_right_elbow_mae": 0.021033238619565964, + "train_right_elbow_r2": 0.9543192386627197, + "train_left_hand_mse": 0.0011515294900164008, + "train_left_hand_rmse": 0.033934193522410415, + "train_left_hand_mae": 0.026606885716319084, + "train_left_hand_r2": 0.9558072686195374, + "train_right_hand_mse": 0.001269071246497333, + "train_right_hand_rmse": 0.03562402625332141, + "train_right_hand_mae": 0.02720487117767334, + "train_right_hand_r2": 0.954587996006012, + "train_left_hip_mse": 0.0006197979091666639, + "train_left_hip_rmse": 0.0248957407836494, + "train_left_hip_mae": 0.019247792661190033, + "train_left_hip_r2": 0.9224674701690674, + "train_right_hip_mse": 0.000614702352322638, + "train_right_hip_rmse": 0.024793191652601687, + "train_right_hip_mae": 0.018980462104082108, + "train_right_hip_r2": 0.9259294271469116, + "train_left_knee_mse": 0.000668326742015779, + "train_left_knee_rmse": 0.02585201620794361, + "train_left_knee_mae": 0.01923758164048195, + "train_left_knee_r2": 0.9300936460494995, + "train_right_knee_mse": 0.0006693300092592835, + "train_right_knee_rmse": 0.025871412973768625, + "train_right_knee_mae": 0.01943545788526535, + "train_right_knee_r2": 0.9377626776695251, + "train_left_foot_mse": 0.0008432724280282855, + "train_left_foot_rmse": 0.029039153362801152, + "train_left_foot_mae": 0.019444221630692482, + "train_left_foot_r2": 0.8085814118385315, + "train_right_foot_mse": 0.0008435030467808247, + "train_right_foot_rmse": 0.029043123915667622, + "train_right_foot_mae": 0.019649378955364227, + "train_right_foot_r2": 0.8099750876426697 + }, + "val_metrics": { + "val_mse": 0.0009299911325797439, + "val_rmse": 0.030495755976524732, + "val_mae": 0.022221975028514862, + "val_r2": 0.9547322988510132, + "val_head_mse": 0.0009193362202495337, + "val_head_rmse": 0.030320557716663685, + "val_head_mae": 0.02286026068031788, + "val_head_r2": 0.9014921188354492, + "val_left_shoulder_mse": 0.0005636455025523901, + "val_left_shoulder_rmse": 0.023741219483261387, + "val_left_shoulder_mae": 0.018062619492411613, + "val_left_shoulder_r2": 0.891297459602356, + "val_left_elbow_mse": 0.0008848309516906738, + "val_left_elbow_rmse": 0.029746108177216627, + "val_left_elbow_mae": 0.02332305908203125, + "val_left_elbow_r2": 0.935871422290802, + "val_right_shoulder_mse": 0.00054282380733639, + "val_right_shoulder_rmse": 0.02329857951327484, + "val_right_shoulder_mae": 0.017673306167125702, + "val_right_shoulder_r2": 0.9121713042259216, + "val_right_elbow_mse": 0.0009586682426743209, + "val_right_elbow_rmse": 0.03096236816967205, + "val_right_elbow_mae": 0.02341270074248314, + "val_right_elbow_r2": 0.936889111995697, + "val_left_hand_mse": 0.0014097543898969889, + "val_left_hand_rmse": 0.03754669612491875, + "val_left_hand_mae": 0.02972496673464775, + "val_left_hand_r2": 0.9480786323547363, + "val_right_hand_mse": 0.0015748271252959967, + "val_right_hand_rmse": 0.03968409158965336, + "val_right_hand_mae": 0.030409449711441994, + "val_right_hand_r2": 0.9426539540290833, + "val_left_hip_mse": 0.0006610432174056768, + "val_left_hip_rmse": 0.025710760731757373, + "val_left_hip_mae": 0.019595736637711525, + "val_left_hip_r2": 0.9057207703590393, + "val_right_hip_mse": 0.0006655704346485436, + "val_right_hip_rmse": 0.025798651799048406, + "val_right_hip_mae": 0.019682275131344795, + "val_right_hip_r2": 0.9098149538040161, + "val_left_knee_mse": 0.0007578296936117113, + "val_left_knee_rmse": 0.027528706718836452, + "val_left_knee_mae": 0.019695069640874863, + "val_left_knee_r2": 0.9174781441688538, + "val_right_knee_mse": 0.0008769836276769638, + "val_right_knee_rmse": 0.029613909361598374, + "val_right_knee_mae": 0.021676691249012947, + "val_right_knee_r2": 0.9184427261352539, + "val_left_foot_mse": 0.0011972666252404451, + "val_left_foot_rmse": 0.034601540792867085, + "val_left_foot_mae": 0.02116885408759117, + "val_left_foot_r2": 0.7180234789848328, + "val_right_foot_mse": 0.0010773048270493746, + "val_right_foot_rmse": 0.03282232208496794, + "val_right_foot_mae": 0.02160065807402134, + "val_right_foot_r2": 0.7691446542739868 + }, + "best_epoch": 100 + } + }, + "fold_details": [ + { + "fold": 1, + "train_metrics": { + "train_mse": 0.0007644317811354995, + "train_rmse": 0.027648359465536097, + "train_mae": 0.02060619555413723, + "train_r2": 0.9640895128250122, + "train_head_mse": 0.0007996394415386021, + "train_head_rmse": 0.028277896695804695, + "train_head_mae": 0.02181337960064411, + "train_head_r2": 0.9208014011383057, + "train_left_shoulder_mse": 0.0005024113925173879, + "train_left_shoulder_rmse": 0.022414535295593077, + "train_left_shoulder_mae": 0.017242593690752983, + "train_left_shoulder_r2": 0.911067545413971, + "train_left_elbow_mse": 0.0007704654708504677, + "train_left_elbow_rmse": 0.027757259786413854, + "train_left_elbow_mae": 0.02142667956650257, + "train_left_elbow_r2": 0.9441766142845154, + "train_right_shoulder_mse": 0.0004583777626976371, + "train_right_shoulder_rmse": 0.02140975858569258, + "train_right_shoulder_mae": 0.016558000817894936, + "train_right_shoulder_r2": 0.9306325316429138, + "train_right_elbow_mse": 0.0007271860959008336, + "train_right_elbow_rmse": 0.026966388262072354, + "train_right_elbow_mae": 0.021033238619565964, + "train_right_elbow_r2": 0.9543192386627197, + "train_left_hand_mse": 0.0011515294900164008, + "train_left_hand_rmse": 0.033934193522410415, + "train_left_hand_mae": 0.026606885716319084, + "train_left_hand_r2": 0.9558072686195374, + "train_right_hand_mse": 0.001269071246497333, + "train_right_hand_rmse": 0.03562402625332141, + "train_right_hand_mae": 0.02720487117767334, + "train_right_hand_r2": 0.954587996006012, + "train_left_hip_mse": 0.0006197979091666639, + "train_left_hip_rmse": 0.0248957407836494, + "train_left_hip_mae": 0.019247792661190033, + "train_left_hip_r2": 0.9224674701690674, + "train_right_hip_mse": 0.000614702352322638, + "train_right_hip_rmse": 0.024793191652601687, + "train_right_hip_mae": 0.018980462104082108, + "train_right_hip_r2": 0.9259294271469116, + "train_left_knee_mse": 0.000668326742015779, + "train_left_knee_rmse": 0.02585201620794361, + "train_left_knee_mae": 0.01923758164048195, + "train_left_knee_r2": 0.9300936460494995, + "train_right_knee_mse": 0.0006693300092592835, + "train_right_knee_rmse": 0.025871412973768625, + "train_right_knee_mae": 0.01943545788526535, + "train_right_knee_r2": 0.9377626776695251, + "train_left_foot_mse": 0.0008432724280282855, + "train_left_foot_rmse": 0.029039153362801152, + "train_left_foot_mae": 0.019444221630692482, + "train_left_foot_r2": 0.8085814118385315, + "train_right_foot_mse": 0.0008435030467808247, + "train_right_foot_rmse": 0.029043123915667622, + "train_right_foot_mae": 0.019649378955364227, + "train_right_foot_r2": 0.8099750876426697 + }, + "val_metrics": { + "val_mse": 0.0009299911325797439, + "val_rmse": 0.030495755976524732, + "val_mae": 0.022221975028514862, + "val_r2": 0.9547322988510132, + "val_head_mse": 0.0009193362202495337, + "val_head_rmse": 0.030320557716663685, + "val_head_mae": 0.02286026068031788, + "val_head_r2": 0.9014921188354492, + "val_left_shoulder_mse": 0.0005636455025523901, + "val_left_shoulder_rmse": 0.023741219483261387, + "val_left_shoulder_mae": 0.018062619492411613, + "val_left_shoulder_r2": 0.891297459602356, + "val_left_elbow_mse": 0.0008848309516906738, + "val_left_elbow_rmse": 0.029746108177216627, + "val_left_elbow_mae": 0.02332305908203125, + "val_left_elbow_r2": 0.935871422290802, + "val_right_shoulder_mse": 0.00054282380733639, + "val_right_shoulder_rmse": 0.02329857951327484, + "val_right_shoulder_mae": 0.017673306167125702, + "val_right_shoulder_r2": 0.9121713042259216, + "val_right_elbow_mse": 0.0009586682426743209, + "val_right_elbow_rmse": 0.03096236816967205, + "val_right_elbow_mae": 0.02341270074248314, + "val_right_elbow_r2": 0.936889111995697, + "val_left_hand_mse": 0.0014097543898969889, + "val_left_hand_rmse": 0.03754669612491875, + "val_left_hand_mae": 0.02972496673464775, + "val_left_hand_r2": 0.9480786323547363, + "val_right_hand_mse": 0.0015748271252959967, + "val_right_hand_rmse": 0.03968409158965336, + "val_right_hand_mae": 0.030409449711441994, + "val_right_hand_r2": 0.9426539540290833, + "val_left_hip_mse": 0.0006610432174056768, + "val_left_hip_rmse": 0.025710760731757373, + "val_left_hip_mae": 0.019595736637711525, + "val_left_hip_r2": 0.9057207703590393, + "val_right_hip_mse": 0.0006655704346485436, + "val_right_hip_rmse": 0.025798651799048406, + "val_right_hip_mae": 0.019682275131344795, + "val_right_hip_r2": 0.9098149538040161, + "val_left_knee_mse": 0.0007578296936117113, + "val_left_knee_rmse": 0.027528706718836452, + "val_left_knee_mae": 0.019695069640874863, + "val_left_knee_r2": 0.9174781441688538, + "val_right_knee_mse": 0.0008769836276769638, + "val_right_knee_rmse": 0.029613909361598374, + "val_right_knee_mae": 0.021676691249012947, + "val_right_knee_r2": 0.9184427261352539, + "val_left_foot_mse": 0.0011972666252404451, + "val_left_foot_rmse": 0.034601540792867085, + "val_left_foot_mae": 0.02116885408759117, + "val_left_foot_r2": 0.7180234789848328, + "val_right_foot_mse": 0.0010773048270493746, + "val_right_foot_rmse": 0.03282232208496794, + "val_right_foot_mae": 0.02160065807402134, + "val_right_foot_r2": 0.7691446542739868 + }, + "best_epoch": 100 + }, + { + "fold": 2, + "train_metrics": { + "train_mse": 0.0009167075622826815, + "train_rmse": 0.030277178902313233, + "train_mae": 0.022160407155752182, + "train_r2": 0.9562936425209045, + "train_head_mse": 0.0010136786149814725, + "train_head_rmse": 0.03183831991455379, + "train_head_mae": 0.023717226460576057, + "train_head_r2": 0.8963203430175781, + "train_left_shoulder_mse": 0.0005893956986255944, + "train_left_shoulder_rmse": 0.02427747306919718, + "train_left_shoulder_mae": 0.01835184544324875, + "train_left_shoulder_r2": 0.8934473991394043, + "train_left_elbow_mse": 0.0009067013743333519, + "train_left_elbow_rmse": 0.03011148243334014, + "train_left_elbow_mae": 0.02282308042049408, + "train_left_elbow_r2": 0.9347777366638184, + "train_right_shoulder_mse": 0.0005447663133963943, + "train_right_shoulder_rmse": 0.023340229506078003, + "train_right_shoulder_mae": 0.01795187219977379, + "train_right_shoulder_r2": 0.9151157140731812, + "train_right_elbow_mse": 0.0009078680886887014, + "train_right_elbow_rmse": 0.03013084945182763, + "train_right_elbow_mae": 0.02258862927556038, + "train_right_elbow_r2": 0.941989004611969, + "train_left_hand_mse": 0.0014476473443210125, + "train_left_hand_rmse": 0.03804796110596483, + "train_left_hand_mae": 0.02931777387857437, + "train_left_hand_r2": 0.9451899528503418, + "train_right_hand_mse": 0.0015485375188291073, + "train_right_hand_rmse": 0.03935146145734752, + "train_right_hand_mae": 0.029341556131839752, + "train_right_hand_r2": 0.9441172480583191, + "train_left_hip_mse": 0.0007001588237471879, + "train_left_hip_rmse": 0.026460514427107947, + "train_left_hip_mae": 0.02026459574699402, + "train_left_hip_r2": 0.9073135256767273, + "train_right_hip_mse": 0.0007026353850960732, + "train_right_hip_rmse": 0.026507270419567405, + "train_right_hip_mae": 0.02013166807591915, + "train_right_hip_r2": 0.9100901484489441, + "train_left_knee_mse": 0.0007265947642736137, + "train_left_knee_rmse": 0.026955421797360428, + "train_left_knee_mae": 0.020192233845591545, + "train_left_knee_r2": 0.921817421913147, + "train_right_knee_mse": 0.0007551433518528938, + "train_right_knee_rmse": 0.027479871758305092, + "train_right_knee_mae": 0.02037254348397255, + "train_right_knee_r2": 0.9298735857009888, + "train_left_foot_mse": 0.0010676839156076312, + "train_left_foot_rmse": 0.03267543290620082, + "train_left_foot_mae": 0.02111709862947464, + "train_left_foot_r2": 0.7527708411216736, + "train_right_foot_mse": 0.0010063867084681988, + "train_right_foot_rmse": 0.031723598605268584, + "train_right_foot_mae": 0.021915161982178688, + "train_right_foot_r2": 0.7796924114227295 + }, + "val_metrics": { + "val_mse": 0.0011744454968720675, + "val_rmse": 0.03427018378812795, + "val_mae": 0.025049660354852676, + "val_r2": 0.9460820555686951, + "val_head_mse": 0.0012985533103346825, + "val_head_rmse": 0.03603544519406805, + "val_head_mae": 0.02700039930641651, + "val_head_r2": 0.8777099847793579, + "val_left_shoulder_mse": 0.0006939647137187421, + "val_left_shoulder_rmse": 0.026343210011666045, + "val_left_shoulder_mae": 0.02010258287191391, + "val_left_shoulder_r2": 0.8772221803665161, + "val_left_elbow_mse": 0.0012121484614908695, + "val_left_elbow_rmse": 0.034815922528217885, + "val_left_elbow_mae": 0.02620295248925686, + "val_left_elbow_r2": 0.909471333026886, + "val_right_shoulder_mse": 0.0007308349595405161, + "val_right_shoulder_rmse": 0.027033959375950024, + "val_right_shoulder_mae": 0.02021113969385624, + "val_right_shoulder_r2": 0.8945615887641907, + "val_right_elbow_mse": 0.0012704158434644341, + "val_right_elbow_rmse": 0.03564289330938826, + "val_right_elbow_mae": 0.026572609320282936, + "val_right_elbow_r2": 0.9217061996459961, + "val_left_hand_mse": 0.002135353861376643, + "val_left_hand_rmse": 0.04620988921623426, + "val_left_hand_mae": 0.03460663929581642, + "val_left_hand_r2": 0.916954517364502, + "val_right_hand_mse": 0.002248746110126376, + "val_right_hand_rmse": 0.04742094590079764, + "val_right_hand_mae": 0.03475645184516907, + "val_right_hand_r2": 0.9206403493881226, + "val_left_hip_mse": 0.0008461969555355608, + "val_left_hip_rmse": 0.029089464682863466, + "val_left_hip_mae": 0.022607941180467606, + "val_left_hip_r2": 0.9033893346786499, + "val_right_hip_mse": 0.0008776735048741102, + "val_right_hip_rmse": 0.029625554929386726, + "val_right_hip_mae": 0.022808438166975975, + "val_right_hip_r2": 0.905659019947052, + "val_left_knee_mse": 0.000923420418985188, + "val_left_knee_rmse": 0.03038783340393303, + "val_left_knee_mae": 0.022580325603485107, + "val_left_knee_r2": 0.9097039699554443, + "val_right_knee_mse": 0.0008170143701136112, + "val_right_knee_rmse": 0.02858346322812565, + "val_right_knee_mae": 0.022046072408556938, + "val_right_knee_r2": 0.9236546158790588, + "val_left_foot_mse": 0.0010822099866345525, + "val_left_foot_rmse": 0.03289696014276323, + "val_left_foot_mae": 0.02283785492181778, + "val_left_foot_r2": 0.7636855840682983, + "val_right_foot_mse": 0.001131259254179895, + "val_right_foot_rmse": 0.03363419768895781, + "val_right_foot_mae": 0.02331218309700489, + "val_right_foot_r2": 0.7270930409431458 + }, + "best_epoch": 100 + }, + { + "fold": 3, + "train_metrics": { + "train_mse": 0.0008597103878855705, + "train_rmse": 0.02932081833587819, + "train_mae": 0.021585028618574142, + "train_r2": 0.9588760733604431, + "train_head_mse": 0.0008480582037009299, + "train_head_rmse": 0.02912143890162246, + "train_head_mae": 0.022065419703722, + "train_head_r2": 0.9138222932815552, + "train_left_shoulder_mse": 0.000578926526941359, + "train_left_shoulder_rmse": 0.024060892064538236, + "train_left_shoulder_mae": 0.01811482571065426, + "train_left_shoulder_r2": 0.8937563896179199, + "train_left_elbow_mse": 0.0008744848892092705, + "train_left_elbow_rmse": 0.029571690672149105, + "train_left_elbow_mae": 0.02253888174891472, + "train_left_elbow_r2": 0.9360511898994446, + "train_right_shoulder_mse": 0.0005174375837668777, + "train_right_shoulder_rmse": 0.022747254422608405, + "train_right_shoulder_mae": 0.01774049550294876, + "train_right_shoulder_r2": 0.9196048974990845, + "train_right_elbow_mse": 0.0008336987812072039, + "train_right_elbow_rmse": 0.028873842508526707, + "train_right_elbow_mae": 0.022214313969016075, + "train_right_elbow_r2": 0.9463329911231995, + "train_left_hand_mse": 0.0012868545018136501, + "train_left_hand_rmse": 0.03587275431039064, + "train_left_hand_mae": 0.028118595480918884, + "train_left_hand_r2": 0.950557291507721, + "train_right_hand_mse": 0.0013860348844900727, + "train_right_hand_rmse": 0.03722948944707774, + "train_right_hand_mae": 0.028437957167625427, + "train_right_hand_r2": 0.9493642449378967, + "train_left_hip_mse": 0.0006978180608712137, + "train_left_hip_rmse": 0.026416246154047203, + "train_left_hip_mae": 0.019941169768571854, + "train_left_hip_r2": 0.9100660085678101, + "train_right_hip_mse": 0.0006834437954239547, + "train_right_hip_rmse": 0.026142757991917277, + "train_right_hip_mae": 0.019702685996890068, + "train_right_hip_r2": 0.915703535079956, + "train_left_knee_mse": 0.0007206366863101721, + "train_left_knee_rmse": 0.02684467705728963, + "train_left_knee_mae": 0.01973244734108448, + "train_left_knee_r2": 0.9234588146209717, + "train_right_knee_mse": 0.0007453804137185216, + "train_right_knee_rmse": 0.027301655878692075, + "train_right_knee_mae": 0.020025121048092842, + "train_right_knee_r2": 0.9303073883056641, + "train_left_foot_mse": 0.00103966414462775, + "train_left_foot_rmse": 0.032243823356229794, + "train_left_foot_mae": 0.020844075828790665, + "train_left_foot_r2": 0.7560219168663025, + "train_right_foot_mse": 0.000963796628639102, + "train_right_foot_rmse": 0.031045074144525762, + "train_right_foot_mae": 0.021129414439201355, + "train_right_foot_r2": 0.7831913232803345 + }, + "val_metrics": { + "val_mse": 0.0012135021388530731, + "val_rmse": 0.03483535759617049, + "val_mae": 0.024911390617489815, + "val_r2": 0.9450418949127197, + "val_head_mse": 0.001151363947428763, + "val_head_rmse": 0.033931754263945194, + "val_head_mae": 0.02535792626440525, + "val_head_r2": 0.8891116380691528, + "val_left_shoulder_mse": 0.000877636659424752, + "val_left_shoulder_rmse": 0.029624933070384343, + "val_left_shoulder_mae": 0.021203158423304558, + "val_left_shoulder_r2": 0.8537274599075317, + "val_left_elbow_mse": 0.0012907282216474414, + "val_left_elbow_rmse": 0.03592670624545815, + "val_left_elbow_mae": 0.026820573955774307, + "val_left_elbow_r2": 0.9099006056785583, + "val_right_shoulder_mse": 0.0006933578988537192, + "val_right_shoulder_rmse": 0.02633169001134791, + "val_right_shoulder_mae": 0.0199758168309927, + "val_right_shoulder_r2": 0.899167537689209, + "val_right_elbow_mse": 0.0011662483448162675, + "val_right_elbow_rmse": 0.03415037839931305, + "val_right_elbow_mae": 0.026630831882357597, + "val_right_elbow_r2": 0.9303141832351685, + "val_left_hand_mse": 0.0017727238591760397, + "val_left_hand_rmse": 0.042103727378654254, + "val_left_hand_mae": 0.03284713625907898, + "val_left_hand_r2": 0.935061514377594, + "val_right_hand_mse": 0.002001088345423341, + "val_right_hand_rmse": 0.04473352596681085, + "val_right_hand_mae": 0.03385771065950394, + "val_right_hand_r2": 0.9327567219734192, + "val_left_hip_mse": 0.0009720409871079028, + "val_left_hip_rmse": 0.031177571860359857, + "val_left_hip_mae": 0.022966664284467697, + "val_left_hip_r2": 0.8776881098747253, + "val_right_hip_mse": 0.0008983573061414063, + "val_right_hip_rmse": 0.029972609264817207, + "val_right_hip_mae": 0.022207435220479965, + "val_right_hip_r2": 0.8896557092666626, + "val_left_knee_mse": 0.0010477312607690692, + "val_left_knee_rmse": 0.03236867715506874, + "val_left_knee_mae": 0.02227994240820408, + "val_left_knee_r2": 0.892817497253418, + "val_right_knee_mse": 0.001075215172022581, + "val_right_knee_rmse": 0.03279047379990995, + "val_right_knee_mae": 0.022538840770721436, + "val_right_knee_r2": 0.902241051197052, + "val_left_foot_mse": 0.00167395721655339, + "val_left_foot_rmse": 0.040914022248532224, + "val_left_foot_mae": 0.02378350868821144, + "val_left_foot_r2": 0.6513503193855286, + "val_right_foot_mse": 0.0011550786439329386, + "val_right_foot_rmse": 0.03398644794521691, + "val_right_foot_mae": 0.023378543555736542, + "val_right_foot_r2": 0.7508265972137451 + }, + "best_epoch": 86 + }, + { + "fold": 4, + "train_metrics": { + "train_mse": 0.0007807748625054955, + "train_rmse": 0.0279423489081626, + "train_mae": 0.020667273551225662, + "train_r2": 0.9633270502090454, + "train_head_mse": 0.0008645357447676361, + "train_head_rmse": 0.02940298870468164, + "train_head_mae": 0.022150713950395584, + "train_head_r2": 0.9131859540939331, + "train_left_shoulder_mse": 0.0005195548292249441, + "train_left_shoulder_rmse": 0.02279374539703697, + "train_left_shoulder_mae": 0.017445536330342293, + "train_left_shoulder_r2": 0.9070405960083008, + "train_left_elbow_mse": 0.0008318985346704721, + "train_left_elbow_rmse": 0.028842651311390778, + "train_left_elbow_mae": 0.022172199562191963, + "train_left_elbow_r2": 0.939730167388916, + "train_right_shoulder_mse": 0.00046638536150567234, + "train_right_shoulder_rmse": 0.02159595706389676, + "train_right_shoulder_mae": 0.016546742990612984, + "train_right_shoulder_r2": 0.9287347197532654, + "train_right_elbow_mse": 0.0007279151468537748, + "train_right_elbow_rmse": 0.026979902647225672, + "train_right_elbow_mae": 0.02092181332409382, + "train_right_elbow_r2": 0.9538065195083618, + "train_left_hand_mse": 0.0012127876980230212, + "train_left_hand_rmse": 0.03482510155079266, + "train_left_hand_mae": 0.02735045552253723, + "train_left_hand_r2": 0.9541293978691101, + "train_right_hand_mse": 0.0012643032241612673, + "train_right_hand_rmse": 0.0355570418364811, + "train_right_hand_mae": 0.02705925516784191, + "train_right_hand_r2": 0.9547832012176514, + "train_left_hip_mse": 0.0006273439503274858, + "train_left_hip_rmse": 0.025046835135950525, + "train_left_hip_mae": 0.01918708346784115, + "train_left_hip_r2": 0.9201875329017639, + "train_right_hip_mse": 0.0006278810324147344, + "train_right_hip_rmse": 0.025057554398119828, + "train_right_hip_mae": 0.018978135660290718, + "train_right_hip_r2": 0.9232650995254517, + "train_left_knee_mse": 0.0006776376976631582, + "train_left_knee_rmse": 0.026031475134213163, + "train_left_knee_mae": 0.019337106496095657, + "train_left_knee_r2": 0.9299176335334778, + "train_right_knee_mse": 0.0006459966534748673, + "train_right_knee_rmse": 0.02541646422055726, + "train_right_knee_mae": 0.019229980185627937, + "train_right_knee_r2": 0.9411700963973999, + "train_left_foot_mse": 0.0008651877287775278, + "train_left_foot_rmse": 0.02941407365152824, + "train_left_foot_mae": 0.01859566941857338, + "train_left_foot_r2": 0.7968400716781616, + "train_right_foot_mse": 0.000818645057734102, + "train_right_foot_rmse": 0.02861197402721633, + "train_right_foot_mae": 0.019699858501553535, + "train_right_foot_r2": 0.820303201675415 + }, + "val_metrics": { + "val_mse": 0.0010600995738059282, + "val_rmse": 0.03255917034885761, + "val_mae": 0.022949542850255966, + "val_r2": 0.9483765959739685, + "val_head_mse": 0.001010590116493404, + "val_head_rmse": 0.03178978006362114, + "val_head_mae": 0.024483924731612206, + "val_head_r2": 0.8980485200881958, + "val_left_shoulder_mse": 0.0006293161422945559, + "val_left_shoulder_rmse": 0.025086174325603255, + "val_left_shoulder_mae": 0.019153138622641563, + "val_left_shoulder_r2": 0.8842388987541199, + "val_left_elbow_mse": 0.001182852196507156, + "val_left_elbow_rmse": 0.034392618343289244, + "val_left_elbow_mae": 0.0254084262996912, + "val_left_elbow_r2": 0.9143470525741577, + "val_right_shoulder_mse": 0.0005570442881435156, + "val_right_shoulder_rmse": 0.023601785698194863, + "val_right_shoulder_mae": 0.0179756972938776, + "val_right_shoulder_r2": 0.9134998321533203, + "val_right_elbow_mse": 0.0010449225082993507, + "val_right_elbow_rmse": 0.03232526114820035, + "val_right_elbow_mae": 0.02300298772752285, + "val_right_elbow_r2": 0.9339180588722229, + "val_left_hand_mse": 0.0019194523338228464, + "val_left_hand_rmse": 0.04381155479805352, + "val_left_hand_mae": 0.03187589719891548, + "val_left_hand_r2": 0.9251706004142761, + "val_right_hand_mse": 0.0017755264416337013, + "val_right_hand_rmse": 0.042136996115453004, + "val_right_hand_mae": 0.02994403801858425, + "val_right_hand_r2": 0.9350650906562805, + "val_left_hip_mse": 0.0007467518444173038, + "val_left_hip_rmse": 0.027326760591356303, + "val_left_hip_mae": 0.021101975813508034, + "val_left_hip_r2": 0.9010151624679565, + "val_right_hip_mse": 0.0007550425943918526, + "val_right_hip_rmse": 0.02747803840145531, + "val_right_hip_mae": 0.021024703979492188, + "val_right_hip_r2": 0.9037333726882935, + "val_left_knee_mse": 0.0008019152446649969, + "val_left_knee_rmse": 0.028318108070014086, + "val_left_knee_mae": 0.020441273227334023, + "val_left_knee_r2": 0.9084137082099915, + "val_right_knee_mse": 0.0008016329957172275, + "val_right_knee_rmse": 0.028313124089673104, + "val_right_knee_mae": 0.02027195133268833, + "val_right_knee_r2": 0.9186673164367676, + "val_left_foot_mse": 0.0012683238601312041, + "val_left_foot_rmse": 0.035613534788493044, + "val_left_foot_mae": 0.021143611520528793, + "val_left_foot_r2": 0.7371780872344971, + "val_right_foot_mse": 0.0012879223795607686, + "val_right_foot_rmse": 0.03588763546906885, + "val_right_foot_mae": 0.022516433149576187, + "val_right_foot_r2": 0.6932873725891113 + }, + "best_epoch": 100 + }, + { + "fold": 5, + "train_metrics": { + "train_mse": 0.0010029308032244444, + "train_rmse": 0.03166908276575822, + "train_mae": 0.02316099777817726, + "train_r2": 0.9527897834777832, + "train_head_mse": 0.0010040325578302145, + "train_head_rmse": 0.03168647278934995, + "train_head_mae": 0.024135028943419456, + "train_head_r2": 0.9003176093101501, + "train_left_shoulder_mse": 0.0006734625203534961, + "train_left_shoulder_rmse": 0.025951156435764015, + "train_left_shoulder_mae": 0.019522475078701973, + "train_left_shoulder_r2": 0.8792332410812378, + "train_left_elbow_mse": 0.0010366452625021338, + "train_left_elbow_rmse": 0.032196975983811486, + "train_left_elbow_mae": 0.0247371643781662, + "train_left_elbow_r2": 0.9251657128334045, + "train_right_shoulder_mse": 0.0005212143296375871, + "train_right_shoulder_rmse": 0.022830118914223534, + "train_right_shoulder_mae": 0.017594972625374794, + "train_right_shoulder_r2": 0.9211886525154114, + "train_right_elbow_mse": 0.0009230468422174454, + "train_right_elbow_rmse": 0.030381685967329815, + "train_right_elbow_mae": 0.02345224656164646, + "train_right_elbow_r2": 0.9423551559448242, + "train_left_hand_mse": 0.0014769204426556826, + "train_left_hand_rmse": 0.0384307226403002, + "train_left_hand_mae": 0.030191879719495773, + "train_left_hand_r2": 0.94426429271698, + "train_right_hand_mse": 0.001645134063437581, + "train_right_hand_rmse": 0.040560252260526945, + "train_right_hand_mae": 0.030592795461416245, + "train_right_hand_r2": 0.9418004155158997, + "train_left_hip_mse": 0.0007783393375575542, + "train_left_hip_rmse": 0.02789873361924434, + "train_left_hip_mae": 0.021123405545949936, + "train_left_hip_r2": 0.9004560112953186, + "train_right_hip_mse": 0.0007476237951777875, + "train_right_hip_rmse": 0.027342710092048073, + "train_right_hip_mae": 0.020731143653392792, + "train_right_hip_r2": 0.9084987640380859, + "train_left_knee_mse": 0.0008674437995068729, + "train_left_knee_rmse": 0.029452398875250772, + "train_left_knee_mae": 0.02169298194348812, + "train_left_knee_r2": 0.9086443185806274, + "train_right_knee_mse": 0.0008740806370042264, + "train_right_knee_rmse": 0.0295648547604115, + "train_right_knee_mae": 0.021198151633143425, + "train_right_knee_r2": 0.9174171686172485, + "train_left_foot_mse": 0.0013307980261743069, + "train_left_foot_rmse": 0.03648010452526564, + "train_left_foot_mae": 0.02321387082338333, + "train_left_foot_r2": 0.7120513916015625, + "train_right_foot_mse": 0.0011593584204092622, + "train_right_foot_rmse": 0.034049352716450605, + "train_right_foot_mae": 0.022906864061951637, + "train_right_foot_r2": 0.7372835278511047 + }, + "val_metrics": { + "val_mse": 0.0012481381418183446, + "val_rmse": 0.03532899859631383, + "val_mae": 0.02552439458668232, + "val_r2": 0.9397807121276855, + "val_head_mse": 0.001111267483793199, + "val_head_rmse": 0.033335678841043555, + "val_head_mae": 0.02554813213646412, + "val_head_r2": 0.882495105266571, + "val_left_shoulder_mse": 0.0007563941180706024, + "val_left_shoulder_rmse": 0.027502620203729723, + "val_left_shoulder_mae": 0.020928282290697098, + "val_left_shoulder_r2": 0.8622440695762634, + "val_left_elbow_mse": 0.00120394432451576, + "val_left_elbow_rmse": 0.03469790086612964, + "val_left_elbow_mae": 0.02694109082221985, + "val_left_elbow_r2": 0.9116610288619995, + "val_right_shoulder_mse": 0.0006667657289654016, + "val_right_shoulder_rmse": 0.02582180723662466, + "val_right_shoulder_mae": 0.019470075145363808, + "val_right_shoulder_r2": 0.8918794989585876, + "val_right_elbow_mse": 0.0012743155239149928, + "val_right_elbow_rmse": 0.03569755627371421, + "val_right_elbow_mae": 0.02677593007683754, + "val_right_elbow_r2": 0.9140562415122986, + "val_left_hand_mse": 0.0019465163350105286, + "val_left_hand_rmse": 0.04411934196030726, + "val_left_hand_mae": 0.03384125232696533, + "val_left_hand_r2": 0.923511266708374, + "val_right_hand_mse": 0.002424483187496662, + "val_right_hand_rmse": 0.04923904129343566, + "val_right_hand_mae": 0.03626913204789162, + "val_right_hand_r2": 0.9074498414993286, + "val_left_hip_mse": 0.0010060323402285576, + "val_left_hip_rmse": 0.031718012866958696, + "val_left_hip_mae": 0.023858873173594475, + "val_left_hip_r2": 0.869533121585846, + "val_right_hip_mse": 0.0010086087277159095, + "val_right_hip_rmse": 0.031758600846320505, + "val_right_hip_mae": 0.023590611293911934, + "val_right_hip_r2": 0.8722031116485596, + "val_left_knee_mse": 0.0009801719570532441, + "val_left_knee_rmse": 0.03130769804781636, + "val_left_knee_mae": 0.023882297798991203, + "val_left_knee_r2": 0.8963133096694946, + "val_right_knee_mse": 0.0013291214127093554, + "val_right_knee_rmse": 0.03645711744926298, + "val_right_knee_mae": 0.025596212595701218, + "val_right_knee_r2": 0.8839012980461121, + "val_left_foot_mse": 0.0009157413151115179, + "val_left_foot_rmse": 0.03026121800442801, + "val_left_foot_mae": 0.020988987758755684, + "val_left_foot_r2": 0.7289378046989441, + "val_right_foot_mse": 0.0016024326905608177, + "val_right_foot_rmse": 0.04003039708222762, + "val_right_foot_mae": 0.024126270785927773, + "val_right_foot_r2": 0.6630392074584961 + }, + "best_epoch": 78 + } + ], + "test_metrics": { + "test_mse": 0.0009610747219994664, + "test_rmse": 0.03100120517011341, + "test_mae": 0.022901641204953194, + "test_r2": 0.9549335241317749, + "test_head_mse": 0.00098160351626575, + "test_head_rmse": 0.03133055244111967, + "test_head_mae": 0.024242691695690155, + "test_head_r2": 0.9036559462547302, + "test_left_shoulder_mse": 0.000643729348666966, + "test_left_shoulder_rmse": 0.025371821942205212, + "test_left_shoulder_mae": 0.01927311345934868, + "test_left_shoulder_r2": 0.887075662612915, + "test_left_elbow_mse": 0.0009166893432848155, + "test_left_elbow_rmse": 0.03027687803068235, + "test_left_elbow_mae": 0.02309303544461727, + "test_left_elbow_r2": 0.9323437809944153, + "test_right_shoulder_mse": 0.0006455500843003392, + "test_right_shoulder_rmse": 0.025407677664445037, + "test_right_shoulder_mae": 0.019159311428666115, + "test_right_shoulder_r2": 0.9030331969261169, + "test_right_elbow_mse": 0.001014905283227563, + "test_right_elbow_rmse": 0.03185757811302615, + "test_right_elbow_mae": 0.02456730231642723, + "test_right_elbow_r2": 0.9350865483283997, + "test_left_hand_mse": 0.0014239158481359482, + "test_left_hand_rmse": 0.03773480950178427, + "test_left_hand_mae": 0.029283981770277023, + "test_left_hand_r2": 0.9456302523612976, + "test_right_hand_mse": 0.0017269558738917112, + "test_right_hand_rmse": 0.041556658598733746, + "test_right_hand_mae": 0.03181558847427368, + "test_right_hand_r2": 0.938625156879425, + "test_left_hip_mse": 0.0007330836378969252, + "test_left_hip_rmse": 0.027075517315407388, + "test_left_hip_mae": 0.020525647327303886, + "test_left_hip_r2": 0.9168252944946289, + "test_right_hip_mse": 0.0007723739254288375, + "test_right_hip_rmse": 0.02779161609962324, + "test_right_hip_mae": 0.020440800115466118, + "test_right_hip_r2": 0.9157125949859619, + "test_left_knee_mse": 0.0008320861379615963, + "test_left_knee_rmse": 0.0288459033133233, + "test_left_knee_mae": 0.021188151091337204, + "test_left_knee_r2": 0.9144752025604248, + "test_right_knee_mse": 0.0008250048849731684, + "test_right_knee_rmse": 0.02872289826903212, + "test_right_knee_mae": 0.02160467393696308, + "test_right_knee_r2": 0.9230653047561646, + "test_left_foot_mse": 0.0009359756368212402, + "test_left_foot_rmse": 0.030593718911260857, + "test_left_foot_mae": 0.02092571370303631, + "test_left_foot_r2": 0.7987405061721802, + "test_right_foot_mse": 0.0010420968756079674, + "test_right_foot_rmse": 0.032281525298659096, + "test_right_foot_mae": 0.02160133421421051, + "test_right_foot_r2": 0.8014936447143555 + } +} \ No newline at end of file diff --git a/A9/cv_results_optimizer/conv1d_v3_rmsprop_fold1_best.h5 b/A9/cv_results_optimizer/conv1d_v3_rmsprop_fold1_best.h5 new file mode 100644 index 0000000000000000000000000000000000000000..6a36bf187b19348de99f9f5c70356cd698c27234 --- /dev/null +++ b/A9/cv_results_optimizer/conv1d_v3_rmsprop_fold1_best.h5 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:4de660a582fd621745fa3ca760b5cbb351eff8e2d22c83e5f061621ff16baeee +size 2658520 diff --git a/A9/cv_results_optimizer/conv1d_v3_rmsprop_fold2_best.h5 b/A9/cv_results_optimizer/conv1d_v3_rmsprop_fold2_best.h5 new file mode 100644 index 0000000000000000000000000000000000000000..dbe13241a1dc523d0a52c71b7f599dca97966b00 --- /dev/null +++ b/A9/cv_results_optimizer/conv1d_v3_rmsprop_fold2_best.h5 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:4019d529e4b3ba33d40e6ac11706343872ef76534a437e6e2b12b8923626d516 +size 2658520 diff --git a/A9/cv_results_optimizer/conv1d_v3_rmsprop_fold3_best.h5 b/A9/cv_results_optimizer/conv1d_v3_rmsprop_fold3_best.h5 new file mode 100644 index 0000000000000000000000000000000000000000..18b94292ae89bad03f84d5043999c74eff57decc --- /dev/null +++ b/A9/cv_results_optimizer/conv1d_v3_rmsprop_fold3_best.h5 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b2cde06f9e86f996d64271a693e2ac32a9e4d6993a8f6e5ab95ee0dc7fedbe30 +size 2658520 diff --git a/A9/cv_results_optimizer/conv1d_v3_rmsprop_fold4_best.h5 b/A9/cv_results_optimizer/conv1d_v3_rmsprop_fold4_best.h5 new file mode 100644 index 0000000000000000000000000000000000000000..3b30c596ac7b3d00437f31929a9fbc6f0cd359fc --- /dev/null +++ b/A9/cv_results_optimizer/conv1d_v3_rmsprop_fold4_best.h5 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:17c84a57f6e38f4356e6a9d76b497097b044a3bd848ad8593a481efed5c2bbce +size 2658520 diff --git a/A9/cv_results_optimizer/conv1d_v3_rmsprop_fold5_best.h5 b/A9/cv_results_optimizer/conv1d_v3_rmsprop_fold5_best.h5 new file mode 100644 index 0000000000000000000000000000000000000000..dfb6dea70ea620d22da32ea44a508b1192754777 --- /dev/null +++ b/A9/cv_results_optimizer/conv1d_v3_rmsprop_fold5_best.h5 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:36ed93bca7871a784eec6fb6e3d0e88902fea57cebb2ca0bedc9c269f134fcf3 +size 2658520 diff --git a/A9/cv_results_optimizer/conv1d_v3_rmsprop_predictions.png b/A9/cv_results_optimizer/conv1d_v3_rmsprop_predictions.png new file mode 100644 index 0000000000000000000000000000000000000000..52751d7fd7b8f54cd5aeb9d83c9d3dd18f7c05f6 --- /dev/null +++ b/A9/cv_results_optimizer/conv1d_v3_rmsprop_predictions.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:221ec3e73b0368303b056401d77d9f9988b86d15296e689ad8de8db3ea4eb8cc +size 550513 diff --git a/A9/cv_results_optimizer/conv1d_v3_rmsprop_test_predictions.png b/A9/cv_results_optimizer/conv1d_v3_rmsprop_test_predictions.png new file mode 100644 index 0000000000000000000000000000000000000000..75f7949b52a82d23fb8bdefd6cc51126ccdb36ff --- /dev/null +++ b/A9/cv_results_optimizer/conv1d_v3_rmsprop_test_predictions.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:f7edc2a19dd8dacf0850f320c7b9d8614af11bdbaccf365d8b9979e8c186a9a6 +size 464919 diff --git a/A9/cv_results_optimizer/conv1d_v3_rmsprop_training_history.png b/A9/cv_results_optimizer/conv1d_v3_rmsprop_training_history.png new file mode 100644 index 0000000000000000000000000000000000000000..8e0369e893a728e8bf512c4fa72f5ca1eb322a8b --- /dev/null +++ b/A9/cv_results_optimizer/conv1d_v3_rmsprop_training_history.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:48bd301eb1f80cfd87b72455358fb455f4e82b7b0b1651fcebbde6bca29f37eb +size 374987 diff --git a/A9/cv_results_optimizer/conv1d_v3_sgd_cv_results.json b/A9/cv_results_optimizer/conv1d_v3_sgd_cv_results.json new file mode 100644 index 0000000000000000000000000000000000000000..76b8732ef81b688cbdbb44c3ac2eaaf953b1cb21 --- /dev/null +++ b/A9/cv_results_optimizer/conv1d_v3_sgd_cv_results.json @@ -0,0 +1,1243 @@ +{ + "model_name": "conv1d_v3_sgd", + "n_splits": 5, + "epochs": 100, + "batch_size": 32, + "patience": 10, + "aggregate_results": { + "n_folds": 5, + "train_metrics": { + "train_mse": { + "mean": 0.0007893650094047189, + "std": 2.9953482437786185e-05 + }, + "train_rmse": { + "mean": 0.028090477753854336, + "std": 0.0005385805092401822 + }, + "train_mae": { + "mean": 0.02087642215192318, + "std": 0.0004479103828863694 + }, + "train_r2": { + "mean": 0.9626668691635132, + "std": 0.0012095668164169022 + }, + "train_head_mse": { + "mean": 0.0008569558151066304, + "std": 6.2599918853009e-05 + }, + "train_head_rmse": { + "mean": 0.029254260264728826, + "std": 0.001069612766443324 + }, + "train_head_mae": { + "mean": 0.022206638008356094, + "std": 0.0006398266898285872 + }, + "train_head_r2": { + "mean": 0.9139099240303039, + "std": 0.005550016092050897 + }, + "train_left_shoulder_mse": { + "mean": 0.0005419198656454683, + "std": 2.7436509273392422e-05 + }, + "train_left_shoulder_rmse": { + "mean": 0.023271691773613608, + "std": 0.0005901081590632518 + }, + "train_left_shoulder_mae": { + "mean": 0.017712372168898583, + "std": 0.0004220556184526061 + }, + "train_left_shoulder_r2": { + "mean": 0.9025272369384766, + "std": 0.004587838538895614 + }, + "train_left_elbow_mse": { + "mean": 0.0008113744901493192, + "std": 3.8699353262124505e-05 + }, + "train_left_elbow_rmse": { + "mean": 0.028476465009554287, + "std": 0.0006822248228771365 + }, + "train_left_elbow_mae": { + "mean": 0.021978935599327086, + "std": 0.0006058776391323216 + }, + "train_left_elbow_r2": { + "mean": 0.9412422180175781, + "std": 0.002607764356986142 + }, + "train_right_shoulder_mse": { + "mean": 0.00047194843064062296, + "std": 2.7328141913656794e-05 + }, + "train_right_shoulder_rmse": { + "mean": 0.021715030769292536, + "std": 0.0006370787465465924 + }, + "train_right_shoulder_mae": { + "mean": 0.0168572049587965, + "std": 0.0004943258217377313 + }, + "train_right_shoulder_r2": { + "mean": 0.9276600480079651, + "std": 0.004107106140753526 + }, + "train_right_elbow_mse": { + "mean": 0.0007561308098956943, + "std": 3.983556448236437e-05 + }, + "train_right_elbow_rmse": { + "mean": 0.027487993445353225, + "std": 0.0007355448619303866 + }, + "train_right_elbow_mae": { + "mean": 0.02132173553109169, + "std": 0.0007703872596614945 + }, + "train_right_elbow_r2": { + "mean": 0.9520796298980713, + "std": 0.0022425133709476743 + }, + "train_left_hand_mse": { + "mean": 0.001197013328783214, + "std": 7.258199471102278e-05 + }, + "train_left_hand_rmse": { + "mean": 0.03458147640243649, + "std": 0.0010652786541287813 + }, + "train_left_hand_mae": { + "mean": 0.027329844236373902, + "std": 0.0007773837033054724 + }, + "train_left_hand_r2": { + "mean": 0.9544754266738892, + "std": 0.002545039034608703 + }, + "train_right_hand_mse": { + "mean": 0.0013189027551561594, + "std": 6.985931669839293e-05 + }, + "train_right_hand_rmse": { + "mean": 0.036303641483511825, + "std": 0.0009738430021303312 + }, + "train_right_hand_mae": { + "mean": 0.02780907414853573, + "std": 0.0008865172022988649 + }, + "train_right_hand_r2": { + "mean": 0.9526547074317933, + "std": 0.002289317031208274 + }, + "train_left_hip_mse": { + "mean": 0.0006275961408391595, + "std": 2.165099333732792e-05 + }, + "train_left_hip_rmse": { + "mean": 0.025048132586247225, + "std": 0.00043266012173483496 + }, + "train_left_hip_mae": { + "mean": 0.019128099828958512, + "std": 0.000351130392340608 + }, + "train_left_hip_r2": { + "mean": 0.9195105314254761, + "std": 0.0023584086792470585 + }, + "train_right_hip_mse": { + "mean": 0.0006125497980974615, + "std": 2.519285311737097e-05 + }, + "train_right_hip_rmse": { + "mean": 0.024744512922498258, + "std": 0.0005088008704574769 + }, + "train_right_hip_mae": { + "mean": 0.018816314265131952, + "std": 0.0002972401720138544 + }, + "train_right_hip_r2": { + "mean": 0.9245050430297852, + "std": 0.00296761971348798 + }, + "train_left_knee_mse": { + "mean": 0.0006606971845030784, + "std": 3.568050315783673e-05 + }, + "train_left_knee_rmse": { + "mean": 0.025694497028594186, + "std": 0.0007000049647273544 + }, + "train_left_knee_mae": { + "mean": 0.01918376125395298, + "std": 0.0004507816636201374 + }, + "train_left_knee_r2": { + "mean": 0.9303759217262269, + "std": 0.0032309277757513947 + }, + "train_right_knee_mse": { + "mean": 0.0006527454475872219, + "std": 2.6505881747933336e-05 + }, + "train_right_knee_rmse": { + "mean": 0.02554367067383729, + "std": 0.0005160776044033341 + }, + "train_right_knee_mae": { + "mean": 0.01917908638715744, + "std": 0.0005575652132491635 + }, + "train_right_knee_r2": { + "mean": 0.9393023014068603, + "std": 0.0027044631701589467 + }, + "train_left_foot_mse": { + "mean": 0.0008817883091978729, + "std": 4.5010139976977726e-05 + }, + "train_left_foot_rmse": { + "mean": 0.029685426155857127, + "std": 0.0007508549414010555 + }, + "train_left_foot_mae": { + "mean": 0.01961040124297142, + "std": 0.00036830056379761526 + }, + "train_left_foot_r2": { + "mean": 0.798191237449646, + "std": 0.011904195561430957 + }, + "train_right_foot_mse": { + "mean": 0.0008721229271031917, + "std": 3.656769424692616e-05 + }, + "train_right_foot_rmse": { + "mean": 0.029525141305314882, + "std": 0.0006236649776768594 + }, + "train_right_foot_mae": { + "mean": 0.020260031148791314, + "std": 9.596626445660859e-05 + }, + "train_right_foot_r2": { + "mean": 0.8055118441581726, + "std": 0.007776106550064155 + } + }, + "val_metrics": { + "val_mse": { + "mean": 0.0010372287128120662, + "std": 4.429347475099714e-05 + }, + "val_rmse": { + "mean": 0.0321987122711828, + "std": 0.0006867611590996281 + }, + "val_mae": { + "mean": 0.023268791288137435, + "std": 0.0005120849035960022 + }, + "val_r2": { + "mean": 0.9508630633354187, + "std": 0.0028151380642020152 + }, + "val_head_mse": { + "mean": 0.0010349174961447715, + "std": 3.813138337763804e-05 + }, + "val_head_rmse": { + "mean": 0.032164636246069016, + "std": 0.0005947027012169949 + }, + "val_head_mae": { + "mean": 0.024127042293548583, + "std": 0.00045306894348818785 + }, + "val_head_r2": { + "mean": 0.8956796765327454, + "std": 0.005591744006796201 + }, + "val_left_shoulder_mse": { + "mean": 0.0006796491565182805, + "std": 5.480713813779655e-05 + }, + "val_left_shoulder_rmse": { + "mean": 0.026048016864764857, + "std": 0.0010723683747696324 + }, + "val_left_shoulder_mae": { + "mean": 0.019587933644652368, + "std": 0.0008687957133171833 + }, + "val_left_shoulder_r2": { + "mean": 0.877679419517517, + "std": 0.007086242355317044 + }, + "val_left_elbow_mse": { + "mean": 0.0010634767590090633, + "std": 8.647802580854986e-05 + }, + "val_left_elbow_rmse": { + "mean": 0.032583288147343105, + "std": 0.001343909415203864 + }, + "val_left_elbow_mae": { + "mean": 0.024827723577618598, + "std": 0.0007434515656825502 + }, + "val_left_elbow_r2": { + "mean": 0.922839629650116, + "std": 0.00657239845699341 + }, + "val_right_shoulder_mse": { + "mean": 0.0006039442494511604, + "std": 5.9897178508103934e-05 + }, + "val_right_shoulder_rmse": { + "mean": 0.024545511642700073, + "std": 0.0012091764342872027 + }, + "val_right_shoulder_mae": { + "mean": 0.018706398829817773, + "std": 0.000857091009286901 + }, + "val_right_shoulder_r2": { + "mean": 0.9071678161621094, + "std": 0.009597699405620854 + }, + "val_right_elbow_mse": { + "mean": 0.0010323634021915495, + "std": 7.661492042541534e-05 + }, + "val_right_elbow_rmse": { + "mean": 0.03210896589593802, + "std": 0.0011737594664320294 + }, + "val_right_elbow_mae": { + "mean": 0.024287331849336624, + "std": 0.0009546161762534057 + }, + "val_right_elbow_r2": { + "mean": 0.9341972708702088, + "std": 0.007436081513979824 + }, + "val_left_hand_mse": { + "mean": 0.0016201801830902697, + "std": 0.0001842209214433906 + }, + "val_left_hand_rmse": { + "mean": 0.040186570529229707, + "std": 0.0022846733222751875 + }, + "val_left_hand_mae": { + "mean": 0.03091343529522419, + "std": 0.001249424589080377 + }, + "val_left_hand_r2": { + "mean": 0.938056468963623, + "std": 0.008552924955422939 + }, + "val_right_hand_mse": { + "mean": 0.0017687404528260232, + "std": 0.00016129310716344816 + }, + "val_right_hand_rmse": { + "mean": 0.04201394005542588, + "std": 0.0018892574851257964 + }, + "val_right_hand_mae": { + "mean": 0.031644441187381744, + "std": 0.0015717757650066423 + }, + "val_right_hand_r2": { + "mean": 0.9361788749694824, + "std": 0.00775691619576947 + }, + "val_left_hip_mse": { + "mean": 0.0007756986422464251, + "std": 5.078317838607621e-05 + }, + "val_left_hip_rmse": { + "mean": 0.027836606859024608, + "std": 0.0009066205504670553 + }, + "val_left_hip_mae": { + "mean": 0.020931635051965714, + "std": 0.000701100680465632 + }, + "val_left_hip_r2": { + "mean": 0.9000558972358703, + "std": 0.008404984056973647 + }, + "val_right_hip_mse": { + "mean": 0.0007685021148063243, + "std": 3.933842007997421e-05 + }, + "val_right_hip_rmse": { + "mean": 0.027712885284095302, + "std": 0.000705764859495233 + }, + "val_right_hip_mae": { + "mean": 0.020752144604921342, + "std": 0.0005741417109151696 + }, + "val_right_hip_r2": { + "mean": 0.9048043370246888, + "std": 0.007448858169816176 + }, + "val_left_knee_mse": { + "mean": 0.0008795416331849992, + "std": 2.5878340132911672e-05 + }, + "val_left_knee_rmse": { + "mean": 0.02965386886704644, + "std": 0.0004355392072053412 + }, + "val_left_knee_mae": { + "mean": 0.021511778980493546, + "std": 0.0006799483477908972 + }, + "val_left_knee_r2": { + "mean": 0.9070152759552002, + "std": 0.004385540899703959 + }, + "val_right_knee_mse": { + "mean": 0.0009175912709906697, + "std": 0.00011323985519514872 + }, + "val_right_knee_rmse": { + "mean": 0.03023760531598945, + "std": 0.001810661687095641 + }, + "val_right_knee_mae": { + "mean": 0.021672187745571135, + "std": 0.0014677484902456445 + }, + "val_right_knee_r2": { + "mean": 0.9147456645965576, + "std": 0.008416033309200339 + }, + "val_left_foot_mse": { + "mean": 0.0011708972975611686, + "std": 0.00035441969648411774 + }, + "val_left_foot_rmse": { + "mean": 0.03379503661711244, + "std": 0.005365892060897021 + }, + "val_left_foot_mae": { + "mean": 0.02151603028178215, + "std": 0.0015097507040927416 + }, + "val_left_foot_r2": { + "mean": 0.7362527132034302, + "std": 0.058585724869857206 + }, + "val_right_foot_mse": { + "mean": 0.001168471307028085, + "std": 0.00020344182988844554 + }, + "val_right_foot_rmse": { + "mean": 0.0340641811625138, + "std": 0.002846553838508027 + }, + "val_right_foot_mae": { + "mean": 0.022016202658414842, + "std": 0.0010293316419476822 + }, + "val_right_foot_r2": { + "mean": 0.7374631881713867, + "std": 0.05487051161255759 + } + }, + "best_fold": "1", + "best_fold_metrics": { + "fold": 1, + "train_metrics": { + "train_mse": 0.000799572269897908, + "train_rmse": 0.028276708965116644, + "train_mae": 0.021132977679371834, + "train_r2": 0.9624387621879578, + "train_head_mse": 0.0008569174096919596, + "train_head_rmse": 0.029273151687031575, + "train_head_mae": 0.022725969552993774, + "train_head_r2": 0.9151284098625183, + "train_left_shoulder_mse": 0.0005232715629972517, + "train_left_shoulder_rmse": 0.02287512979192144, + "train_left_shoulder_mae": 0.017492838203907013, + "train_left_shoulder_r2": 0.9073750972747803, + "train_left_elbow_mse": 0.0008000160451047122, + "train_left_elbow_rmse": 0.02828455488609839, + "train_left_elbow_mae": 0.022016357630491257, + "train_left_elbow_r2": 0.9420355558395386, + "train_right_shoulder_mse": 0.00047687540063634515, + "train_right_shoulder_rmse": 0.021837476975061592, + "train_right_shoulder_mae": 0.016936272382736206, + "train_right_shoulder_r2": 0.9278332591056824, + "train_right_elbow_mse": 0.0007715170504525304, + "train_right_elbow_rmse": 0.027776195751983934, + "train_right_elbow_mae": 0.021656254306435585, + "train_right_elbow_r2": 0.9515344500541687, + "train_left_hand_mse": 0.001203917316161096, + "train_left_hand_rmse": 0.034697511671027595, + "train_left_hand_mae": 0.027403907850384712, + "train_left_hand_r2": 0.9537967443466187, + "train_right_hand_mse": 0.001313422224484384, + "train_right_hand_rmse": 0.0362411675375447, + "train_right_hand_mae": 0.02792787365615368, + "train_right_hand_r2": 0.9530009627342224, + "train_left_hip_mse": 0.0006603015935979784, + "train_left_hip_rmse": 0.02569633424436214, + "train_left_hip_mae": 0.01962256245315075, + "train_left_hip_r2": 0.9174007773399353, + "train_right_hip_mse": 0.0006516720168292522, + "train_right_hip_rmse": 0.025527867455572003, + "train_right_hip_mae": 0.019335323944687843, + "train_right_hip_r2": 0.9214746952056885, + "train_left_knee_mse": 0.0007005551015026867, + "train_left_knee_rmse": 0.026468001464082753, + "train_left_knee_mae": 0.019592225551605225, + "train_left_knee_r2": 0.9267225861549377, + "train_right_knee_mse": 0.0006973783602006733, + "train_right_knee_rmse": 0.026407922299959027, + "train_right_knee_mae": 0.019988395273685455, + "train_right_knee_r2": 0.9351546168327332, + "train_left_foot_mse": 0.0008649443043395877, + "train_left_foot_rmse": 0.02940993546983039, + "train_left_foot_mae": 0.019732290878891945, + "train_left_foot_r2": 0.8036620020866394, + "train_right_foot_mse": 0.0008736516465432942, + "train_right_foot_rmse": 0.029557598795289415, + "train_right_foot_mae": 0.020298441872000694, + "train_right_foot_r2": 0.8031831979751587 + }, + "val_metrics": { + "val_mse": 0.0009712054161354899, + "val_rmse": 0.031164168786211675, + "val_mae": 0.0225379578769207, + "val_r2": 0.9527261853218079, + "val_head_mse": 0.0009846274042502046, + "val_head_rmse": 0.03137877314762648, + "val_head_mae": 0.02346022054553032, + "val_head_r2": 0.8944961428642273, + "val_left_shoulder_mse": 0.0005805276450701058, + "val_left_shoulder_rmse": 0.02409414130177927, + "val_left_shoulder_mae": 0.01791745238006115, + "val_left_shoulder_r2": 0.888041615486145, + "val_left_elbow_mse": 0.0009160825284197927, + "val_left_elbow_rmse": 0.030266855278006544, + "val_left_elbow_mae": 0.023573655635118484, + "val_left_elbow_r2": 0.9336064457893372, + "val_right_shoulder_mse": 0.0005738011677749455, + "val_right_shoulder_rmse": 0.023954147193647815, + "val_right_shoulder_mae": 0.018254082649946213, + "val_right_shoulder_r2": 0.9071592092514038, + "val_right_elbow_mse": 0.0010033507132902741, + "val_right_elbow_rmse": 0.03167571172507848, + "val_right_elbow_mae": 0.024461863562464714, + "val_right_elbow_r2": 0.9339475631713867, + "val_left_hand_mse": 0.0014295694418251514, + "val_left_hand_rmse": 0.03780964747025753, + "val_left_hand_mae": 0.02979523502290249, + "val_left_hand_r2": 0.9473488330841064, + "val_right_hand_mse": 0.0015942009631544352, + "val_right_hand_rmse": 0.039927446238827186, + "val_right_hand_mae": 0.030432142317295074, + "val_right_hand_r2": 0.9419484734535217, + "val_left_hip_mse": 0.0007327157072722912, + "val_left_hip_rmse": 0.027068721936439687, + "val_left_hip_mae": 0.019892238080501556, + "val_left_hip_r2": 0.8954986929893494, + "val_right_hip_mse": 0.0007353114197030663, + "val_right_hip_rmse": 0.027116626259604388, + "val_right_hip_mae": 0.020062042400240898, + "val_right_hip_r2": 0.9003650546073914, + "val_left_knee_mse": 0.0008426023996435106, + "val_left_knee_rmse": 0.02902761443252805, + "val_left_knee_mae": 0.020593440160155296, + "val_left_knee_r2": 0.9082469940185547, + "val_right_knee_mse": 0.0009017440024763346, + "val_right_knee_rmse": 0.030029052640340396, + "val_right_knee_mae": 0.021845249459147453, + "val_right_knee_r2": 0.916140079498291, + "val_left_foot_mse": 0.0012365974253043532, + "val_left_foot_rmse": 0.03516528722055819, + "val_left_foot_mae": 0.021492714062333107, + "val_left_foot_r2": 0.7087603807449341, + "val_right_foot_mse": 0.0010945394169539213, + "val_right_foot_rmse": 0.03308382409809847, + "val_right_foot_mae": 0.021213119849562645, + "val_right_foot_r2": 0.7654514312744141 + }, + "best_epoch": 100 + } + }, + "fold_details": [ + { + "fold": 1, + "train_metrics": { + "train_mse": 0.000799572269897908, + "train_rmse": 0.028276708965116644, + "train_mae": 0.021132977679371834, + "train_r2": 0.9624387621879578, + "train_head_mse": 0.0008569174096919596, + "train_head_rmse": 0.029273151687031575, + "train_head_mae": 0.022725969552993774, + "train_head_r2": 0.9151284098625183, + "train_left_shoulder_mse": 0.0005232715629972517, + "train_left_shoulder_rmse": 0.02287512979192144, + "train_left_shoulder_mae": 0.017492838203907013, + "train_left_shoulder_r2": 0.9073750972747803, + "train_left_elbow_mse": 0.0008000160451047122, + "train_left_elbow_rmse": 0.02828455488609839, + "train_left_elbow_mae": 0.022016357630491257, + "train_left_elbow_r2": 0.9420355558395386, + "train_right_shoulder_mse": 0.00047687540063634515, + "train_right_shoulder_rmse": 0.021837476975061592, + "train_right_shoulder_mae": 0.016936272382736206, + "train_right_shoulder_r2": 0.9278332591056824, + "train_right_elbow_mse": 0.0007715170504525304, + "train_right_elbow_rmse": 0.027776195751983934, + "train_right_elbow_mae": 0.021656254306435585, + "train_right_elbow_r2": 0.9515344500541687, + "train_left_hand_mse": 0.001203917316161096, + "train_left_hand_rmse": 0.034697511671027595, + "train_left_hand_mae": 0.027403907850384712, + "train_left_hand_r2": 0.9537967443466187, + "train_right_hand_mse": 0.001313422224484384, + "train_right_hand_rmse": 0.0362411675375447, + "train_right_hand_mae": 0.02792787365615368, + "train_right_hand_r2": 0.9530009627342224, + "train_left_hip_mse": 0.0006603015935979784, + "train_left_hip_rmse": 0.02569633424436214, + "train_left_hip_mae": 0.01962256245315075, + "train_left_hip_r2": 0.9174007773399353, + "train_right_hip_mse": 0.0006516720168292522, + "train_right_hip_rmse": 0.025527867455572003, + "train_right_hip_mae": 0.019335323944687843, + "train_right_hip_r2": 0.9214746952056885, + "train_left_knee_mse": 0.0007005551015026867, + "train_left_knee_rmse": 0.026468001464082753, + "train_left_knee_mae": 0.019592225551605225, + "train_left_knee_r2": 0.9267225861549377, + "train_right_knee_mse": 0.0006973783602006733, + "train_right_knee_rmse": 0.026407922299959027, + "train_right_knee_mae": 0.019988395273685455, + "train_right_knee_r2": 0.9351546168327332, + "train_left_foot_mse": 0.0008649443043395877, + "train_left_foot_rmse": 0.02940993546983039, + "train_left_foot_mae": 0.019732290878891945, + "train_left_foot_r2": 0.8036620020866394, + "train_right_foot_mse": 0.0008736516465432942, + "train_right_foot_rmse": 0.029557598795289415, + "train_right_foot_mae": 0.020298441872000694, + "train_right_foot_r2": 0.8031831979751587 + }, + "val_metrics": { + "val_mse": 0.0009712054161354899, + "val_rmse": 0.031164168786211675, + "val_mae": 0.0225379578769207, + "val_r2": 0.9527261853218079, + "val_head_mse": 0.0009846274042502046, + "val_head_rmse": 0.03137877314762648, + "val_head_mae": 0.02346022054553032, + "val_head_r2": 0.8944961428642273, + "val_left_shoulder_mse": 0.0005805276450701058, + "val_left_shoulder_rmse": 0.02409414130177927, + "val_left_shoulder_mae": 0.01791745238006115, + "val_left_shoulder_r2": 0.888041615486145, + "val_left_elbow_mse": 0.0009160825284197927, + "val_left_elbow_rmse": 0.030266855278006544, + "val_left_elbow_mae": 0.023573655635118484, + "val_left_elbow_r2": 0.9336064457893372, + "val_right_shoulder_mse": 0.0005738011677749455, + "val_right_shoulder_rmse": 0.023954147193647815, + "val_right_shoulder_mae": 0.018254082649946213, + "val_right_shoulder_r2": 0.9071592092514038, + "val_right_elbow_mse": 0.0010033507132902741, + "val_right_elbow_rmse": 0.03167571172507848, + "val_right_elbow_mae": 0.024461863562464714, + "val_right_elbow_r2": 0.9339475631713867, + "val_left_hand_mse": 0.0014295694418251514, + "val_left_hand_rmse": 0.03780964747025753, + "val_left_hand_mae": 0.02979523502290249, + "val_left_hand_r2": 0.9473488330841064, + "val_right_hand_mse": 0.0015942009631544352, + "val_right_hand_rmse": 0.039927446238827186, + "val_right_hand_mae": 0.030432142317295074, + "val_right_hand_r2": 0.9419484734535217, + "val_left_hip_mse": 0.0007327157072722912, + "val_left_hip_rmse": 0.027068721936439687, + "val_left_hip_mae": 0.019892238080501556, + "val_left_hip_r2": 0.8954986929893494, + "val_right_hip_mse": 0.0007353114197030663, + "val_right_hip_rmse": 0.027116626259604388, + "val_right_hip_mae": 0.020062042400240898, + "val_right_hip_r2": 0.9003650546073914, + "val_left_knee_mse": 0.0008426023996435106, + "val_left_knee_rmse": 0.02902761443252805, + "val_left_knee_mae": 0.020593440160155296, + "val_left_knee_r2": 0.9082469940185547, + "val_right_knee_mse": 0.0009017440024763346, + "val_right_knee_rmse": 0.030029052640340396, + "val_right_knee_mae": 0.021845249459147453, + "val_right_knee_r2": 0.916140079498291, + "val_left_foot_mse": 0.0012365974253043532, + "val_left_foot_rmse": 0.03516528722055819, + "val_left_foot_mae": 0.021492714062333107, + "val_left_foot_r2": 0.7087603807449341, + "val_right_foot_mse": 0.0010945394169539213, + "val_right_foot_rmse": 0.03308382409809847, + "val_right_foot_mae": 0.021213119849562645, + "val_right_foot_r2": 0.7654514312744141 + }, + "best_epoch": 100 + }, + { + "fold": 2, + "train_metrics": { + "train_mse": 0.0007954432512633502, + "train_rmse": 0.028203603515567832, + "train_mae": 0.020952094346284866, + "train_r2": 0.9620752334594727, + "train_head_mse": 0.0008138317498378456, + "train_head_rmse": 0.028527736500427888, + "train_head_mae": 0.02173539809882641, + "train_head_r2": 0.9167608022689819, + "train_left_shoulder_mse": 0.000554614991415292, + "train_left_shoulder_rmse": 0.023550265209022425, + "train_left_shoulder_mae": 0.018124468624591827, + "train_left_shoulder_r2": 0.899735152721405, + "train_left_elbow_mse": 0.0008272298146039248, + "train_left_elbow_rmse": 0.02876160312993566, + "train_left_elbow_mae": 0.02222738415002823, + "train_left_elbow_r2": 0.9404944181442261, + "train_right_shoulder_mse": 0.0004917911719530821, + "train_right_shoulder_rmse": 0.022176365165488283, + "train_right_shoulder_mae": 0.017187468707561493, + "train_right_shoulder_r2": 0.9233701825141907, + "train_right_elbow_mse": 0.0007596459472551942, + "train_right_elbow_rmse": 0.027561675334696077, + "train_right_elbow_mae": 0.021363751962780952, + "train_right_elbow_r2": 0.9514601230621338, + "train_left_hand_mse": 0.001252798712812364, + "train_left_hand_rmse": 0.03539489670577334, + "train_left_hand_mae": 0.027833886444568634, + "train_left_hand_r2": 0.9525672197341919, + "train_right_hand_mse": 0.001390748075209558, + "train_right_hand_rmse": 0.03729273488508932, + "train_right_hand_mae": 0.02846905216574669, + "train_right_hand_r2": 0.9498114585876465, + "train_left_hip_mse": 0.0006267037824727595, + "train_left_hip_rmse": 0.025034052458057194, + "train_left_hip_mae": 0.019026972353458405, + "train_left_hip_r2": 0.9170374274253845, + "train_right_hip_mse": 0.0006142182392068207, + "train_right_hip_rmse": 0.024783426704288104, + "train_right_hip_mae": 0.018734248355031013, + "train_right_hip_r2": 0.9214041233062744, + "train_left_knee_mse": 0.0006384886801242828, + "train_left_knee_rmse": 0.02526833354466184, + "train_left_knee_mae": 0.0189900491386652, + "train_left_knee_r2": 0.931297779083252, + "train_right_knee_mse": 0.0006431613001041114, + "train_right_knee_rmse": 0.02536062499435121, + "train_right_knee_mae": 0.01902724616229534, + "train_right_knee_r2": 0.9402728080749512, + "train_left_foot_mse": 0.0008410234586335719, + "train_left_foot_rmse": 0.029000404456379084, + "train_left_foot_mae": 0.019311005249619484, + "train_left_foot_r2": 0.8052555322647095, + "train_right_foot_mse": 0.0008865066338330507, + "train_right_foot_rmse": 0.029774261264270702, + "train_right_foot_mae": 0.020346298813819885, + "train_right_foot_r2": 0.8059353232383728 + }, + "val_metrics": { + "val_mse": 0.0010257632238790393, + "val_rmse": 0.03202753852357435, + "val_mae": 0.023714881390333176, + "val_r2": 0.9529079794883728, + "val_head_mse": 0.001075838110409677, + "val_head_rmse": 0.03279997119525682, + "val_head_mae": 0.024712134152650833, + "val_head_r2": 0.8986839652061462, + "val_left_shoulder_mse": 0.0006921578897163272, + "val_left_shoulder_rmse": 0.026308893737980076, + "val_left_shoulder_mae": 0.020319566130638123, + "val_left_shoulder_r2": 0.8775418996810913, + "val_left_elbow_mse": 0.0011139002162963152, + "val_left_elbow_rmse": 0.03337514368952312, + "val_left_elbow_mae": 0.025507641956210136, + "val_left_elbow_r2": 0.9168089628219604, + "val_right_shoulder_mse": 0.0006990677793510258, + "val_right_shoulder_rmse": 0.02643988992698392, + "val_right_shoulder_mae": 0.019846629351377487, + "val_right_shoulder_r2": 0.8991446495056152, + "val_right_elbow_mse": 0.0010325706098228693, + "val_right_elbow_rmse": 0.03213363673509224, + "val_right_elbow_mae": 0.024471038952469826, + "val_right_elbow_r2": 0.9363642930984497, + "val_left_hand_mse": 0.0018351346952840686, + "val_left_hand_rmse": 0.04283847213993595, + "val_left_hand_mae": 0.03246122971177101, + "val_left_hand_r2": 0.9286302924156189, + "val_right_hand_mse": 0.0018411694327369332, + "val_right_hand_rmse": 0.04290885028449182, + "val_right_hand_mae": 0.0327257476747036, + "val_right_hand_r2": 0.9350240230560303, + "val_left_hip_mse": 0.0007457703468389809, + "val_left_hip_rmse": 0.02730879614408114, + "val_left_hip_mae": 0.021019859239459038, + "val_left_hip_r2": 0.9148551225662231, + "val_right_hip_mse": 0.0007766678463667631, + "val_right_hip_rmse": 0.027868761120056325, + "val_right_hip_mae": 0.020919732749462128, + "val_right_hip_r2": 0.916516125202179, + "val_left_knee_mse": 0.0008762734360061586, + "val_left_knee_rmse": 0.029601916086735982, + "val_left_knee_mae": 0.021764036267995834, + "val_left_knee_r2": 0.9143141508102417, + "val_right_knee_mse": 0.0008043404668569565, + "val_right_knee_rmse": 0.028360896792184773, + "val_right_knee_mae": 0.021442532539367676, + "val_right_knee_r2": 0.9248389005661011, + "val_left_foot_mse": 0.0008700679754838347, + "val_left_foot_rmse": 0.029496914677366422, + "val_left_foot_mae": 0.02151750959455967, + "val_left_foot_r2": 0.8100094795227051, + "val_right_foot_mse": 0.0009719639201648533, + "val_right_foot_rmse": 0.031176335900244167, + "val_right_foot_mae": 0.021585779264569283, + "val_right_foot_r2": 0.7655217051506042 + }, + "best_epoch": 100 + }, + { + "fold": 3, + "train_metrics": { + "train_mse": 0.0007330648950301111, + "train_rmse": 0.027075171191150594, + "train_mae": 0.019997116178274155, + "train_r2": 0.9649341106414795, + "train_head_mse": 0.0007687760517001152, + "train_head_rmse": 0.027726811062581923, + "train_head_mae": 0.021219803020358086, + "train_head_r2": 0.9218787550926208, + "train_left_shoulder_mse": 0.0005016583600081503, + "train_left_shoulder_rmse": 0.022397731135276856, + "train_left_shoulder_mae": 0.017016062512993813, + "train_left_shoulder_r2": 0.9079365134239197, + "train_left_elbow_mse": 0.0007477794424630702, + "train_left_elbow_rmse": 0.02734555617395759, + "train_left_elbow_mae": 0.02082834579050541, + "train_left_elbow_r2": 0.9453168511390686, + "train_right_shoulder_mse": 0.00042302312795072794, + "train_right_shoulder_rmse": 0.02056752605324053, + "train_right_shoulder_mae": 0.01595819927752018, + "train_right_shoulder_r2": 0.9342742562294006, + "train_right_elbow_mse": 0.0006807534955441952, + "train_right_elbow_rmse": 0.026091253238282656, + "train_right_elbow_mae": 0.01984962448477745, + "train_right_elbow_r2": 0.9561784267425537, + "train_left_hand_mse": 0.0010637163650244474, + "train_left_hand_rmse": 0.03261466487677663, + "train_left_hand_mae": 0.025896506384015083, + "train_left_hand_r2": 0.959130585193634, + "train_right_hand_mse": 0.001192230498418212, + "train_right_hand_rmse": 0.034528690945621034, + "train_right_hand_mae": 0.026152966544032097, + "train_right_hand_r2": 0.9564444422721863, + "train_left_hip_mse": 0.0005933017819188535, + "train_left_hip_rmse": 0.024357786884666954, + "train_left_hip_mae": 0.018542489036917686, + "train_left_hip_r2": 0.923535943031311, + "train_right_hip_mse": 0.0005751334829255939, + "train_right_hip_rmse": 0.023981940766451614, + "train_right_hip_mae": 0.01841118186712265, + "train_right_hip_r2": 0.9290626049041748, + "train_left_knee_mse": 0.0006020760047249496, + "train_left_knee_rmse": 0.024537237104550903, + "train_left_knee_mae": 0.01843397691845894, + "train_left_knee_r2": 0.9360515475273132, + "train_right_knee_mse": 0.0006197855109348893, + "train_right_knee_rmse": 0.02489549177933405, + "train_right_knee_mae": 0.01833152398467064, + "train_right_knee_r2": 0.9420504570007324, + "train_left_foot_mse": 0.0008414968615397811, + "train_left_foot_rmse": 0.02900856531336531, + "train_left_foot_mae": 0.019092494621872902, + "train_left_foot_r2": 0.8025258779525757, + "train_right_foot_mse": 0.000920112244784832, + "train_right_foot_rmse": 0.030333352020257043, + "train_right_foot_mae": 0.02022935077548027, + "train_right_foot_r2": 0.7930182218551636 + }, + "val_metrics": { + "val_mse": 0.0010304610477760434, + "val_rmse": 0.032100795126850726, + "val_mae": 0.022817054763436317, + "val_r2": 0.9533315896987915, + "val_head_mse": 0.0009931100066751242, + "val_head_rmse": 0.031513647942996444, + "val_head_mae": 0.02387259341776371, + "val_head_r2": 0.904353141784668, + "val_left_shoulder_mse": 0.0007283498416654766, + "val_left_shoulder_rmse": 0.026987957345184103, + "val_left_shoulder_mae": 0.01961475796997547, + "val_left_shoulder_r2": 0.8786085247993469, + "val_left_elbow_mse": 0.0010832642437890172, + "val_left_elbow_rmse": 0.03291297986796421, + "val_left_elbow_mae": 0.02455228567123413, + "val_left_elbow_r2": 0.9243826270103455, + "val_right_shoulder_mse": 0.0005285093211568892, + "val_right_shoulder_rmse": 0.02298933059392746, + "val_right_shoulder_mae": 0.017434336245059967, + "val_right_shoulder_r2": 0.9231408834457397, + "val_right_elbow_mse": 0.0009437507833354175, + "val_right_elbow_rmse": 0.030720527067995065, + "val_right_elbow_mae": 0.023300176486372948, + "val_right_elbow_r2": 0.9436088800430298, + "val_left_hand_mse": 0.0014170584036037326, + "val_left_hand_rmse": 0.0376438361966967, + "val_left_hand_mae": 0.029399018734693527, + "val_left_hand_r2": 0.9480902552604675, + "val_right_hand_mse": 0.0016830416861921549, + "val_right_hand_rmse": 0.04102489105643249, + "val_right_hand_mae": 0.03055286593735218, + "val_right_hand_r2": 0.9434441328048706, + "val_left_hip_mse": 0.0008435713825747371, + "val_left_hip_rmse": 0.029044300345760388, + "val_left_hip_mae": 0.020816070958971977, + "val_left_hip_r2": 0.8938534259796143, + "val_right_hip_mse": 0.0007749336655251682, + "val_right_hip_rmse": 0.027837630386316436, + "val_right_hip_mae": 0.020374655723571777, + "val_right_hip_r2": 0.9048157334327698, + "val_left_knee_mse": 0.0009216305334120989, + "val_left_knee_rmse": 0.030358368424737502, + "val_left_knee_mae": 0.021605879068374634, + "val_left_knee_r2": 0.9057175517082214, + "val_right_knee_mse": 0.0008666404755786061, + "val_right_knee_rmse": 0.029438758050886013, + "val_right_knee_mae": 0.020404024049639702, + "val_right_knee_r2": 0.9212047457695007, + "val_left_foot_mse": 0.0014755066949874163, + "val_left_foot_rmse": 0.03841232477978151, + "val_left_foot_mae": 0.022392049431800842, + "val_left_foot_r2": 0.6926833391189575, + "val_right_foot_mse": 0.0011366282124072313, + "val_right_foot_rmse": 0.033713917191676664, + "val_right_foot_mae": 0.0223030224442482, + "val_right_foot_r2": 0.7548067569732666 + }, + "best_epoch": 100 + }, + { + "fold": 4, + "train_metrics": { + "train_mse": 0.0008230587118305266, + "train_rmse": 0.028688999840191826, + "train_mae": 0.02121608704328537, + "train_r2": 0.9613409638404846, + "train_head_mse": 0.0008968102629296482, + "train_head_rmse": 0.029946790528029014, + "train_head_mae": 0.022410815581679344, + "train_head_r2": 0.9099450707435608, + "train_left_shoulder_mse": 0.0005818497738800943, + "train_left_shoulder_rmse": 0.024121562426179907, + "train_left_shoulder_mae": 0.018138756975531578, + "train_left_shoulder_r2": 0.8958947062492371, + "train_left_elbow_mse": 0.0008665864588692784, + "train_left_elbow_rmse": 0.029437840594535435, + "train_left_elbow_mae": 0.022604407742619514, + "train_left_elbow_r2": 0.9372170567512512, + "train_right_shoulder_mse": 0.0005017842631787062, + "train_right_shoulder_rmse": 0.02240054158226328, + "train_right_shoulder_mae": 0.017400674521923065, + "train_right_shoulder_r2": 0.9233256578445435, + "train_right_elbow_mse": 0.0007986649870872498, + "train_right_elbow_rmse": 0.028260661476463174, + "train_right_elbow_mae": 0.022077562287449837, + "train_right_elbow_r2": 0.949316680431366, + "train_left_hand_mse": 0.0012705287663266063, + "train_left_hand_rmse": 0.03564447736082837, + "train_left_hand_mae": 0.028170930221676826, + "train_left_hand_r2": 0.9519454836845398, + "train_right_hand_mse": 0.0013749937061220407, + "train_right_hand_rmse": 0.03708090756874811, + "train_right_hand_mae": 0.02866746112704277, + "train_right_hand_r2": 0.9508244395256042, + "train_left_hip_mse": 0.0006356998346745968, + "train_left_hip_rmse": 0.025213088558813986, + "train_left_hip_mae": 0.019211633130908012, + "train_left_hip_r2": 0.9191244840621948, + "train_right_hip_mse": 0.0006216883775778115, + "train_right_hip_rmse": 0.02493367958360361, + "train_right_hip_mae": 0.018777886405587196, + "train_right_hip_r2": 0.9240219593048096, + "train_left_knee_mse": 0.0006792038329876959, + "train_left_knee_rmse": 0.02606153934416952, + "train_left_knee_mae": 0.019220683723688126, + "train_left_knee_r2": 0.9297556281089783, + "train_right_knee_mse": 0.0006386565510183573, + "train_right_knee_rmse": 0.025271655090602144, + "train_right_knee_mae": 0.019009625539183617, + "train_right_knee_r2": 0.9418385624885559, + "train_left_foot_mse": 0.0009604742517694831, + "train_left_foot_rmse": 0.030991519029719777, + "train_left_foot_mae": 0.019779635593295097, + "train_left_foot_r2": 0.7744653224945068, + "train_right_foot_mse": 0.0008728221873752773, + "train_right_foot_rmse": 0.029543564229376207, + "train_right_foot_mae": 0.02033906988799572, + "train_right_foot_r2": 0.8084110021591187 + }, + "val_metrics": { + "val_mse": 0.0011087137972936034, + "val_rmse": 0.03329735420860948, + "val_mae": 0.023410344496369362, + "val_r2": 0.9460092186927795, + "val_head_mse": 0.0010583829134702682, + "val_head_rmse": 0.032532797504522544, + "val_head_mae": 0.024052107706665993, + "val_head_r2": 0.8932270407676697, + "val_left_shoulder_mse": 0.0007298073614947498, + "val_left_shoulder_rmse": 0.027014947001516583, + "val_left_shoulder_mae": 0.020169688388705254, + "val_left_shoulder_r2": 0.8657537698745728, + "val_left_elbow_mse": 0.0011719056637957692, + "val_left_elbow_rmse": 0.03423310771454688, + "val_left_elbow_mae": 0.02563452534377575, + "val_left_elbow_r2": 0.9151396751403809, + "val_right_shoulder_mse": 0.0005757335457019508, + "val_right_shoulder_rmse": 0.023994448226661743, + "val_right_shoulder_mae": 0.018562840297818184, + "val_right_shoulder_r2": 0.9105977416038513, + "val_right_elbow_mse": 0.001008121413178742, + "val_right_elbow_rmse": 0.03175092775303963, + "val_right_elbow_mae": 0.023313378915190697, + "val_right_elbow_r2": 0.9362453818321228, + "val_left_hand_mse": 0.0018318590009585023, + "val_left_hand_rmse": 0.04280022197323867, + "val_left_hand_mae": 0.032255031168460846, + "val_left_hand_r2": 0.928585410118103, + "val_right_hand_mse": 0.0016763833118602633, + "val_right_hand_rmse": 0.04094366021571915, + "val_right_hand_mae": 0.03028409741818905, + "val_right_hand_r2": 0.9386909604072571, + "val_left_hip_mse": 0.0007256250246427953, + "val_left_hip_rmse": 0.026937427951510058, + "val_left_hip_mae": 0.020837603136897087, + "val_left_hip_r2": 0.9038155674934387, + "val_right_hip_mse": 0.0007214293582364917, + "val_right_hip_rmse": 0.0268594370424343, + "val_right_hip_mae": 0.020656084641814232, + "val_right_hip_r2": 0.9080190062522888, + "val_left_knee_mse": 0.0008686006767675281, + "val_left_knee_rmse": 0.029472032111266575, + "val_left_knee_mae": 0.021012838929891586, + "val_left_knee_r2": 0.9007976055145264, + "val_right_knee_mse": 0.000880661653354764, + "val_right_knee_rmse": 0.02967594401792071, + "val_right_knee_mae": 0.02030934952199459, + "val_right_knee_r2": 0.9106491804122925, + "val_left_foot_mse": 0.0016041588969528675, + "val_left_foot_rmse": 0.040051952473666844, + "val_left_foot_mae": 0.02335432730615139, + "val_left_foot_r2": 0.6675864458084106, + "val_right_foot_mse": 0.0015606109518557787, + "val_right_foot_rmse": 0.03950456874661181, + "val_right_foot_mae": 0.023892585188150406, + "val_right_foot_r2": 0.6283478140830994 + }, + "best_epoch": 100 + }, + { + "fold": 5, + "train_metrics": { + "train_mse": 0.0007956859190016985, + "train_rmse": 0.028207905257244793, + "train_mae": 0.021083835512399673, + "train_r2": 0.9625452756881714, + "train_head_mse": 0.0009484436013735831, + "train_head_rmse": 0.030796811545573725, + "train_head_mae": 0.02294120378792286, + "train_head_r2": 0.9058365821838379, + "train_left_shoulder_mse": 0.0005482046399265528, + "train_left_shoulder_rmse": 0.023413770305667405, + "train_left_shoulder_mae": 0.01778973452746868, + "train_left_shoulder_r2": 0.9016947150230408, + "train_left_elbow_mse": 0.0008152606897056103, + "train_left_elbow_rmse": 0.02855277026324434, + "train_left_elbow_mae": 0.022218182682991028, + "train_left_elbow_r2": 0.9411472082138062, + "train_right_shoulder_mse": 0.0004662681894842535, + "train_right_shoulder_rmse": 0.021593244070409003, + "train_right_shoulder_mae": 0.016803409904241562, + "train_right_shoulder_r2": 0.9294968843460083, + "train_right_elbow_mse": 0.0007700725691393018, + "train_right_elbow_rmse": 0.0277501814253403, + "train_right_elbow_mae": 0.021661484614014626, + "train_right_elbow_r2": 0.9519084692001343, + "train_left_hand_mse": 0.0011941054835915565, + "train_left_hand_rmse": 0.03455583139777651, + "train_left_hand_mae": 0.02734399028122425, + "train_left_hand_r2": 0.9549371004104614, + "train_right_hand_mse": 0.0013231192715466022, + "train_right_hand_rmse": 0.036374706480556, + "train_right_hand_mae": 0.027828017249703407, + "train_right_hand_r2": 0.9531922340393066, + "train_left_hip_mse": 0.0006219737115316093, + "train_left_hip_rmse": 0.024939400785335828, + "train_left_hip_mae": 0.019236842170357704, + "train_left_hip_r2": 0.9204540252685547, + "train_right_hip_mse": 0.000600036873947829, + "train_right_hip_rmse": 0.02449565010257595, + "train_right_hip_mae": 0.01882293075323105, + "train_right_hip_r2": 0.9265618324279785, + "train_left_knee_mse": 0.0006831623031757772, + "train_left_knee_rmse": 0.026137373685505918, + "train_left_knee_mae": 0.019681870937347412, + "train_left_knee_r2": 0.9280520677566528, + "train_right_knee_mse": 0.0006647455156780779, + "train_right_knee_rmse": 0.025782659204940012, + "train_right_knee_mae": 0.01953864097595215, + "train_right_knee_r2": 0.9371950626373291, + "train_left_foot_mse": 0.0009010026697069407, + "train_left_foot_rmse": 0.030016706509991078, + "train_left_foot_mae": 0.020136579871177673, + "train_left_foot_r2": 0.8050474524497986, + "train_right_foot_mse": 0.0008075219229795039, + "train_right_foot_rmse": 0.028416930217381044, + "train_right_foot_mae": 0.020086994394659996, + "train_right_foot_r2": 0.8170114755630493 + }, + "val_metrics": { + "val_mse": 0.0010500000789761543, + "val_rmse": 0.032403704710667794, + "val_mae": 0.023863717913627625, + "val_r2": 0.9493403434753418, + "val_head_mse": 0.0010626290459185839, + "val_head_rmse": 0.0325979914399428, + "val_head_mae": 0.024538155645132065, + "val_head_r2": 0.8876380920410156, + "val_left_shoulder_mse": 0.0006674030446447432, + "val_left_shoulder_rmse": 0.025834144937364256, + "val_left_shoulder_mae": 0.019918203353881836, + "val_left_shoulder_r2": 0.8784512877464294, + "val_left_elbow_mse": 0.001032231142744422, + "val_left_elbow_rmse": 0.032128354186674764, + "val_left_elbow_mae": 0.024870509281754494, + "val_left_elbow_r2": 0.9242604374885559, + "val_right_shoulder_mse": 0.0006426094332709908, + "val_right_shoulder_rmse": 0.025349742272279433, + "val_right_shoulder_mae": 0.01943410560488701, + "val_right_shoulder_r2": 0.8957965970039368, + "val_right_elbow_mse": 0.0011740234913304448, + "val_right_elbow_rmse": 0.03426402619848468, + "val_right_elbow_mae": 0.025890201330184937, + "val_right_elbow_r2": 0.9208202362060547, + "val_left_hand_mse": 0.001587279373779893, + "val_left_hand_rmse": 0.039840674866019686, + "val_left_hand_mae": 0.030656661838293076, + "val_left_hand_r2": 0.9376275539398193, + "val_right_hand_mse": 0.002048906870186329, + "val_right_hand_rmse": 0.045264852481658756, + "val_right_hand_mae": 0.03422735258936882, + "val_right_hand_r2": 0.9217867851257324, + "val_left_hip_mse": 0.0008308107499033213, + "val_left_hip_rmse": 0.028823787917331777, + "val_left_hip_mae": 0.02209240384399891, + "val_left_hip_r2": 0.8922566771507263, + "val_right_hip_mse": 0.0008341682842001319, + "val_right_hip_rmse": 0.028881971612065058, + "val_right_hip_mae": 0.02174820750951767, + "val_right_hip_r2": 0.8943057656288147, + "val_left_knee_mse": 0.0008886011200957, + "val_left_knee_rmse": 0.0298094132799641, + "val_left_knee_mae": 0.022582700476050377, + "val_left_knee_r2": 0.9060000777244568, + "val_right_knee_mse": 0.0011345697566866875, + "val_right_knee_rmse": 0.03368337507861538, + "val_right_knee_mae": 0.02435978315770626, + "val_right_knee_r2": 0.9008954167366028, + "val_left_foot_mse": 0.0006681554950773716, + "val_left_foot_rmse": 0.025848703934189265, + "val_left_foot_mae": 0.018823551014065742, + "val_left_foot_r2": 0.8022239208221436, + "val_right_foot_mse": 0.0010786140337586403, + "val_right_foot_rmse": 0.0328422598759379, + "val_right_foot_mae": 0.02108650654554367, + "val_right_foot_r2": 0.7731882333755493 + }, + "best_epoch": 100 + } + ], + "test_metrics": { + "test_mse": 0.0010160746751353145, + "test_rmse": 0.031875926263174134, + "test_mae": 0.023561371490359306, + "test_r2": 0.9523544311523438, + "test_head_mse": 0.0010892950231209397, + "test_head_rmse": 0.033004469744580656, + "test_head_mae": 0.025483572855591774, + "test_head_r2": 0.8930860757827759, + "test_left_shoulder_mse": 0.000668752589263022, + "test_left_shoulder_rmse": 0.025860251144623904, + "test_left_shoulder_mae": 0.01916496641933918, + "test_left_shoulder_r2": 0.8826860189437866, + "test_left_elbow_mse": 0.0009737054933793843, + "test_left_elbow_rmse": 0.031204254411528316, + "test_left_elbow_mae": 0.023772895336151123, + "test_left_elbow_r2": 0.9281356930732727, + "test_right_shoulder_mse": 0.0006809979095123708, + "test_right_shoulder_rmse": 0.026095936647539033, + "test_right_shoulder_mae": 0.01966923102736473, + "test_right_shoulder_r2": 0.8977086544036865, + "test_right_elbow_mse": 0.0010758178541436791, + "test_right_elbow_rmse": 0.0327996624089895, + "test_right_elbow_mae": 0.02562992088496685, + "test_right_elbow_r2": 0.9311906099319458, + "test_left_hand_mse": 0.0015080207958817482, + "test_left_hand_rmse": 0.03883324343757225, + "test_left_hand_mae": 0.030159827321767807, + "test_left_hand_r2": 0.9424188733100891, + "test_right_hand_mse": 0.0017818583874031901, + "test_right_hand_rmse": 0.042212064476914536, + "test_right_hand_mae": 0.03312547504901886, + "test_right_hand_r2": 0.9366739392280579, + "test_left_hip_mse": 0.000800485082436353, + "test_left_hip_rmse": 0.028292845074971745, + "test_left_hip_mae": 0.020972896367311478, + "test_left_hip_r2": 0.9091780185699463, + "test_right_hip_mse": 0.0008328694384545088, + "test_right_hip_rmse": 0.028859477445970998, + "test_right_hip_mae": 0.02094716578722, + "test_right_hip_r2": 0.9091108441352844, + "test_left_knee_mse": 0.000877024547662586, + "test_left_knee_rmse": 0.02961460024485534, + "test_left_knee_mae": 0.02191762998700142, + "test_left_knee_r2": 0.9098562598228455, + "test_right_knee_mse": 0.0008561272406950593, + "test_right_knee_rmse": 0.02925965209456632, + "test_right_knee_mae": 0.0220179446041584, + "test_right_knee_r2": 0.9201630353927612, + "test_left_foot_mse": 0.0009719498339109123, + "test_left_foot_rmse": 0.031176109986829854, + "test_left_foot_mae": 0.02157977968454361, + "test_left_foot_r2": 0.7910050749778748, + "test_right_foot_mse": 0.0010920674540102482, + "test_right_foot_rmse": 0.033046443893560595, + "test_right_foot_mae": 0.02185649983584881, + "test_right_foot_r2": 0.7919749021530151 + } +} \ No newline at end of file diff --git a/A9/cv_results_optimizer/conv1d_v3_sgd_fold1_best.h5 b/A9/cv_results_optimizer/conv1d_v3_sgd_fold1_best.h5 new file mode 100644 index 0000000000000000000000000000000000000000..3ca2022d8747623551da535249c1e9359b458d51 --- /dev/null +++ b/A9/cv_results_optimizer/conv1d_v3_sgd_fold1_best.h5 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:6ba5f0a355766de7111fb53f45ffdd8879e3e66e0e5c740a72ac5475f31aae6d +size 2658520 diff --git a/A9/cv_results_optimizer/conv1d_v3_sgd_fold2_best.h5 b/A9/cv_results_optimizer/conv1d_v3_sgd_fold2_best.h5 new file mode 100644 index 0000000000000000000000000000000000000000..a31dbe1076733ca96948f947bb5555a1ce81a8c7 --- /dev/null +++ b/A9/cv_results_optimizer/conv1d_v3_sgd_fold2_best.h5 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:012dcc37b5716d7fdcef5ebf6a5afc4a2b9a2a9950c98017f2cfcf07364a860f +size 2658520 diff --git a/A9/cv_results_optimizer/conv1d_v3_sgd_fold3_best.h5 b/A9/cv_results_optimizer/conv1d_v3_sgd_fold3_best.h5 new file mode 100644 index 0000000000000000000000000000000000000000..a3b3074d352591318fbbf34c76dd72e49fc745e4 --- /dev/null +++ b/A9/cv_results_optimizer/conv1d_v3_sgd_fold3_best.h5 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:1e1958837391e3fa871e4af55eb1a7a90029a0ad025c6522f9d756fbaadf5b6d +size 2658520 diff --git a/A9/cv_results_optimizer/conv1d_v3_sgd_fold4_best.h5 b/A9/cv_results_optimizer/conv1d_v3_sgd_fold4_best.h5 new file mode 100644 index 0000000000000000000000000000000000000000..271fd25460b74653f7153811fc2b00b0c6d147a5 --- /dev/null +++ b/A9/cv_results_optimizer/conv1d_v3_sgd_fold4_best.h5 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:82c472f5b8dcc29ae1241fb63ed934bba9a14fe80c47dc6b794e1f03729d2e41 +size 2658520 diff --git a/A9/cv_results_optimizer/conv1d_v3_sgd_fold5_best.h5 b/A9/cv_results_optimizer/conv1d_v3_sgd_fold5_best.h5 new file mode 100644 index 0000000000000000000000000000000000000000..e5e34a9da8cdbf6a0f186f82fcdceb21b5f4fa0f --- /dev/null +++ b/A9/cv_results_optimizer/conv1d_v3_sgd_fold5_best.h5 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b6fdd7f78f1379c44aab534c185ba1d05947746aab6a58398620648831a30d4c +size 2658520 diff --git a/A9/cv_results_optimizer/conv1d_v3_sgd_predictions.png b/A9/cv_results_optimizer/conv1d_v3_sgd_predictions.png new file mode 100644 index 0000000000000000000000000000000000000000..6aabcbc641054c378baa5a0d26df860fcc68ca2d --- /dev/null +++ b/A9/cv_results_optimizer/conv1d_v3_sgd_predictions.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:5ff49d3d646092132c41184df833d4a526da8b3447c152daefd4172b048fac95 +size 561958 diff --git a/A9/cv_results_optimizer/conv1d_v3_sgd_test_predictions.png b/A9/cv_results_optimizer/conv1d_v3_sgd_test_predictions.png new file mode 100644 index 0000000000000000000000000000000000000000..9cead8ea8475d13e990a7bc32f3d0867a784e4ed --- /dev/null +++ b/A9/cv_results_optimizer/conv1d_v3_sgd_test_predictions.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:822e7f43815606710b5cfd047b7ab6ae7e7f5d7bb1d5e3c87602a6d7ca5c6188 +size 464998 diff --git a/A9/cv_results_optimizer/conv1d_v3_sgd_training_history.png b/A9/cv_results_optimizer/conv1d_v3_sgd_training_history.png new file mode 100644 index 0000000000000000000000000000000000000000..3ba713f7808ee5dfd768168e4237b191c58238d0 --- /dev/null +++ b/A9/cv_results_optimizer/conv1d_v3_sgd_training_history.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:10ee6f2b561dd39f5ad62910aa3e98fd1997da2e216ca65fabe556d8b9ef00c8 +size 392949 diff --git a/A9/cv_results_optimizer/summary.txt b/A9/cv_results_optimizer/summary.txt new file mode 100644 index 0000000000000000000000000000000000000000..80f83ad9c89a6b096e47a67bd8ab30e49644ec94 --- /dev/null +++ b/A9/cv_results_optimizer/summary.txt @@ -0,0 +1,392 @@ +Cross-Validation Training Summary - Optimizer Variants +================================================== + +CONV1D_V3: +---------------------------------------- + + Optimizer: SGD + Best Fold: 1 + Val RMSE: 0.032199 ± 0.000687 + Val MAE: 0.023269 ± 0.000512 + Val R²: 0.950863 ± 0.002815 + Test RMSE: 0.031876 + Test MAE: 0.023561 + Test R²: 0.952354 + + Optimizer: RMSPROP + Best Fold: 1 + Val RMSE: 0.033498 ± 0.001768 + Val MAE: 0.024131 ± 0.001299 + Val R²: 0.946803 ± 0.004864 + Test RMSE: 0.031001 + Test MAE: 0.022902 + Test R²: 0.954934 + + Optimizer: ADAM + Best Fold: 1 + Val RMSE: 0.032558 ± 0.001222 + Val MAE: 0.023490 ± 0.000625 + Val R²: 0.949678 ± 0.004671 + Test RMSE: 0.031496 + Test MAE: 0.023255 + Test R²: 0.953484 + +CONV1D_V3_SGD +------------------------------ +Best Fold: 1 + +Validation Metrics (mean ± std): + val_mse: 0.001037 ± 0.000044 + val_rmse: 0.032199 ± 0.000687 + val_mae: 0.023269 ± 0.000512 + val_r2: 0.950863 ± 0.002815 + val_head_mse: 0.001035 ± 0.000038 + val_head_rmse: 0.032165 ± 0.000595 + val_head_mae: 0.024127 ± 0.000453 + val_head_r2: 0.895680 ± 0.005592 + val_left_shoulder_mse: 0.000680 ± 0.000055 + val_left_shoulder_rmse: 0.026048 ± 0.001072 + val_left_shoulder_mae: 0.019588 ± 0.000869 + val_left_shoulder_r2: 0.877679 ± 0.007086 + val_left_elbow_mse: 0.001063 ± 0.000086 + val_left_elbow_rmse: 0.032583 ± 0.001344 + val_left_elbow_mae: 0.024828 ± 0.000743 + val_left_elbow_r2: 0.922840 ± 0.006572 + val_right_shoulder_mse: 0.000604 ± 0.000060 + val_right_shoulder_rmse: 0.024546 ± 0.001209 + val_right_shoulder_mae: 0.018706 ± 0.000857 + val_right_shoulder_r2: 0.907168 ± 0.009598 + val_right_elbow_mse: 0.001032 ± 0.000077 + val_right_elbow_rmse: 0.032109 ± 0.001174 + val_right_elbow_mae: 0.024287 ± 0.000955 + val_right_elbow_r2: 0.934197 ± 0.007436 + val_left_hand_mse: 0.001620 ± 0.000184 + val_left_hand_rmse: 0.040187 ± 0.002285 + val_left_hand_mae: 0.030913 ± 0.001249 + val_left_hand_r2: 0.938056 ± 0.008553 + val_right_hand_mse: 0.001769 ± 0.000161 + val_right_hand_rmse: 0.042014 ± 0.001889 + val_right_hand_mae: 0.031644 ± 0.001572 + val_right_hand_r2: 0.936179 ± 0.007757 + val_left_hip_mse: 0.000776 ± 0.000051 + val_left_hip_rmse: 0.027837 ± 0.000907 + val_left_hip_mae: 0.020932 ± 0.000701 + val_left_hip_r2: 0.900056 ± 0.008405 + val_right_hip_mse: 0.000769 ± 0.000039 + val_right_hip_rmse: 0.027713 ± 0.000706 + val_right_hip_mae: 0.020752 ± 0.000574 + val_right_hip_r2: 0.904804 ± 0.007449 + val_left_knee_mse: 0.000880 ± 0.000026 + val_left_knee_rmse: 0.029654 ± 0.000436 + val_left_knee_mae: 0.021512 ± 0.000680 + val_left_knee_r2: 0.907015 ± 0.004386 + val_right_knee_mse: 0.000918 ± 0.000113 + val_right_knee_rmse: 0.030238 ± 0.001811 + val_right_knee_mae: 0.021672 ± 0.001468 + val_right_knee_r2: 0.914746 ± 0.008416 + val_left_foot_mse: 0.001171 ± 0.000354 + val_left_foot_rmse: 0.033795 ± 0.005366 + val_left_foot_mae: 0.021516 ± 0.001510 + val_left_foot_r2: 0.736253 ± 0.058586 + val_right_foot_mse: 0.001168 ± 0.000203 + val_right_foot_rmse: 0.034064 ± 0.002847 + val_right_foot_mae: 0.022016 ± 0.001029 + val_right_foot_r2: 0.737463 ± 0.054871 + +Test Metrics: + test_mse: 0.001016 + test_rmse: 0.031876 + test_mae: 0.023561 + test_r2: 0.952354 + test_head_mse: 0.001089 + test_head_rmse: 0.033004 + test_head_mae: 0.025484 + test_head_r2: 0.893086 + test_left_shoulder_mse: 0.000669 + test_left_shoulder_rmse: 0.025860 + test_left_shoulder_mae: 0.019165 + test_left_shoulder_r2: 0.882686 + test_left_elbow_mse: 0.000974 + test_left_elbow_rmse: 0.031204 + test_left_elbow_mae: 0.023773 + test_left_elbow_r2: 0.928136 + test_right_shoulder_mse: 0.000681 + test_right_shoulder_rmse: 0.026096 + test_right_shoulder_mae: 0.019669 + test_right_shoulder_r2: 0.897709 + test_right_elbow_mse: 0.001076 + test_right_elbow_rmse: 0.032800 + test_right_elbow_mae: 0.025630 + test_right_elbow_r2: 0.931191 + test_left_hand_mse: 0.001508 + test_left_hand_rmse: 0.038833 + test_left_hand_mae: 0.030160 + test_left_hand_r2: 0.942419 + test_right_hand_mse: 0.001782 + test_right_hand_rmse: 0.042212 + test_right_hand_mae: 0.033125 + test_right_hand_r2: 0.936674 + test_left_hip_mse: 0.000800 + test_left_hip_rmse: 0.028293 + test_left_hip_mae: 0.020973 + test_left_hip_r2: 0.909178 + test_right_hip_mse: 0.000833 + test_right_hip_rmse: 0.028859 + test_right_hip_mae: 0.020947 + test_right_hip_r2: 0.909111 + test_left_knee_mse: 0.000877 + test_left_knee_rmse: 0.029615 + test_left_knee_mae: 0.021918 + test_left_knee_r2: 0.909856 + test_right_knee_mse: 0.000856 + test_right_knee_rmse: 0.029260 + test_right_knee_mae: 0.022018 + test_right_knee_r2: 0.920163 + test_left_foot_mse: 0.000972 + test_left_foot_rmse: 0.031176 + test_left_foot_mae: 0.021580 + test_left_foot_r2: 0.791005 + test_right_foot_mse: 0.001092 + test_right_foot_rmse: 0.033046 + test_right_foot_mae: 0.021856 + test_right_foot_r2: 0.791975 + +CONV1D_V3_RMSPROP +------------------------------ +Best Fold: 1 + +Validation Metrics (mean ± std): + val_mse: 0.001125 ± 0.000116 + val_rmse: 0.033498 ± 0.001768 + val_mae: 0.024131 ± 0.001299 + val_r2: 0.946803 ± 0.004864 + val_head_mse: 0.001098 ± 0.000129 + val_head_rmse: 0.033083 ± 0.001939 + val_head_mae: 0.025050 ± 0.001361 + val_head_r2: 0.889771 ± 0.008997 + val_left_shoulder_mse: 0.000704 ± 0.000108 + val_left_shoulder_rmse: 0.026460 ± 0.002020 + val_left_shoulder_mae: 0.019890 ± 0.001160 + val_left_shoulder_r2: 0.873746 ± 0.013880 + val_left_elbow_mse: 0.001155 ± 0.000140 + val_left_elbow_rmse: 0.033916 ± 0.002148 + val_left_elbow_mae: 0.025739 ± 0.001325 + val_left_elbow_r2: 0.916250 ± 0.009959 + val_right_shoulder_mse: 0.000638 ± 0.000075 + val_right_shoulder_rmse: 0.025218 ± 0.001497 + val_right_shoulder_mae: 0.019061 ± 0.001042 + val_right_shoulder_r2: 0.902256 ± 0.008957 + val_right_elbow_mse: 0.001143 ± 0.000125 + val_right_elbow_rmse: 0.033756 ± 0.001862 + val_right_elbow_mae: 0.025279 ± 0.001697 + val_right_elbow_r2: 0.927377 ± 0.008384 + val_left_hand_mse: 0.001837 ± 0.000243 + val_left_hand_rmse: 0.042758 ± 0.002914 + val_left_hand_mae: 0.032579 ± 0.001698 + val_left_hand_r2: 0.929755 ± 0.010842 + val_right_hand_mse: 0.002005 ± 0.000308 + val_right_hand_rmse: 0.044643 ± 0.003456 + val_right_hand_mae: 0.033047 ± 0.002472 + val_right_hand_r2: 0.927713 ± 0.012356 + val_left_hip_mse: 0.000846 ± 0.000131 + val_left_hip_rmse: 0.029005 ± 0.002270 + val_left_hip_mae: 0.022026 ± 0.001506 + val_left_hip_r2: 0.891469 ± 0.014882 + val_right_hip_mse: 0.000841 ± 0.000119 + val_right_hip_rmse: 0.028927 ± 0.002073 + val_right_hip_mae: 0.021863 ± 0.001375 + val_right_hip_r2: 0.896213 ± 0.013786 + val_left_knee_mse: 0.000902 ± 0.000108 + val_left_knee_rmse: 0.029982 ± 0.001811 + val_left_knee_mae: 0.021776 ± 0.001513 + val_left_knee_r2: 0.904945 ± 0.009092 + val_right_knee_mse: 0.000980 ± 0.000200 + val_right_knee_rmse: 0.031152 ± 0.003094 + val_right_knee_mae: 0.022426 ± 0.001755 + val_right_knee_r2: 0.909381 ± 0.014644 + val_left_foot_mse: 0.001227 ± 0.000253 + val_left_foot_rmse: 0.034857 ± 0.003530 + val_left_foot_mae: 0.021985 ± 0.001125 + val_left_foot_r2: 0.719835 ± 0.037416 + val_right_foot_mse: 0.001251 ± 0.000189 + val_right_foot_rmse: 0.035272 ± 0.002583 + val_right_foot_mae: 0.022987 ± 0.000860 + val_right_foot_r2: 0.720678 ± 0.038397 + +Test Metrics: + test_mse: 0.000961 + test_rmse: 0.031001 + test_mae: 0.022902 + test_r2: 0.954934 + test_head_mse: 0.000982 + test_head_rmse: 0.031331 + test_head_mae: 0.024243 + test_head_r2: 0.903656 + test_left_shoulder_mse: 0.000644 + test_left_shoulder_rmse: 0.025372 + test_left_shoulder_mae: 0.019273 + test_left_shoulder_r2: 0.887076 + test_left_elbow_mse: 0.000917 + test_left_elbow_rmse: 0.030277 + test_left_elbow_mae: 0.023093 + test_left_elbow_r2: 0.932344 + test_right_shoulder_mse: 0.000646 + test_right_shoulder_rmse: 0.025408 + test_right_shoulder_mae: 0.019159 + test_right_shoulder_r2: 0.903033 + test_right_elbow_mse: 0.001015 + test_right_elbow_rmse: 0.031858 + test_right_elbow_mae: 0.024567 + test_right_elbow_r2: 0.935087 + test_left_hand_mse: 0.001424 + test_left_hand_rmse: 0.037735 + test_left_hand_mae: 0.029284 + test_left_hand_r2: 0.945630 + test_right_hand_mse: 0.001727 + test_right_hand_rmse: 0.041557 + test_right_hand_mae: 0.031816 + test_right_hand_r2: 0.938625 + test_left_hip_mse: 0.000733 + test_left_hip_rmse: 0.027076 + test_left_hip_mae: 0.020526 + test_left_hip_r2: 0.916825 + test_right_hip_mse: 0.000772 + test_right_hip_rmse: 0.027792 + test_right_hip_mae: 0.020441 + test_right_hip_r2: 0.915713 + test_left_knee_mse: 0.000832 + test_left_knee_rmse: 0.028846 + test_left_knee_mae: 0.021188 + test_left_knee_r2: 0.914475 + test_right_knee_mse: 0.000825 + test_right_knee_rmse: 0.028723 + test_right_knee_mae: 0.021605 + test_right_knee_r2: 0.923065 + test_left_foot_mse: 0.000936 + test_left_foot_rmse: 0.030594 + test_left_foot_mae: 0.020926 + test_left_foot_r2: 0.798741 + test_right_foot_mse: 0.001042 + test_right_foot_rmse: 0.032282 + test_right_foot_mae: 0.021601 + test_right_foot_r2: 0.801494 + +CONV1D_V3_ADAM +------------------------------ +Best Fold: 1 + +Validation Metrics (mean ± std): + val_mse: 0.001061 ± 0.000080 + val_rmse: 0.032558 ± 0.001222 + val_mae: 0.023490 ± 0.000625 + val_r2: 0.949678 ± 0.004671 + val_head_mse: 0.001057 ± 0.000050 + val_head_rmse: 0.032506 ± 0.000775 + val_head_mae: 0.024242 ± 0.000888 + val_head_r2: 0.893498 ± 0.005390 + val_left_shoulder_mse: 0.000687 ± 0.000055 + val_left_shoulder_rmse: 0.026196 ± 0.001032 + val_left_shoulder_mae: 0.019805 ± 0.000777 + val_left_shoulder_r2: 0.876301 ± 0.006259 + val_left_elbow_mse: 0.001092 ± 0.000116 + val_left_elbow_rmse: 0.033003 ± 0.001775 + val_left_elbow_mae: 0.025224 ± 0.001185 + val_left_elbow_r2: 0.920813 ± 0.008027 + val_right_shoulder_mse: 0.000620 ± 0.000035 + val_right_shoulder_rmse: 0.024899 ± 0.000708 + val_right_shoulder_mae: 0.019004 ± 0.000565 + val_right_shoulder_r2: 0.904604 ± 0.007138 + val_right_elbow_mse: 0.001087 ± 0.000077 + val_right_elbow_rmse: 0.032954 ± 0.001168 + val_right_elbow_mae: 0.024840 ± 0.000611 + val_right_elbow_r2: 0.930762 ± 0.006868 + val_left_hand_mse: 0.001702 ± 0.000200 + val_left_hand_rmse: 0.041186 ± 0.002421 + val_left_hand_mae: 0.031638 ± 0.001225 + val_left_hand_r2: 0.934933 ± 0.009125 + val_right_hand_mse: 0.001837 ± 0.000114 + val_right_hand_rmse: 0.042840 ± 0.001324 + val_right_hand_mae: 0.031801 ± 0.000866 + val_right_hand_r2: 0.933713 ± 0.006597 + val_left_hip_mse: 0.000794 ± 0.000070 + val_left_hip_rmse: 0.028142 ± 0.001231 + val_left_hip_mae: 0.021356 ± 0.000849 + val_left_hip_r2: 0.897912 ± 0.009256 + val_right_hip_mse: 0.000786 ± 0.000069 + val_right_hip_rmse: 0.028011 ± 0.001216 + val_right_hip_mae: 0.021144 ± 0.000849 + val_right_hip_r2: 0.902697 ± 0.009752 + val_left_knee_mse: 0.000878 ± 0.000057 + val_left_knee_rmse: 0.029619 ± 0.000964 + val_left_knee_mae: 0.021396 ± 0.000713 + val_left_knee_r2: 0.906944 ± 0.009307 + val_right_knee_mse: 0.000904 ± 0.000168 + val_right_knee_rmse: 0.029953 ± 0.002670 + val_right_knee_mae: 0.021673 ± 0.001551 + val_right_knee_r2: 0.916034 ± 0.013548 + val_left_foot_mse: 0.001139 ± 0.000358 + val_left_foot_rmse: 0.033324 ± 0.005357 + val_left_foot_mae: 0.020969 ± 0.001667 + val_left_foot_r2: 0.742631 ± 0.060462 + val_right_foot_mse: 0.001214 ± 0.000259 + val_right_foot_rmse: 0.034657 ± 0.003636 + val_right_foot_mae: 0.022278 ± 0.001070 + val_right_foot_r2: 0.727374 ± 0.064691 + +Test Metrics: + test_mse: 0.000992 + test_rmse: 0.031496 + test_mae: 0.023255 + test_r2: 0.953484 + test_head_mse: 0.001028 + test_head_rmse: 0.032069 + test_head_mae: 0.024457 + test_head_r2: 0.899060 + test_left_shoulder_mse: 0.000682 + test_left_shoulder_rmse: 0.026116 + test_left_shoulder_mae: 0.019677 + test_left_shoulder_r2: 0.880357 + test_left_elbow_mse: 0.000944 + test_left_elbow_rmse: 0.030731 + test_left_elbow_mae: 0.023302 + test_left_elbow_r2: 0.930298 + test_right_shoulder_mse: 0.000662 + test_right_shoulder_rmse: 0.025733 + test_right_shoulder_mae: 0.019634 + test_right_shoulder_r2: 0.900537 + test_right_elbow_mse: 0.001062 + test_right_elbow_rmse: 0.032594 + test_right_elbow_mae: 0.025117 + test_right_elbow_r2: 0.932051 + test_left_hand_mse: 0.001478 + test_left_hand_rmse: 0.038446 + test_left_hand_mae: 0.029303 + test_left_hand_r2: 0.943562 + test_right_hand_mse: 0.001863 + test_right_hand_rmse: 0.043167 + test_right_hand_mae: 0.032781 + test_right_hand_r2: 0.933778 + test_left_hip_mse: 0.000748 + test_left_hip_rmse: 0.027352 + test_left_hip_mae: 0.020815 + test_left_hip_r2: 0.915119 + test_right_hip_mse: 0.000771 + test_right_hip_rmse: 0.027764 + test_right_hip_mae: 0.020667 + test_right_hip_r2: 0.915880 + test_left_knee_mse: 0.000813 + test_left_knee_rmse: 0.028515 + test_left_knee_mae: 0.021313 + test_left_knee_r2: 0.916425 + test_right_knee_mse: 0.000822 + test_right_knee_rmse: 0.028678 + test_right_knee_mae: 0.022045 + test_right_knee_r2: 0.923304 + test_left_foot_mse: 0.000916 + test_left_foot_rmse: 0.030272 + test_left_foot_mae: 0.020706 + test_left_foot_r2: 0.802957 + test_right_foot_mse: 0.001104 + test_right_foot_rmse: 0.033229 + test_right_foot_mae: 0.022503 + test_right_foot_r2: 0.789667 diff --git a/A9/cv_training.py b/A9/cv_training.py new file mode 100644 index 0000000000000000000000000000000000000000..f7da086adbcb457ddad8054a94b00513627e0637 --- /dev/null +++ b/A9/cv_training.py @@ -0,0 +1,755 @@ +""" +Cross-Validation Training Script for Kinect Movement Prediction Models + +This script trains all models from models.py using k-fold cross-validation +and provides comprehensive evaluation metrics and visualizations. + +1. **K-Fold Cross-Validation** - Uses 5-fold CV with shuffled splits +2. **Multiple Model Training** - Trains Dense, Conv1D, LSTM, and GRU models +3. **Comprehensive Metrics** - Calculates MSE, RMSE, MAE, and R² scores +4. **Early Stopping** - Prevents overfitting with patience-based stopping +5. **Learning Rate Reduction** - Automatically reduces LR on plateau +6. **Model Checkpointing** - Saves best model weights for each fold +7. **Visualization** - Creates training history plots and prediction analysis +8. **Result Export** - Saves all results to JSON and text files + +Data Flow: + 179 files → 24,005 frames (flat) / 3,831 sequences (windowed) + ↓ + 90% train+val | 10% test + ↓ + 5-fold CV on train+val + ↓ + Select best fold (by val RMSE) + ↓ + Evaluate best model on test set + +""" + +import os +import sys +import numpy as np +import pandas as pd +import matplotlib.pyplot as plt +from datetime import datetime +import json +import warnings +warnings.filterwarnings('ignore') + +# Import tensorflow +import tensorflow as tf +from tensorflow import keras + +# Add the current directory to path to import models +sys.path.append(os.path.dirname(os.path.abspath(__file__))) + +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 +) + +# Import scikit-learn for cross-validation +from sklearn.model_selection import KFold +from sklearn.metrics import mean_squared_error, mean_absolute_error, r2_score + +# Set random seeds for reproducibility +np.random.seed(42) +tf.random.set_seed(42) + +# Enable XLA Compilation (optional) +tf.config.optimizer.set_jit(True) # Enable XLA + +# Configuration +N_SPLITS = 5 +BATCH_SIZE = 32 +EPOCHS = 100 +PATIENCE = 10 +WINDOW_SIZE = 30 + +# Optimizer configurations with learning rate settings +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' + }, +} + +# Model configurations +MODEL_CONFIGS = { + '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', + } +} + +# Loss functions to test +LOSS_FUNCTIONS = { + 'mse': { + 'name': 'Mean Squared Error', + 'description': 'MSE penalizes larger errors more heavily' + }, + 'mae': { + 'name': 'Mean Absolute Error', + 'description': 'MAE treats all errors equally' + } +} + + +def create_callbacks(model_name, fold, save_dir): + """Create early stopping and model checkpoint callbacks.""" + checkpoint_path = os.path.join(save_dir, f'{model_name}_fold{fold}_best.h5') + + early_stopping = tf.keras.callbacks.EarlyStopping( + monitor='val_loss', + patience=PATIENCE, + restore_best_weights=True, + verbose=1 + ) + + model_checkpoint = tf.keras.callbacks.ModelCheckpoint( + checkpoint_path, + monitor='val_loss', + save_best_only=True, + verbose=1 + ) + + reduce_lr = tf.keras.callbacks.ReduceLROnPlateau( + monitor='val_loss', + factor=0.5, + patience=7, + min_lr=1e-7, + verbose=1 + ) + + return [early_stopping, model_checkpoint, reduce_lr] + + +def evaluate_model(y_true, y_pred, prefix=""): + """Calculate comprehensive evaluation metrics.""" + # Flatten for frame-level metrics + y_true_flat = y_true.reshape(-1) + y_pred_flat = y_pred.reshape(-1) + + metrics = { + f'{prefix}mse': float(mean_squared_error(y_true_flat, y_pred_flat)), + f'{prefix}rmse': float(np.sqrt(mean_squared_error(y_true_flat, y_pred_flat))), + f'{prefix}mae': float(mean_absolute_error(y_true_flat, y_pred_flat)), + f'{prefix}r2': float(r2_score(y_true_flat, y_pred_flat)), + } + + # Per-joint metrics + for i, joint in enumerate(JOINTS): + joint_mse = float(mean_squared_error(y_true[:, i], y_pred[:, i])) + joint_rmse = float(np.sqrt(joint_mse)) + joint_mae = float(mean_absolute_error(y_true[:, i], y_pred[:, i])) + joint_r2 = float(r2_score(y_true[:, i], y_pred[:, i])) + + metrics[f'{prefix}{joint}_mse'] = joint_mse + metrics[f'{prefix}{joint}_rmse'] = joint_rmse + metrics[f'{prefix}{joint}_mae'] = joint_mae + metrics[f'{prefix}{joint}_r2'] = joint_r2 + + return metrics + + +def train_model_with_cv(X, y, model_name, model_config, save_dir, verbose=1, test_size=0.1): + """ + Train a model using k-fold cross-validation with a test split. + + Args: + X: Input data + y: Target data + model_name: Name of the model + model_config: Configuration dictionary for the model + save_dir: Directory to save results + verbose: Verbosity level + test_size: Proportion of data to use for testing (default 0.2) + + Returns: + Dictionary containing cross-validation results + """ + print(f"\n{'='*60}") + print(f"Training {model_name.upper()} with {N_SPLITS}-Fold Cross-Validation") + print(f"{'='*60}") + + # Prepare data based on data type + data_type = model_config['data_type'] + + if data_type == 'flat': + X_cv = X['flat'] + y_cv = y['flat'] + else: # windowed + X_cv = X['windowed'] + y_cv = y['windowed_last'] + + print(f"Data shape: X={X_cv.shape}, y={y_cv.shape}") + + # Split data into train+val and test sets + from sklearn.model_selection import train_test_split + X_trainval, X_test, y_trainval, y_test = train_test_split( + X_cv, y_cv, test_size=test_size, random_state=42, shuffle=True + ) + + print(f"Training+Validation samples: {len(X_trainval)}, Test samples: {len(X_test)}") + + # Initialize KFold on training+validation set + kfold = KFold(n_splits=N_SPLITS, shuffle=True, random_state=42) + + # Storage for results + fold_results = [] + history_list = [] + best_models = [] + best_val_metrics = [] # Store best validation metrics for each fold + + print(f"\nStarting {N_SPLITS}-fold cross-validation on training+validation set...") + + for fold, (train_idx, val_idx) in enumerate(kfold.split(X_trainval), 1): + print(f"\n{'='*40}") + print(f"Fold {fold}/{N_SPLITS}") + print(f"{'='*40}") + + # Split data + X_train, X_val = X_trainval[train_idx], X_trainval[val_idx] + y_train, y_val = y_trainval[train_idx], y_trainval[val_idx] + + print(f"Training samples: {len(X_train)}, Validation samples: {len(X_val)}") + + # Build model + model = model_config['build_fn'](**model_config['params']) + + # Get loss function from config, default to 'mse' + loss_function = model_config.get('loss_function', 'mse') + + model.compile( + optimizer='adam', + loss=loss_function, + metrics=['mae', 'mse'] + ) + + # Print model summary + if fold == 1 and verbose > 1: + model.summary() + + # Create callbacks + callbacks = create_callbacks(model_name, fold, save_dir) + + # Train model + history = model.fit( + X_train, y_train, + validation_data=(X_val, y_val), + epochs=EPOCHS, + batch_size=BATCH_SIZE, + callbacks=callbacks, + verbose=verbose + ) + + history_list.append(history) + best_models.append(model) + + # Evaluate on validation set + y_val_pred = model.predict(X_val, verbose=0) + val_metrics = evaluate_model(y_val, y_val_pred, prefix='val_') + best_val_metrics.append(val_metrics) # Store for later test evaluation + + # Evaluate on training set + y_train_pred = model.predict(X_train, verbose=0) + train_metrics = evaluate_model(y_train, y_train_pred, prefix='train_') + + fold_result = { + 'fold': fold, + 'train_metrics': train_metrics, + 'val_metrics': val_metrics, + 'best_epoch': len(history.history['loss']), + } + fold_results.append(fold_result) + + print(f"\nValidation Results:") + print(f" RMSE: {val_metrics['val_rmse']:.6f}") + print(f" MAE: {val_metrics['val_mae']:.6f}") + print(f" R²: {val_metrics['val_r2']:.6f}") + + # Calculate aggregate statistics + aggregate_results = calculate_aggregate_results(fold_results) + + # Find best fold based on validation RMSE and evaluate on test set + best_fold_idx = np.argmin([m['val_rmse'] for m in best_val_metrics]) + best_model = best_models[best_fold_idx] + best_fold_num = best_fold_idx + 1 + + print(f"\n{'='*40}") + print(f"Best Fold: {best_fold_num}") + print(f"{'='*40}") + + # Evaluate best model on test set + print(f"\nEvaluating best model on test set...") + y_test_pred = best_model.predict(X_test, verbose=0) + test_metrics = evaluate_model(y_test, y_test_pred, prefix='test_') + + print(f"\nTest Results:") + print(f" RMSE: {test_metrics['test_rmse']:.6f}") + print(f" MAE: {test_metrics['test_mae']:.6f}") + print(f" R²: {test_metrics['test_r2']:.6f}") + + # Save results + save_cv_results(fold_results, aggregate_results, model_name, save_dir, test_metrics=test_metrics) + + # Plot training history + plot_training_history(history_list, model_name, save_dir) + + # Plot predictions vs actual + plot_predictions(best_models, X_cv, y_cv, model_name, save_dir) + + # Plot test predictions + plot_test_predictions(best_model, X_test, y_test, model_name, save_dir) + + return { + 'fold_results': fold_results, + 'aggregate_results': aggregate_results, + 'history_list': history_list, + 'best_models': best_models, + 'best_fold': best_fold_num, + 'test_metrics': test_metrics, + } + + +def calculate_aggregate_results(fold_results): + """Calculate aggregate statistics across all folds.""" + aggregate = { + 'n_folds': len(fold_results), + 'train_metrics': {}, + 'val_metrics': {}, + } + + # Collect all metric keys + train_keys = fold_results[0]['train_metrics'].keys() + val_keys = fold_results[0]['val_metrics'].keys() + + # Calculate mean and std for each metric + for key in train_keys: + values = [fr['train_metrics'][key] for fr in fold_results] + aggregate['train_metrics'][key] = { + 'mean': float(np.mean(values)), + 'std': float(np.std(values)), + } + + for key in val_keys: + values = [fr['val_metrics'][key] for fr in fold_results] + aggregate['val_metrics'][key] = { + 'mean': float(np.mean(values)), + 'std': float(np.std(values)), + } + + # Best fold (based on val_loss) + best_fold_idx = np.argmin([fr['val_metrics']['val_mse'] for fr in fold_results]) + aggregate['best_fold'] = best_fold_idx + 1 + aggregate['best_fold_metrics'] = fold_results[best_fold_idx] + + return aggregate + + +def save_cv_results(fold_results, aggregate_results, model_name, save_dir, test_metrics=None): + """Save cross-validation results to JSON file.""" + results_path = os.path.join(save_dir, f'{model_name}_cv_results.json') + + # Convert to serializable format + output = { + 'model_name': model_name, + 'n_splits': N_SPLITS, + 'epochs': EPOCHS, + 'batch_size': BATCH_SIZE, + 'patience': PATIENCE, + 'aggregate_results': aggregate_results, + 'fold_details': fold_results, + } + + # Add test metrics if available + if test_metrics is not None: + output['test_metrics'] = test_metrics + + with open(results_path, 'w') as f: + json.dump(output, f, indent=2, default=str) + + print(f"\nResults saved to: {results_path}") + + +def plot_training_history(history_list, model_name, save_dir): + """Plot training history for all folds.""" + fig, axes = plt.subplots(2, 2, figsize=(15, 10)) + fig.suptitle(f'{model_name.upper()} - Training History across Folds', fontsize=14) + + colors = plt.cm.tab10(np.linspace(0, 1, len(history_list))) + + for i, history in enumerate(history_list): + color = colors[i] + + # Loss + axes[0, 0].plot(history.history['loss'], label=f'Fold {i+1} (train)', + color=color, linestyle='-') + axes[0, 0].plot(history.history['val_loss'], label=f'Fold {i+1} (val)', + color=color, linestyle='--') + + # MAE + axes[0, 1].plot(history.history['mae'], label=f'Fold {i+1} (train)', + color=color, linestyle='-') + axes[0, 1].plot(history.history['val_mae'], label=f'Fold {i+1} (val)', + color=color, linestyle='--') + + axes[0, 0].set_xlabel('Epoch') + axes[0, 0].set_ylabel('Loss (MSE)') + axes[0, 0].set_title('Training and Validation Loss') + axes[0, 0].legend(loc='upper right', fontsize=8) + axes[0, 0].grid(True) + + axes[0, 1].set_xlabel('Epoch') + axes[0, 1].set_ylabel('MAE') + axes[0, 1].set_title('Training and Validation MAE') + axes[0, 1].legend(loc='upper right', fontsize=8) + axes[0, 1].grid(True) + + # Learning rate (if available) + if 'lr' in history.history: + axes[1, 0].plot(history.history['lr']) + axes[1, 0].set_xlabel('Epoch') + axes[1, 0].set_ylabel('Learning Rate') + axes[1, 0].set_title('Learning Rate Schedule') + axes[1, 0].grid(True) + + # Combined plot + for i, history in enumerate(history_list): + color = colors[i] + axes[1, 1].plot(history.history['val_loss'], label=f'Fold {i+1}', + color=color, alpha=0.7) + + axes[1, 1].set_xlabel('Epoch') + axes[1, 1].set_ylabel('Validation Loss') + axes[1, 1].set_title('Validation Loss per Fold') + axes[1, 1].legend(loc='upper right', fontsize=8) + axes[1, 1].grid(True) + + plt.tight_layout() + plt.savefig(os.path.join(save_dir, f'{model_name}_training_history.png'), dpi=150) + plt.close() + + +def plot_predictions(best_models, X, y, model_name, save_dir): + """Plot predicted vs actual values.""" + # Use the best model (first fold's best model as representative) + model = best_models[0] + + # Predict + y_pred = model.predict(X, verbose=0) + + # Flatten for scatter plot + y_true_flat = y.reshape(-1) + y_pred_flat = y_pred.reshape(-1) + + # Create figure + fig, axes = plt.subplots(2, 2, figsize=(15, 12)) + fig.suptitle(f'{model_name.upper()} - Predictions vs Actual', fontsize=14) + + # Scatter plot + axes[0, 0].scatter(y_true_flat, y_pred_flat, alpha=0.3, s=1) + axes[0, 0].plot([y_true_flat.min(), y_true_flat.max()], + [y_true_flat.min(), y_true_flat.max()], 'r--', lw=2) + axes[0, 0].set_xlabel('Actual') + axes[0, 0].set_ylabel('Predicted') + axes[0, 0].set_title('Predicted vs Actual (All Joints)') + axes[0, 0].grid(True) + + # Per-joint scatter + for i, joint in enumerate(JOINTS[:6]): # Plot first 6 joints + axes[0, 1].scatter(y[:, i], y_pred[:, i], alpha=0.5, s=1, label=joint) + axes[0, 1].plot([y[:, :6].min(), y[:, :6].max()], + [y[:, :6].min(), y[:, :6].max()], 'r--', lw=2) + axes[0, 1].set_xlabel('Actual') + axes[0, 1].set_ylabel('Predicted') + axes[0, 1].set_title('Predicted vs Actual (First 6 Joints)') + axes[0, 1].legend(loc='upper right', fontsize=8) + axes[0, 1].grid(True) + + # Residual plot + residuals = y_true_flat - y_pred_flat + axes[1, 0].scatter(y_pred_flat, residuals, alpha=0.3, s=1) + axes[1, 0].axhline(y=0, color='r', linestyle='--', lw=2) + axes[1, 0].set_xlabel('Predicted') + axes[1, 0].set_ylabel('Residual') + axes[1, 0].set_title('Residual Plot') + axes[1, 0].grid(True) + + # Histogram of residuals + axes[1, 1].hist(residuals, bins=50, edgecolor='black', alpha=0.7) + axes[1, 1].set_xlabel('Residual') + axes[1, 1].set_ylabel('Frequency') + axes[1, 1].set_title(f'Residual Distribution (Mean: {residuals.mean():.4f}, Std: {residuals.std():.4f})') + axes[1, 1].grid(True) + + plt.tight_layout() + plt.savefig(os.path.join(save_dir, f'{model_name}_predictions.png'), dpi=150) + plt.close() + + +def plot_test_predictions(model, X_test, y_test, model_name, save_dir): + """Plot test predictions vs actual values.""" + # Predict on test set + y_test_pred = model.predict(X_test, verbose=0) + + # Flatten for scatter plot + y_true_flat = y_test.reshape(-1) + y_pred_flat = y_test_pred.reshape(-1) + + # Create figure + fig, axes = plt.subplots(2, 2, figsize=(15, 12)) + fig.suptitle(f'{model_name.upper()} - Test Set Predictions', fontsize=14) + + # Scatter plot + axes[0, 0].scatter(y_true_flat, y_pred_flat, alpha=0.3, s=1) + axes[0, 0].plot([y_true_flat.min(), y_true_flat.max()], + [y_true_flat.min(), y_true_flat.max()], 'r--', lw=2) + axes[0, 0].set_xlabel('Actual') + axes[0, 0].set_ylabel('Predicted') + axes[0, 0].set_title('Test Set: Predicted vs Actual (All Joints)') + axes[0, 0].grid(True) + + # Per-joint scatter (first 6 joints) + for i, joint in enumerate(JOINTS[:6]): + axes[0, 1].scatter(y_test[:, i], y_test_pred[:, i], alpha=0.5, s=1, label=joint) + axes[0, 1].plot([y_test[:, :6].min(), y_test[:, :6].max()], + [y_test[:, :6].min(), y_test[:, :6].max()], 'r--', lw=2) + axes[0, 1].set_xlabel('Actual') + axes[0, 1].set_ylabel('Predicted') + axes[0, 1].set_title('Test Set: Predicted vs Actual (First 6 Joints)') + axes[0, 1].legend(loc='upper right', fontsize=8) + axes[0, 1].grid(True) + + # Residual plot + residuals = y_true_flat - y_pred_flat + axes[1, 0].scatter(y_pred_flat, residuals, alpha=0.3, s=1) + axes[1, 0].axhline(y=0, color='r', linestyle='--', lw=2) + axes[1, 0].set_xlabel('Predicted') + axes[1, 0].set_ylabel('Residual') + axes[1, 0].set_title('Test Set: Residual Plot') + axes[1, 0].grid(True) + + # Histogram of residuals + axes[1, 1].hist(residuals, bins=50, edgecolor='black', alpha=0.7) + axes[1, 1].set_xlabel('Residual') + axes[1, 1].set_ylabel('Frequency') + axes[1, 1].set_title(f'Test Set: Residual Distribution (Mean: {residuals.mean():.4f}, Std: {residuals.std():.4f})') + axes[1, 1].grid(True) + + plt.tight_layout() + plt.savefig(os.path.join(save_dir, f'{model_name}_test_predictions.png'), dpi=150) + plt.close() + + + +def main(): + """Main function to run cross-validation for all models and optimizer variants.""" + print("="*60) + print("Cross-Validation Training for Kinect Movement Prediction") + print("Testing Multiple Optimizer Variants") + print("="*60) + + # Create save directory + timestamp = datetime.now().strftime("%Y%m%d_%H%M%S") + save_dir = os.path.join(os.path.dirname(os.path.abspath(__file__)), + f"cv_results_{timestamp}") + os.makedirs(save_dir, exist_ok=True) + + print(f"\nResults will be saved to: {save_dir}") + + # Load data + print("\n" + "="*60) + print("Loading Data") + print("="*60) + + REPO_ROOT = os.path.abspath(os.path.join(os.path.dirname(__file__), '..')) + DATA_DIR = os.path.join(REPO_ROOT, 'Datasets_all') + KINECT_DATA_PATH = os.path.join(DATA_DIR, 'kinect_good_preprocessed') + + sequences, file_names = load_all_sequences(KINECT_DATA_PATH) + + # Prepare data + X_flat, y_flat = flatten_sequences(sequences) + X_seq, y_seq = make_windowed_sequences(sequences, window_size=WINDOW_SIZE, stride=5) + y_seq_last = y_seq[:, -1, :] + + X = { + 'flat': X_flat, + 'windowed': X_seq, + } + y = { + 'flat': y_flat, + 'windowed_last': y_seq_last, + } + + print(f"\nFlat dataset: X={X_flat.shape} y={y_flat.shape}") + print(f"Windowed dataset: X={X_seq.shape} y_last={y_seq_last.shape}") + + # Define optimizer variants to test + optimizers_to_test = ['sgd', 'rmsprop', 'adam'] + + # Define loss functions to test + loss_functions_to_test = ['mse', 'mae'] + + print(f"\n{'='*60}") + print("Optimizer Variants to Test:") + print("="*60) + for opt_name in optimizers_to_test: + opt_config = OPTIMIZER_CONFIGS[opt_name] + print(f" - {opt_config['name']}: {opt_config['description']}") + print(f" Default LR: {opt_config['default_lr']}") + + print(f"\n{'='*60}") + print("Loss Functions to Test:") + print("="*60) + for loss_name in loss_functions_to_test: + loss_config = LOSS_FUNCTIONS[loss_name] + print(f" - {loss_name.upper()}: {loss_config['name']}") + print(f" {loss_config['description']}") + + # Train each model with each optimizer and loss function variant + all_results = {} + + for model_name, model_config in MODEL_CONFIGS.items(): + print(f"\n{'='*60}") + print(f"Testing Model: {model_name.upper()}") + print(f"{'='*60}") + + for optimizer_name in optimizers_to_test: + for loss_function in loss_functions_to_test: + # Create a copy of model config with optimizer and loss function specified + config_with_optimizer = model_config.copy() + config_with_optimizer['optimizer'] = optimizer_name + config_with_optimizer['loss_function'] = loss_function + + # Generate a unique run name for this optimizer variant + run_name = f"{model_name}_{optimizer_name}_{loss_function}" + + try: + print(f"\n{'='*60}") + print(f"Training {run_name.upper()}") + print(f"{'='*60}") + + results = train_model_with_cv( + X, y, run_name, config_with_optimizer, save_dir, verbose=1, test_size=0.2 + ) + all_results[run_name] = results + + # Clear session to free memory + tf.keras.backend.clear_session() + + except Exception as e: + print(f"\nError training {run_name}: {str(e)}") + import traceback + traceback.print_exc() + continue + + # Save summary + print("\n" + "="*60) + print("Cross-Validation Summary") + print("="*60) + + summary_path = os.path.join(save_dir, 'summary.txt') + with open(summary_path, 'w') as f: + f.write("Cross-Validation Training Summary - Optimizer & Loss Function Variants\n") + f.write("="*50 + "\n\n") + + for run_name, results in all_results.items(): + agg = results['aggregate_results'] + f.write(f"\n{run_name.upper()}\n") + f.write("-"*30 + "\n") + f.write(f"Best Fold: {agg['best_fold']}\n\n") + + f.write("Validation Metrics (mean ± std):\n") + for metric, values in agg['val_metrics'].items(): + if not metric.startswith('train_'): + f.write(f" {metric}: {values['mean']:.6f} ± {values['std']:.6f}\n") + + # Add test metrics if available + if 'test_metrics' in results: + f.write("\nTest Metrics:\n") + for metric, value in results['test_metrics'].items(): + if not metric.startswith('train_'): + f.write(f" {metric}: {value:.6f}\n") + + print(f"\nSummary saved to: {summary_path}") + + # Print summary to console + print("\n" + "="*60) + print("Final Results - Optimizer Comparison") + print("="*60) + + # Group results by model for comparison + results_by_model = {} + for run_name, results in all_results.items(): + model_name = run_name.rsplit('_', 2)[0] # Remove optimizer and loss suffix + if model_name not in results_by_model: + results_by_model[model_name] = [] + results_by_model[model_name].append((run_name, results)) + + for model_name, runs in results_by_model.items(): + print(f"\n{model_name.upper()}:") + print("-" * 40) + + for run_name, results in runs: + agg = results['aggregate_results'] + # Extract optimizer and loss function from run_name + parts = run_name.replace(model_name + '_', '').rsplit('_', 1) + optimizer_name = parts[0] + loss_function = parts[1] if len(parts) > 1 else 'unknown' + + print(f"\n Optimizer: {optimizer_name.upper()}, Loss: {loss_function.upper()}") + print(f" Best Fold: {agg['best_fold']}") + print(f" Val RMSE: {agg['val_metrics']['val_rmse']['mean']:.6f} ± {agg['val_metrics']['val_rmse']['std']:.6f}") + print(f" Val MAE: {agg['val_metrics']['val_mae']['mean']:.6f} ± {agg['val_metrics']['val_mae']['std']:.6f}") + print(f" Val R²: {agg['val_metrics']['val_r2']['mean']:.6f} ± {agg['val_metrics']['val_r2']['std']:.6f}") + + # Print test results if available + if 'test_metrics' in results: + print(f" Test RMSE: {results['test_metrics']['test_rmse']:.6f}") + print(f" Test MAE: {results['test_metrics']['test_mae']:.6f}") + print(f" Test R²: {results['test_metrics']['test_r2']:.6f}") + + print(f"\n\nAll results saved to: {save_dir}") + print("="*60) + + +if __name__ == "__main__": + main() diff --git a/A9/models.py b/A9/models.py new file mode 100644 index 0000000000000000000000000000000000000000..216b88258fcb97611d572a928f8a7d12118521bb --- /dev/null +++ b/A9/models.py @@ -0,0 +1,203 @@ +import os +import zipfile +import io +from pathlib import Path + +import numpy as np +import pandas as pd +import matplotlib.pyplot as plt + +import tensorflow as tf +from tensorflow import keras +from tensorflow.keras import layers, regularizers + +# Load Kinect movement data + +JOINTS = ["head", "left_shoulder", "left_elbow", "right_shoulder", "right_elbow", + "left_hand", "right_hand", "left_hip", "right_hip", "left_knee", "right_knee", + "left_foot", "right_foot", +] +N_JOINTS = len(JOINTS) # total number of body joints +N_INPUT = N_JOINTS * 2 # input for each joint(x and y) +N_OUTPUT = N_JOINTS * 1 # output/target z coordiante +print(f'Input: {N_INPUT} \tOutput:{N_OUTPUT}') +print(f'Joints: {N_JOINTS}') + +# Loads single csv file and splits into input and target array +def load_single_csv(filepath_or_bytes): + if isinstance(filepath_or_bytes, (str, os.PathLike)): + df = pd.read_csv(filepath_or_bytes) + else: + df = pd.read_csv(io.BytesIO(filepath_or_bytes)) + df.columns = df.columns.str.strip() + + x_cols = [f"{j}_x" for j in JOINTS] + y_cols = [f"{j}_y" for j in JOINTS] + z_cols = [f"{j}_z" for j in JOINTS] + + xy_cols = [] + for j in JOINTS: + xy_cols += [f"{j}_x", f"{j}_y"] + + X = df[xy_cols].values.astype(np.float32) # input + y = df[z_cols].values.astype(np.float32) # Target + + return X, y + +# load all csv file from the folder +def load_all_sequences(folder_path): + sequences, file_names = [], [] + + # Get all CSV files in the folder + csv_files = [f for f in os.listdir(folder_path) if f.endswith('.csv')] + csv_files.sort() + + print(f"Found {len(csv_files)} CSV files in folder.") + + for name in csv_files: + file_path = os.path.join(folder_path, name) + + with open(file_path, 'rb') as f: + raw = f.read() + + X, y = load_single_csv(raw) + + sequences.append((X, y)) # stores as (input,target) + file_names.append(name) + + return sequences, file_names + +# For Dense MLP model, which treats each frame independently +def flatten_sequences(sequences): + X_flat = np.concatenate([s[0] for s in sequences], axis=0) + y_flat = np.concatenate([s[1] for s in sequences], axis=0) + return X_flat, y_flat + +# Create fixed-length windows of consecutive frames from each session for conv1d, lstm and gru +def make_windowed_sequences(sequences, window_size=30, stride=1): + X_list, y_list = [], [] + for X, y in sequences: + n = len(X) + for start in range(0, n - window_size + 1, stride): + X_list.append(X[start : start + window_size]) + y_list.append(y[start : start + window_size]) + + X_seq = np.array(X_list, dtype=np.float32) # (N, window, 26) + y_seq = np.array(y_list, dtype=np.float32) # (N, window, 13) + return X_seq, y_seq + +REPO_ROOT = os.path.abspath(os.path.join(os.getcwd(), '..')) +DATA_DIR = os.path.join(REPO_ROOT, 'Datasets_all') +KINECT_DATA_PATH = os.path.join(DATA_DIR, 'kinect_good_preprocessed') + +# sequences contain list of tuples (X,y) +sequences, file_names = load_all_sequences(KINECT_DATA_PATH) + +# Frame-level flat data (for Dense models) +X_flat, y_flat = flatten_sequences(sequences) +print(f"\nFlat dataset: X={X_flat.shape} y={y_flat.shape}") + +# Windowed sequences (for Conv1D / LSTM / GRU models) +WINDOW_SIZE = 30 +X_seq, y_seq = make_windowed_sequences(sequences, window_size=WINDOW_SIZE, stride=5) +y_seq_last = y_seq[:, -1, :] # (N, 13) +print(f"Windowed dataset: X={X_seq.shape} y_last={y_seq_last.shape}") + +# Define Deep Learning network architectures + +# DEnse MLP + +def build_dense_model( hidden_units=(128, 64), activation="relu", dropout_rate=0.2, + l2_reg=1e-4,optimizer="adam", loss="mse", +): + inputs = keras.Input(shape=(N_INPUT,), name="xy_input") + x = inputs + for i, units in enumerate(hidden_units): + x = layers.Dense( + units, + activation=activation, + kernel_regularizer=regularizers.l2(l2_reg) if l2_reg else None, + name=f"dense_{i+1}", + )(x) + if dropout_rate > 0: + x = layers.Dropout(dropout_rate, name=f"dropout_{i+1}")(x) + + outputs = layers.Dense(N_OUTPUT, activation="linear", name="z_output")(x) + model = keras.Model(inputs, outputs, name="DenseModel") + return model + +# Conv1D CNN + +def build_conv1d_model(filters=(64, 128), kernel_size=3, pool_size=2, dense_units=(64,), + activation="relu", dropout_rate=0.2,optimizer="adam", loss="mse", +): + inputs = keras.Input(shape=(WINDOW_SIZE, N_INPUT), name="xy_seq_input") + x = inputs + for i, f in enumerate(filters): + x = layers.Conv1D(f, kernel_size, activation=activation, padding="same", + name=f"conv_{i+1}")(x) + x = layers.MaxPooling1D(pool_size, padding="same", name=f"pool_{i+1}")(x) + if dropout_rate > 0: + x = layers.Dropout(dropout_rate, name=f"drop_conv_{i+1}")(x) + + x = layers.GlobalAveragePooling1D(name="gap")(x) + + for i, units in enumerate(dense_units): + x = layers.Dense(units, activation=activation, name=f"fc_{i+1}")(x) + if dropout_rate > 0: + x = layers.Dropout(dropout_rate, name=f"drop_fc_{i+1}")(x) + + outputs = layers.Dense(N_OUTPUT, activation="linear", name="z_output")(x) + model = keras.Model(inputs, outputs, name="Conv1DModel") + return model + +# layers.LSTM + +def build_lstm_model(lstm_units=(64, 32), dense_units=(32,), activation="tanh", + dropout_rate=0.2, recurrent_dropout=0.0, optimizer="adam", loss="mse", +): + inputs = keras.Input(shape=(WINDOW_SIZE, N_INPUT), name="xy_seq_input") + x = inputs + for i, units in enumerate(lstm_units): + return_sequences = (i < len(lstm_units) - 1) + x = layers.LSTM( + units, + return_sequences=return_sequences, + dropout=dropout_rate, + recurrent_dropout=recurrent_dropout, + name=f"lstm_{i+1}", + )(x) + + for i, units in enumerate(dense_units): + x = layers.Dense(units, activation="relu", name=f"fc_{i+1}")(x) + if dropout_rate > 0: + x = layers.Dropout(dropout_rate, name=f"drop_fc_{i+1}")(x) + + outputs = layers.Dense(N_OUTPUT, activation="linear", name="z_output")(x) + model = keras.Model(inputs, outputs, name="LSTMModel") + return model + +# layers.GRU + +def build_gru_model(gru_units=(64, 32), dense_units=(32,), dropout_rate=0.2, + recurrent_dropout=0.0, optimizer="adam", loss="mse",): + inputs = keras.Input(shape=(WINDOW_SIZE, N_INPUT), name="xy_seq_input") + x = inputs + for i, units in enumerate(gru_units): + return_sequences = (i < len(gru_units) - 1) + x = layers.GRU( + units, + return_sequences=return_sequences, + dropout=dropout_rate, + recurrent_dropout=recurrent_dropout, + name=f"gru_{i+1}", + )(x) + + for i, units in enumerate(dense_units): + x = layers.Dense(units, activation="relu", name=f"fc_{i+1}")(x) + if dropout_rate > 0: + x = layers.Dropout(dropout_rate, name=f"drop_fc_{i+1}")(x) + + outputs = layers.Dense(N_OUTPUT, activation="linear", name="z_output")(x) + model = keras.Model(inputs, outputs, name="GRUModel") + return model diff --git a/A9/models/conv1d_v3_adam_mse_fold4_best.h5 b/A9/models/conv1d_v3_adam_mse_fold4_best.h5 new file mode 100644 index 0000000000000000000000000000000000000000..2b5cf3cdd52102654efb99139b0b33b647036448 --- /dev/null +++ b/A9/models/conv1d_v3_adam_mse_fold4_best.h5 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:1a093463228a0d9d95a8bf8658e90fa3823bf24ac70c118d57f2808199fc115e +size 2658520