From 789285e31247b06141e35a739bf1ce93512c9db5 Mon Sep 17 00:00:00 2001
From: haotian <2421912570@qq.com>
Date: Mon, 17 Feb 2025 14:45:22 +0800
Subject: [PATCH] =?UTF-8?q?=E5=88=9D=E6=AD=A5=E5=AE=9E=E7=8E=B0--=E8=AF=BB?=
 =?UTF-8?q?=E5=8F=96method.md=E4=B8=AD=E7=9A=84=E6=96=B9=E6=B3=95?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 data_process/method_reader.py | 101 ++++++++++++++++++++++++++++++++++
 1 file changed, 101 insertions(+)
 create mode 100644 data_process/method_reader.py

diff --git a/data_process/method_reader.py b/data_process/method_reader.py
new file mode 100644
index 0000000..3d01904
--- /dev/null
+++ b/data_process/method_reader.py
@@ -0,0 +1,101 @@
+import yaml
+from typing import Dict, List
+import os
+import logging
+from pathlib import Path
+
+class MethodReader:
+    """方法配置读取器"""
+    
+    def __init__(self):
+        """初始化方法读取器"""
+        self.logger = logging.getLogger(__name__)
+        self.method_config = self._load_method_config()
+        
+    def _load_method_config(self) -> Dict:
+        """加载方法配置文件"""
+        try:
+            config_path = Path('date_preprocessing/method.yaml')
+            if not config_path.exists():
+                raise FileNotFoundError(f"Method config file not found at {config_path}")
+                
+            with open(config_path, 'r', encoding='utf-8') as f:
+                config = yaml.safe_load(f)
+                
+            self.logger.info("Successfully loaded method config")
+            return config
+            
+        except Exception as e:
+            self.logger.error(f"Error loading method config: {str(e)}")
+            raise
+            
+    def get_preprocessing_methods(self) -> Dict:
+        """获取预处理方法列表"""
+        try:
+            methods = []
+            
+            # 数据缩放方法
+            scaler_methods = list(self.method_config.get('data_scaler_methods', {}).keys())
+            if scaler_methods:
+                methods.append({
+                    "name": "data_scaler",
+                    "description": "数据缩放处理",
+                    "method": scaler_methods
+                })
+                
+            # 缺失值处理方法
+            missing_methods = list(self.method_config.get('missing_value_handling_methods', {}).keys())
+            if missing_methods:
+                methods.append({
+                    "name": "missing_value_handler",
+                    "description": "缺失值处理",
+                    "method": missing_methods
+                })
+                
+            # 异常值检测方法
+            outlier_methods = list(self.method_config.get('outlier_detection_methods', {}).keys())
+            if outlier_methods:
+                methods.append({
+                    "name": "outlier_detector",
+                    "description": "异常值检测",
+                    "method": outlier_methods
+                })
+                
+            return {
+                "status": "success",
+                "methods": methods
+            }
+            
+        except Exception as e:
+            self.logger.error(f"Error getting preprocessing methods: {str(e)}")
+            return {
+                "status": "error",
+                "error": str(e)
+            }
+            
+    def get_method_details(self, method_name: str) -> Dict:
+        """获取指定方法的详细信息"""
+        try:
+            # 在各个方法类别中查找
+            for category in ['data_scaler_methods', 'missing_value_handling_methods', 'outlier_detection_methods']:
+                if method_name in self.method_config.get(category, {}):
+                    method_info = self.method_config[category][method_name]
+                    return {
+                        "status": "success",
+                        "method": {
+                            "name": method_name,
+                            "principle": method_info.get('principle', ''),
+                            "advantages": method_info.get('advantages', []),
+                            "disadvantages": method_info.get('disadvantages', []),
+                            "applicable_scenarios": method_info.get('applicable_scenarios', [])
+                        }
+                    }
+                    
+            raise ValueError(f"Method {method_name} not found")
+            
+        except Exception as e:
+            self.logger.error(f"Error getting method details: {str(e)}")
+            return {
+                "status": "error",
+                "error": str(e)
+            } 
\ No newline at end of file