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    :