From 6496a6af8e29b4a83afd0db4d4c2562676e25f8a Mon Sep 17 00:00:00 2001 From: stochastic-sisyphus <102266523+stochastic-sisyphus@users.noreply.github.com> Date: Tue, 10 Dec 2024 04:32:01 +0000 Subject: [PATCH] Refactor main script imports; add new metric calculations and style parameter handling for enhanced functionality --- src/main.py | 14 ++++++++++++-- 1 file changed, 12 insertions(+), 2 deletions(-) diff --git a/src/main.py b/src/main.py index 241028b..6474037 100644 --- a/src/main.py +++ b/src/main.py @@ -17,8 +17,13 @@ from src.evaluation.metrics import EvaluationMetrics from src.clustering.dynamic_cluster_manager import DynamicClusterManager from src.utils.metrics_utils import calculate_cluster_metrics - from src.utils.style_selector import determine_cluster_style + from src.utils.style_selector import determine_cluster_style, get_style_parameters from src.utils.logging_config import setup_logging + from src.utils.metrics_utils import ( + calculate_cluster_variance, + calculate_lexical_diversity, + calculate_cluster_metrics + ) else: # Use relative imports when imported as module from .data_loader import DataLoader @@ -30,8 +35,13 @@ from .evaluation.metrics import EvaluationMetrics from .clustering.dynamic_cluster_manager import DynamicClusterManager from .utils.metrics_utils import calculate_cluster_metrics - from .utils.style_selector import determine_cluster_style + from .utils.style_selector import determine_cluster_style, get_style_parameters from .utils.logging_config import setup_logging + from .utils.metrics_utils import ( + calculate_cluster_variance, + calculate_lexical_diversity, + calculate_cluster_metrics + ) import yaml import pandas as pd