diff --git a/model_selection/cv_metrics_distr.py b/model_selection/cv_metrics_distr.py
index 04d6c5bf9d15857951270fb96c25e4f7181ccde3..4ebbfc2775c17a37f325cec54929e0e0d1c2eecc 100644
--- a/model_selection/cv_metrics_distr.py
+++ b/model_selection/cv_metrics_distr.py
@@ -26,14 +26,23 @@ if __name__ == "__main__":
else:
model_names = model_names_simple
# Create figure for current sheet, one row per metric
- fig, axes = plt.subplots(len(metric_names), 1, figsize=(15, 8 * len(metric_names)))
+ fig, axes = plt.subplots(len(metric_names), 1, figsize=(10, 10 * len(metric_names)))
for metric_id, metric_name in enumerate(metric_names):
# Get the axis for the current metric
ax = axes[metric_id]
+ # List to store the metric array for each model
+ metric_data = []
for model_name in model_names:
+ # Construct the row name
row_name = f'{model_name}_{metric_name}'
- # Collect data for the current model's metric
metric_row = df.loc[df['Unnamed: 0'] == row_name].iloc[0, 1:].values
- if group == 'pre' and method == '_ORIG' and metric_id == 0 and model_name == 'DT':
- print(metric_row)
+ metric_data.append(metric_row)
+ # Plot boxplot for the current metric across all models
+ ax.boxplot(metric_data, labels=model_names)
+ ax.set_title(f'{metric_name} for {group}{method}')
+ plt.tight_layout()
+ fig.savefig(f'./output_cv_metrics/distributions/{group}{method}.svg', format='svg', dpi=600)
+ plt.close(fig) # Close the figure to free up memory
+
+ print("Succesful distribution plots generation")
# --------------------------------------------------------------------------------------------------------
\ No newline at end of file
diff --git a/model_selection/output_cv_metrics/distributions/post_ORIG.svg b/model_selection/output_cv_metrics/distributions/post_ORIG.svg
new file mode 100644
index 0000000000000000000000000000000000000000..cbb514eaf76ad58bcf62d16267d844fc71172f11
--- /dev/null
+++ b/model_selection/output_cv_metrics/distributions/post_ORIG.svg
@@ -0,0 +1,7034 @@
+
+
+
diff --git a/model_selection/output_cv_metrics/distributions/post_ORIG_CW.svg b/model_selection/output_cv_metrics/distributions/post_ORIG_CW.svg
new file mode 100644
index 0000000000000000000000000000000000000000..90ac73ccf2f2ebb4cb0f66577e4e6b9b39c40db3
--- /dev/null
+++ b/model_selection/output_cv_metrics/distributions/post_ORIG_CW.svg
@@ -0,0 +1,6275 @@
+
+
+
diff --git a/model_selection/output_cv_metrics/distributions/post_OVER.svg b/model_selection/output_cv_metrics/distributions/post_OVER.svg
new file mode 100644
index 0000000000000000000000000000000000000000..0225c5e9053c934fe70213f58d9255e7cc9a9c96
--- /dev/null
+++ b/model_selection/output_cv_metrics/distributions/post_OVER.svg
@@ -0,0 +1,7003 @@
+
+
+
diff --git a/model_selection/output_cv_metrics/distributions/post_UNDER.svg b/model_selection/output_cv_metrics/distributions/post_UNDER.svg
new file mode 100644
index 0000000000000000000000000000000000000000..50ed9591673c3b324c857e68d3a1ce60d32ba3ef
--- /dev/null
+++ b/model_selection/output_cv_metrics/distributions/post_UNDER.svg
@@ -0,0 +1,7152 @@
+
+
+
diff --git a/model_selection/output_cv_metrics/distributions/pre_ORIG.svg b/model_selection/output_cv_metrics/distributions/pre_ORIG.svg
new file mode 100644
index 0000000000000000000000000000000000000000..c71850c3c13c1bfc88206c9fcd5826dd830754fe
--- /dev/null
+++ b/model_selection/output_cv_metrics/distributions/pre_ORIG.svg
@@ -0,0 +1,7169 @@
+
+
+
diff --git a/model_selection/output_cv_metrics/distributions/pre_ORIG_CW.svg b/model_selection/output_cv_metrics/distributions/pre_ORIG_CW.svg
new file mode 100644
index 0000000000000000000000000000000000000000..ff7a561736597b70013aeea3469b3c868e619aa7
--- /dev/null
+++ b/model_selection/output_cv_metrics/distributions/pre_ORIG_CW.svg
@@ -0,0 +1,5947 @@
+
+
+
diff --git a/model_selection/output_cv_metrics/distributions/pre_OVER.svg b/model_selection/output_cv_metrics/distributions/pre_OVER.svg
new file mode 100644
index 0000000000000000000000000000000000000000..c934b66badd0cc208ae2156915c12a24d2c98c79
--- /dev/null
+++ b/model_selection/output_cv_metrics/distributions/pre_OVER.svg
@@ -0,0 +1,6969 @@
+
+
+
diff --git a/model_selection/output_cv_metrics/distributions/pre_UNDER.svg b/model_selection/output_cv_metrics/distributions/pre_UNDER.svg
new file mode 100644
index 0000000000000000000000000000000000000000..bab7eb421fbf06b851278d7d70c14034051fa47e
--- /dev/null
+++ b/model_selection/output_cv_metrics/distributions/pre_UNDER.svg
@@ -0,0 +1,7009 @@
+
+
+