Python：使用导入函数时出现类型错误

Question

我有一个方法如下：

def importFrom(module, name): module = importlib.import_module(module) return getattr(module, name)

然后按如下方式使用：

def imputation_LR (df, name): reg = importFrom('sklearn.linear_model', name) reg.fit(X_train, y_train)

然后调用如下：

data = imputation_LR (data, 'LinearRegression')

并得到以下错误：
reg.fit(X_train, y_train) TypeError: fit() missing 1 required positional argument: 'y'

我觉得这是关于 LinearRegression / LinearRegression() 的事情，但无法弄清楚。

谢谢。

Answer 1

您需要创建一个LinearRegression 对象。您收到错误的原因是因为fit 是一个实例方法（所有这些都将self 作为第一个参数传递）。所以函数接口真的是fit(self, x, y)。 self 是一个特殊变量，在调用实例方法时由对象引用自动传入。试试这个（注意我添加了()，见内联评论）：

def imputation_LR (df, name): 
    reg = importFrom('sklearn.linear_model', name)() # Note I added () here
    reg.fit(X_train, y_train)

请注意，它位于此处记录的 API 的“方法”部分：http://scikit-learn.org/stable/modules/generated/sklearn.linear_model.LinearRegression.html。

如有疑问，请阅读文档。

附：目前还不完全清楚您为什么要尝试将模块作为字符串导入。您可以使用要获取的包/模块的名称导入：

from sklearn.linear_model import LinearRegression
reg = LinearRegression()
...