函数来决定哪种分类方法与可选参数一起使用-Python

Question

我的目标是包括一个“ mod_type”参数，该参数指示要运行的模型类型，即决策树或knn，使用kwargs让用户通过可选的关键字params“ k” for KNN和决策树的“ max_depth” 。如果用户将其传递给它们，则在初始化模型时使用它们。返回模型对象。

为此，我正在使用以下功能：

from sklearn.tree import DecisionTreeClassifier
from sklearn.datasets import load_iris
from sklearn.model_selection import train_test_split
from sklearn.neighbors import KNeighborsClassifier
import pandas as pd
import numpy as np
from sklearn.neighbors import KNeighborsClassifier


def my_classification(x,y,mod_type,**kwargs):
    if mod_type == "dt":
        if max_d in kwargs.keys():
            dt = DecisionTreeClassifier(max_depth=max_d.values())
            dt.fit(x,y)
            return dt
        else:
            dt = DecisionTreeClassifier()
            dt.fit(x,y)     
            return dt       
    elif mod_type == "knn":
        if k in kwargs.keys():
            knn = KNeighborsClassifier(k.values())
            knn.fit(x,y)
            return knn
        else:
            knn = KNeighborsClassifier()
            knn.fit(x,y)
            return knn
    else:
        print("unavailable type")

iris = load_iris()
x = pd.DataFrame(iris.data)
y = iris.target
my_classification(x,y,"dt")

了解Kwargs并不容易，但我想我现在可能有它，错误它给我的是：name error：name'max_d'未定义。我已经尝试在功能之前创建它们，然后更改其中的功能，但它在没有任何更改的情况下打印了模型。

有人可以帮忙吗？

Answer 1

字典对象中的键相当于字符串。

if 'max_d' in kwargs.keys():

如果不是字符串，它会在 kwargs.keys() 列表中查找名为 max_d 的对象。

Answer 2

kwargs是一个命令，其名称为键，其值是值。

这就是您可以使用它的方式：

if mod_type == "dt":
    if "max_d" in kwargs:
        dt = DecisionTreeClassifier(max_depth=kwargs["max_d"])
        ...
elif mod_type == "knn":
    if "k" in kwargs:
        knn = KNeighborsClassifier(kwargs["k"])
        ...