From fdc49775e613036f6da3169a1298a28792aae018 Mon Sep 17 00:00:00 2001
From: lidanyang <lidanyang@fuzhi.ai>
Date: Fri, 24 Nov 2023 17:23:39 +0800
Subject: [PATCH] reduce hierarchy of machine learning

---
 metagpt/tools/functions/__init__.py                        | 2 +-
 .../libs/{machine_learning => }/feature_engineering.py     | 2 +-
 metagpt/tools/functions/libs/machine_learning/__init__.py  | 7 -------
 .../schemas/{machine_learning => }/feature_engineering.py  | 0
 .../tools/functions/schemas/machine_learning/__init__.py   | 6 ------
 5 files changed, 2 insertions(+), 15 deletions(-)
 rename metagpt/tools/functions/libs/{machine_learning => }/feature_engineering.py (98%)
 delete mode 100644 metagpt/tools/functions/libs/machine_learning/__init__.py
 rename metagpt/tools/functions/schemas/{machine_learning => }/feature_engineering.py (100%)
 delete mode 100644 metagpt/tools/functions/schemas/machine_learning/__init__.py

diff --git a/metagpt/tools/functions/__init__.py b/metagpt/tools/functions/__init__.py
index 069e4297b..b81e85833 100644
--- a/metagpt/tools/functions/__init__.py
+++ b/metagpt/tools/functions/__init__.py
@@ -5,4 +5,4 @@
 # @File    : __init__.py
 # @Desc    :
 from metagpt.tools.functions.register.register import registry
-import metagpt.tools.functions.libs.machine_learning
+import metagpt.tools.functions.libs.feature_engineering
diff --git a/metagpt/tools/functions/libs/machine_learning/feature_engineering.py b/metagpt/tools/functions/libs/feature_engineering.py
similarity index 98%
rename from metagpt/tools/functions/libs/machine_learning/feature_engineering.py
rename to metagpt/tools/functions/libs/feature_engineering.py
index 584bd125d..0573f362d 100644
--- a/metagpt/tools/functions/libs/machine_learning/feature_engineering.py
+++ b/metagpt/tools/functions/libs/feature_engineering.py
@@ -11,7 +11,7 @@ from pandas.api.types import is_numeric_dtype
 from sklearn.preprocessing import PolynomialFeatures, OneHotEncoder
 
 from metagpt.tools.functions import registry
-from metagpt.tools.functions.schemas.machine_learning.feature_engineering import *
+from metagpt.tools.functions.schemas.feature_engineering import *
 
 
 @registry.register("feature_engineering", PolynomialExpansion)
diff --git a/metagpt/tools/functions/libs/machine_learning/__init__.py b/metagpt/tools/functions/libs/machine_learning/__init__.py
deleted file mode 100644
index 5e9760c64..000000000
--- a/metagpt/tools/functions/libs/machine_learning/__init__.py
+++ /dev/null
@@ -1,7 +0,0 @@
-#!/usr/bin/env python
-# -*- coding: utf-8 -*-
-# @Time    : 2023/11/16 16:36
-# @Author  : lidanyang
-# @File    : __init__.py
-# @Desc    :
-from metagpt.tools.functions.libs.machine_learning.feature_engineering import *
diff --git a/metagpt/tools/functions/schemas/machine_learning/feature_engineering.py b/metagpt/tools/functions/schemas/feature_engineering.py
similarity index 100%
rename from metagpt/tools/functions/schemas/machine_learning/feature_engineering.py
rename to metagpt/tools/functions/schemas/feature_engineering.py
diff --git a/metagpt/tools/functions/schemas/machine_learning/__init__.py b/metagpt/tools/functions/schemas/machine_learning/__init__.py
deleted file mode 100644
index c80872750..000000000
--- a/metagpt/tools/functions/schemas/machine_learning/__init__.py
+++ /dev/null
@@ -1,6 +0,0 @@
-#!/usr/bin/env python
-# -*- coding: utf-8 -*-
-# @Time    : 2023/11/16 16:37
-# @Author  : lidanyang
-# @File    : __init__.py
-# @Desc    :