reduce hierarchy of machine learning

This commit is contained in:
lidanyang 2023-11-24 17:23:39 +08:00
parent 142b04fa76
commit fdc49775e6
5 changed files with 2 additions and 15 deletions

View file

@ -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

View file

@ -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)

View file

@ -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 *

View file

@ -1,6 +0,0 @@
#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Time : 2023/11/16 16:37
# @Author : lidanyang
# @File : __init__.py
# @Desc :